#!/usr/bin/env python3
"""
将“缺省全局配置项”同步到数据库 global_strategy_config 表。

- 已在 UI 保存过的项不会覆盖（只插入缺失的 key）。
- 用于新上线配置项（如 MAX_RSI_FOR_LONG、MIN_RSI_FOR_SHORT 等）一次性写入默认值，
  便于在数据库中可见、可备份，且不依赖“先在页面改一次再保存”。

使用方式（在项目根目录）：
  cd backend && python sync_global_config_defaults.py
  或
  python backend/sync_global_config_defaults.py
"""
import os
import sys
from pathlib import Path

# 确保 backend 在路径中
backend_dir = Path(__file__).resolve().parent
if str(backend_dir) not in sys.path:
    sys.path.insert(0, str(backend_dir))

# 需要同步的缺省项（仅插入数据库中不存在的 key）
DEFAULTS_TO_SYNC = [
    {"config_key": "MAX_RSI_FOR_LONG", "config_value": "70", "config_type": "number", "category": "strategy",
     "description": "做多时 RSI 超过此值则不开多（避免超买区追多）。2026-01-31新增。"},
    {"config_key": "MAX_CHANGE_PERCENT_FOR_LONG", "config_value": "25", "config_type": "number", "category": "strategy",
     "description": "做多时 24h 涨跌幅超过此值则不开多（避免追大涨）。单位：百分比数值，如 25 表示 25%。2026-01-31新增。"},
    {"config_key": "MIN_RSI_FOR_SHORT", "config_value": "30", "config_type": "number", "category": "strategy",
     "description": "做空时 RSI 低于此值则不做空（避免深超卖反弹）。2026-01-31新增。"},
    {"config_key": "MAX_CHANGE_PERCENT_FOR_SHORT", "config_value": "10", "config_type": "number", "category": "strategy",
     "description": "做空时 24h 涨跌幅超过此值则不做空（24h 仍大涨时不做空）。单位：百分比数值。2026-01-31新增。"},
    {"config_key": "TAKE_PROFIT_1_PERCENT", "config_value": "0.15", "config_type": "number", "category": "strategy",
     "description": "分步止盈第一目标（保证金百分比，如 0.15=15%）。第一目标触发后了结50%仓位，剩余追求第二目标。"},
    {"config_key": "SCAN_EXTRA_SYMBOLS_FOR_SUPPLEMENT", "config_value": "8", "config_type": "number", "category": "scan",
     "description": "智能补单：多返回的候选数量。当前 TOP_N 中部分因冷却等被跳过时，仍会尝试这批额外候选，避免无单可下。"},
    {"config_key": "BETA_FILTER_ENABLED", "config_value": "true", "config_type": "boolean", "category": "strategy",
     "description": "大盘共振过滤：BTC/ETH 下跌时屏蔽多单。"},
    {"config_key": "BETA_FILTER_THRESHOLD", "config_value": "-0.005", "config_type": "number", "category": "strategy",
     "description": "大盘共振阈值（比例，如 -0.005 表示 -0.5%）。"},
]


def main():
    try:
        from database.models import GlobalStrategyConfig
        from database.connection import db
    except ImportError as e:
        print(f"无法导入数据库模块，请确保在 backend 目录或设置 PYTHONPATH: {e}")
        sys.exit(1)

    def _table_has_column(table: str, col: str) -> bool:
        try:
            db.execute_one(f"SELECT {col} FROM {table} LIMIT 1")
            return True
        except Exception:
            return False

    if not _table_has_column("global_strategy_config", "config_key"):
        print("表 global_strategy_config 不存在或结构异常，请先执行 backend/database/add_global_strategy_config.sql")
        sys.exit(1)

    inserted = 0
    skipped = 0
    for row in DEFAULTS_TO_SYNC:
        key = row["config_key"]
        existing = GlobalStrategyConfig.get(key)
        if existing:
            skipped += 1
            print(f"  已有: {key}")
            continue
        GlobalStrategyConfig.set(
            key,
            row["config_value"],
            row["config_type"],
            row["category"],
            row.get("description"),
            updated_by="sync_global_config_defaults",
        )
        inserted += 1
        print(f"  插入: {key} = {row['config_value']}")

    print(f"\n同步完成: 新增 {inserted} 项，已存在跳过 {skipped} 项。")


if __name__ == "__main__":
    main()