102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
#!/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()
|