init
This commit is contained in:
101
init_data.py
Normal file
101
init_data.py
Normal file
@@ -0,0 +1,101 @@
|
||||
#!/usr/bin/env python
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
初始化数据脚本 - 插入解析接口配置
|
||||
"""
|
||||
|
||||
import os
|
||||
os.environ['SKIP_SCHEDULER'] = 'true'
|
||||
|
||||
from app import create_app
|
||||
from models import db, ParserAPI, HealthCheckConfig
|
||||
|
||||
def init_parser_apis():
|
||||
"""初始化解析接口配置"""
|
||||
app = create_app()
|
||||
|
||||
with app.app_context():
|
||||
# 抖音解析接口
|
||||
douyin_api = ParserAPI(
|
||||
name='抖音自部署API',
|
||||
platform='douyin',
|
||||
api_url='https://dy.zoomcaronline.net',
|
||||
weight=1,
|
||||
is_enabled=True
|
||||
)
|
||||
|
||||
# TikTok解析接口
|
||||
tiktok_api = ParserAPI(
|
||||
name='TikTok自部署API',
|
||||
platform='tiktok',
|
||||
api_url='https://dy.zoomcaronline.net',
|
||||
weight=1,
|
||||
is_enabled=True
|
||||
)
|
||||
|
||||
# 哔哩哔哩解析接口 - 米人API
|
||||
bilibili_mir = ParserAPI(
|
||||
name='哔哩哔哩-米人API',
|
||||
platform='bilibili',
|
||||
api_url='https://api.mir6.com',
|
||||
weight=1,
|
||||
is_enabled=True
|
||||
)
|
||||
|
||||
# 哔哩哔哩解析接口 - BugPK API
|
||||
bilibili_bugpk = ParserAPI(
|
||||
name='哔哩哔哩-BugPK API',
|
||||
platform='bilibili',
|
||||
api_url='https://api.bugpk.com',
|
||||
weight=1,
|
||||
is_enabled=True
|
||||
)
|
||||
|
||||
# 哔哩哔哩解析接口 - 妖狐API
|
||||
bilibili_yaohu = ParserAPI(
|
||||
name='哔哩哔哩-妖狐API',
|
||||
platform='bilibili',
|
||||
api_url='https://api.yaohud.cn',
|
||||
api_key='SM227DLC0ZgJ6DXJhAx',
|
||||
weight=1,
|
||||
is_enabled=True
|
||||
)
|
||||
|
||||
# 检查是否已存在
|
||||
if not ParserAPI.query.filter_by(name='抖音自部署API').first():
|
||||
db.session.add(douyin_api)
|
||||
print("✓ 已添加抖音解析接口")
|
||||
|
||||
if not ParserAPI.query.filter_by(name='TikTok自部署API').first():
|
||||
db.session.add(tiktok_api)
|
||||
print("✓ 已添加TikTok解析接口")
|
||||
|
||||
if not ParserAPI.query.filter_by(name='哔哩哔哩-米人API').first():
|
||||
db.session.add(bilibili_mir)
|
||||
print("✓ 已添加哔哩哔哩-米人API")
|
||||
|
||||
if not ParserAPI.query.filter_by(name='哔哩哔哩-BugPK API').first():
|
||||
db.session.add(bilibili_bugpk)
|
||||
print("✓ 已添加哔哩哔哩-BugPK API")
|
||||
|
||||
if not ParserAPI.query.filter_by(name='哔哩哔哩-妖狐API').first():
|
||||
db.session.add(bilibili_yaohu)
|
||||
print("✓ 已添加哔哩哔哩-妖狐API")
|
||||
|
||||
# 添加健康检查配置
|
||||
if not HealthCheckConfig.query.filter_by(platform='bilibili').first():
|
||||
health_config = HealthCheckConfig(
|
||||
platform='bilibili',
|
||||
test_url='https://www.bilibili.com/video/BV1vrU6B4ELQ/',
|
||||
check_interval=300, # 5分钟
|
||||
is_enabled=True,
|
||||
alert_email='admin@example.com'
|
||||
)
|
||||
db.session.add(health_config)
|
||||
print("✓ 已添加哔哩哔哩健康检查配置")
|
||||
|
||||
db.session.commit()
|
||||
print("\n数据初始化完成!")
|
||||
|
||||
if __name__ == '__main__':
|
||||
init_parser_apis()
|
||||
Reference in New Issue
Block a user