#!/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()