From 31a20a590c3952ec55b770995ac6624a7e3ecc5d Mon Sep 17 00:00:00 2001
From: ngfchl <ngfchl@126.com>
Date: Fri, 14 Oct 2022 21:22:23 +0800
Subject: [PATCH] =?UTF-8?q?=E6=AF=8F=E6=AC=A1=E5=90=AF=E5=8A=A8=E6=97=B6?=
 =?UTF-8?q?=E5=88=9D=E5=A7=8B=E5=8C=96=E7=AB=99=E7=82=B9=E4=BF=A1=E6=81=AF?=
 =?UTF-8?q?=E8=A1=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pt_site/apps.py | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/pt_site/apps.py b/pt_site/apps.py
index 56972bc..0100543 100644
--- a/pt_site/apps.py
+++ b/pt_site/apps.py
@@ -1,7 +1,33 @@
+import json
+import logging
+
 from django.apps import AppConfig
 
+from ptools.settings import BASE_DIR
+
+logger = logging.getLogger('ptools')
+
 
 class PtSiteConfig(AppConfig):
     default_auto_field = 'django.db.models.BigAutoField'
     name = 'pt_site'
     verbose_name = 'PT站点管理'
+
+    def ready(self):
+        from pt_site.models import Site
+        logger.info('初始化站点信息')
+        try:
+            with open('pt_site_site.json', 'r') as f:
+                # print(f.readlines())
+                data = json.load(f)
+                logger.info('更新规则中，返回结果为True为新建，为False为更新，其他是错误了')
+                for site_rules in data:
+                    if site_rules.get('pk'):
+                        del site_rules['pk']
+                    if site_rules.get('id'):
+                        del site_rules['id']
+                    site_obj = Site.objects.update_or_create(defaults=site_rules, url=site_rules.get('url'))
+                    msg = site_obj[0].name + (' 规则新增成功！' if site_obj[1] else '规则更新成功！')
+                    logger.info(msg)
+        except Exception as e:
+            logger.error('初始化站点信息出错！{}'.format(e))