优化每次同步数据库后初始化站点信息表

pt_site/apps.py

@@ -1,33 +1,42 @@
 import json
 import logging
+import os
 
 from django.apps import AppConfig
-
-from ptools.settings import BASE_DIR
+from django.db.models.signals import post_migrate
 
 logger = logging.getLogger('ptools')
 
 
+def app_ready_handler(sender, **kwargs):
+    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('正在初始化站点规则信息表')
+            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))
+
+
 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))
+        # 环境变量不存在，说明数据库还未初始化，先跳过初始化站点数据
+        if os.getenv('CONTAINER_ALREADY_STARTED'):
+            post_migrate.connect(app_ready_handler, sender=self)
+        else:
+            logger.info('第一次启动容器，初始化数据库中')