import json
import logging
import os

from django.apps import AppConfig
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):
        # 环境变量不存在，说明数据库还未初始化，先跳过初始化站点数据
        if os.getenv('CONTAINER_ALREADY_STARTED'):
            post_migrate.connect(app_ready_handler, sender=self)
        else:
            logger.info('第一次启动容器，初始化数据库中')