mirror of
https://github.com/ngfchl/ptools
synced 2023-07-10 13:41:22 +08:00
49 lines
1.9 KiB
Python
49 lines
1.9 KiB
Python
import json
|
||
import logging
|
||
|
||
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为更新，其他是错误了')
|
||
btschool = Site.objects.filter(url='http://47.242.110.63/').first()
|
||
if btschool:
|
||
bt_school = Site.objects.filter(url='https://pt.btschool.club/').first()
|
||
if bt_school:
|
||
bt_school.delete()
|
||
btschool.url = 'https://pt.btschool.club/'
|
||
btschool.save()
|
||
for site_rules in data:
|
||
if site_rules.get('pk'):
|
||
del site_rules['pk']
|
||
if site_rules.get('id'):
|
||
del site_rules['id']
|
||
url = site_rules.get('url')
|
||
site_obj = Site.objects.update_or_create(defaults=site_rules, url=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.path.exists('CONTAINER_ALREADY_STARTED_PLACEHOLDER'):
|
||
# logger.info('第一次启动容器，初始化数据库中')
|
||
post_migrate.connect(app_ready_handler, sender=self)
|