# 1. 从数据库获取实例、策略、交易对信息
# 2. 验证实例状态是否为 RUNNING
# 3. 读取 .env 中的 OKX API 配置
# 4. 创建 OKX API 客户端

# 1. 调用 OKX API 获取最新 200 根K线
# 2. 解析时间戳、OHLCV 数据
# 3. 转换为 DataFrame 格式
# 4. 计算技术指标（MA/RSI/MACD/BOLL等）

# 1. 解析策略配置（JSON格式的规则集）
# 2. 调用 should_buy() 判断买入信号
# 3. 调用 should_sell() 判断卖出信号

# 1. 查询历史交易记录
# 2. 计算当前净持仓（BUY 增加，SELL 减少）
# 3. 决定是否需要交易

# 交易规则：
# - 如果有买入信号 且 当前无持仓 → 市价买入 1 单位
# - 如果有卖出信号 且 当前有持仓 → 市价卖出全部
# 
# 交易记录包含：
# - 时间戳、方向（BUY/SELL）、价格、数量
# - OKX 订单ID、订单状态
# - 完整的 OKX 响应 JSON（用于调试）

# 调用 OKX API 获取账户总权益
# 保存到 account_equity_snapshots 表
# 用于后续绘制权益曲线

# 启动实例时
start_strategy_instance(instance_id, interval_sec)
    ↓
# 注册到 APScheduler 调度器
scheduler.add_job(
    func=_run_strategy_instance,
    trigger="interval",
    seconds=interval_sec,  # 例如：60秒
    id=f"strategy-{instance_id}"
)
    ↓
# 调度器每隔 60 秒自动执行一次策略函数

# 获取交易摘要
GET /instances/{instance_id}/summary

# 获取所有交易记录
GET /instances/{instance_id}/trades

# 运行 Python 脚本查询
from app.db.session import SessionLocal
from app.models import Symbol

db = SessionLocal()
symbols = db.query(Symbol).all()
for s in symbols:
    print(f"ID: {s.id}, 品种: {s.inst_id}")
db.close()

from app.db.session import SessionLocal
from app.models import Symbol

db = SessionLocal()

# 添加 BTC 永续合约
btc_swap = Symbol(
    exchange_name="OKX",
    inst_id="BTC-USDT-SWAP",
    base_ccy="BTC",
    quote_ccy="USDT",
    inst_type="SWAP",
    is_active=True
)
db.add(btc_swap)
db.commit()
print(f"BTC-USDT-SWAP 的 Symbol ID: {btc_swap.id}")

# 添加 ETH 永续合约
eth_swap = Symbol(
    exchange_name="OKX",
    inst_id="ETH-USDT-SWAP",
    base_ccy="ETH",
    quote_ccy="USDT",
    inst_type="SWAP"
)
db.add(eth_swap)
db.commit()
print(f"ETH-USDT-SWAP 的 Symbol ID: {eth_swap.id}")

db.close()

from app.db.session import SessionLocal
from app.models import ExchangeAccount, User

db = SessionLocal()

# 1. 首先确保有用户（如果没有，先创建）
user = db.query(User).filter(User.id == 1).first()
if not user:
    from passlib.hash import bcrypt
    user = User(
        username="admin",
        password_hash=bcrypt.hash("admin123")
    )
    db.add(user)
    db.commit()

# 2. 创建交易账户
account = ExchangeAccount(
    user_id=1,
    exchange_name="OKX",
    api_key="你的OKX API Key",
    api_secret="你的OKX API Secret",
    passphrase="你的OKX API Passphrase",
    is_active=True
)
db.add(account)
db.commit()
print(f"交易账户 ID: {account.id}")

db.close()

from app.db.session import SessionLocal
from app.models import ExchangeAccount

db = SessionLocal()
accounts = db.query(ExchangeAccount).all()
for acc in accounts:
    print(f"ID: {acc.id}, 用户: {acc.user_id}, 交易所: {acc.exchange_name}")
db.close()

from app.db.session import SessionLocal
from app.models import Symbol, ExchangeAccount, User
from passlib.hash import bcrypt

def setup():
    db = SessionLocal()
    
    # 1. 创建用户
    user = db.query(User).filter(User.username == "admin").first()
    if not user:
        user = User(
            username="admin",
            password_hash=bcrypt.hash("admin123")
        )
        db.add(user)
        db.commit()
        print(f"✅ 创建用户: admin (ID: {user.id})")
    
    # 2. 创建交易品种
    symbols = [
        {"inst_id": "BTC-USDT-SWAP", "base": "BTC", "quote": "USDT"},
        {"inst_id": "ETH-USDT-SWAP", "base": "ETH", "quote": "USDT"},
    ]
    
    for s in symbols:
        existing = db.query(Symbol).filter(Symbol.inst_id == s["inst_id"]).first()
        if not existing:
            symbol = Symbol(
                exchange_name="OKX",
                inst_id=s["inst_id"],
                base_ccy=s["base"],
                quote_ccy=s["quote"],
                inst_type="SWAP"
            )
            db.add(symbol)
            db.commit()
            print(f"✅ 创建品种: {symbol.inst_id} (Symbol ID: {symbol.id})")
    
    # 3. 创建交易账户
    account = db.query(ExchangeAccount).filter(ExchangeAccount.user_id == user.id).first()
    if not account:
        account = ExchangeAccount(
            user_id=user.id,
            exchange_name="OKX",
            api_key="YOUR_API_KEY",  # 替换为真实的
            api_secret="YOUR_API_SECRET",
            passphrase="YOUR_PASSPHRASE",
            is_active=True
        )
        db.add(account)
        db.commit()
        print(f"✅ 创建交易账户 (账户 ID: {account.id})")
    
    # 4. 显示汇总信息
    print("\n" + "=" * 60)
    print("设置完成！以下是可用的 ID：")
    print("=" * 60)
    
    symbols = db.query(Symbol).all()
    print("\n📊 Symbol ID（创建策略时使用）：")
    for s in symbols:
        print(f"  - Symbol ID: {s.id} -> {s.inst_id}")
    
    accounts = db.query(ExchangeAccount).all()
    print("\n🔑 账户 ID（实盘交易时使用）：")
    for acc in accounts:
        print(f"  - 账户 ID: {acc.id} -> {acc.exchange_name}")
    
    db.close()

if __name__ == "__main__":
    setup()

import requests
from datetime import datetime, timedelta

BASE_URL = "http://127.0.0.1:8000"

# 同步 BTC 1小时K线（最近7天）
payload = {
    "inst_id": "BTC-USDT-SWAP",
    "timeframe": "1H",
    "start_ts": (datetime.now() - timedelta(days=7)).isoformat(),
    "end_ts": datetime.now().isoformat(),
    "limit_per_call": 100
}

response = requests.post(f"{BASE_URL}/market/klines/sync", json=payload)
print(response.json())
# 输出: {"inserted": 168}  # 7天 * 24小时

import requests
from datetime import datetime, timedelta

BASE_URL = "http://127.0.0.1:8000"

payload = {
    "strategy_id": 5,  # ← 你的策略 ID
    "start_ts": (datetime.now() - timedelta(days=7)).isoformat(),
    "end_ts": datetime.now().isoformat(),
    "initial_balance": 10000.0
}

response = requests.post(f"{BASE_URL}/backtests/", json=payload)
result = response.json()

print(f"回测ID: {result['id']}")
print(f"状态: {result['status']}")
print(f"结果: {result['result_json']}")

import requests

BASE_URL = "http://127.0.0.1:8000"

# 创建实盘实例
response = requests.post(
    f"{BASE_URL}/instances/",
    params={
        "strategy_id": 5,  # ← 策略 ID
        "exchange_account_id": 1  # ← 账户 ID
    }
)

instance = response.json()
print(f"实盘实例 ID: {instance['id']}")
print(f"状态: {instance['status']}")

# 启动实盘交易
instance_id = instance['id']
response = requests.post(f"{BASE_URL}/instances/{instance_id}/start")
print("✅ 实盘交易已启动！")

# 停止实盘交易
# response = requests.post(f"{BASE_URL}/instances/{instance_id}/stop")

from app.db.session import SessionLocal
from app.models import Symbol

db = SessionLocal()
symbols = db.query(Symbol).all()
for s in symbols:
    print(f"Symbol ID: {s.id} = {s.inst_id}")
db.close()

from app.db.session import SessionLocal
from app.models import ExchangeAccount

db = SessionLocal()
accounts = db.query(ExchangeAccount).all()
for acc in accounts:
    print(f"账户 ID: {acc.id}")
db.close()

"""
快速开始：一键设置环境并创建示例策略
"""
from app.db.session import SessionLocal
from app.models import Symbol, ExchangeAccount, User
from passlib.hash import bcrypt
import requests
import json

BASE_URL = "http://127.0.0.1:8000"

def quick_start():
    db = SessionLocal()
    
    print("🚀 开始快速设置...\n")
    
    # 1. 创建用户
    user = db.query(User).first()
    if not user:
        user = User(username="admin", password_hash=bcrypt.hash("admin123"))
        db.add(user)
        db.commit()
    print(f"✅ 用户 ID: {user.id}")
    
    # 2. 创建BTC品种
    btc = db.query(Symbol).filter(Symbol.inst_id == "BTC-USDT-SWAP").first()
    if not btc:
        btc = Symbol(
            exchange_name="OKX",
            inst_id="BTC-USDT-SWAP",
            base_ccy="BTC",
            quote_ccy="USDT",
            inst_type="SWAP"
        )
        db.add(btc)
        db.commit()
    print(f"✅ BTC Symbol ID: {btc.id}")
    
    # 3. 创建账户（请修改为真实API密钥）
    account = db.query(ExchangeAccount).filter(ExchangeAccount.user_id == user.id).first()
    if not account:
        account = ExchangeAccount(
            user_id=user.id,
            exchange_name="OKX",
            api_key="YOUR_API_KEY",
            api_secret="YOUR_SECRET",
            passphrase="YOUR_PASSPHRASE"
        )
        db.add(account)
        db.commit()
    print(f"✅ 账户 ID: {account.id}")
    
    db.close()
    
    # 4. 创建示例策略
    strategy_config = {
        "buy_groups": [{
            "logic": "AND",
            "conditions": [
                {"side": "BUY", "indicator_type": "MACD", "signal_type": "MACD_GOLDEN_CROSS", "params": {}},
                {"side": "BUY", "indicator_type": "RSI", "signal_type": "RSI_OVERSOLD", "params": {"threshold": 30}}
            ]
        }],
        "sell_groups": [{
            "logic": "OR",
            "conditions": [
                {"side": "SELL", "indicator_type": "MACD", "signal_type": "MACD_DEAD_CROSS", "params": {}},
                {"side": "SELL", "indicator_type": "RSI", "signal_type": "RSI_OVERBOUGHT", "params": {"threshold": 70}}
            ]
        }]
    }
    
    payload = {
        "name": "快速开始示例策略",
        "description": "MACD金叉+RSI超卖买入，MACD死叉或RSI超买卖出",
        "symbol_id": btc.id,
        "timeframe": "1H",
        "leverage": 1.0,
        "monitor_interval_sec": 60,
        "config_json": json.dumps(strategy_config)
    }
    
    try:
        response = requests.post(f"{BASE_URL}/strategies/", json=payload)
        strategy = response.json()
        print(f"✅ 策略 ID: {strategy['id']}")
        
        print("\n" + "="*60)
        print("✅ 设置完成！请记住以下 ID：")
        print("="*60)
        print(f"Symbol ID (BTC): {btc.id}")
        print(f"账户 ID: {account.id}")
        print(f"策略 ID: {strategy['id']}")
        print("\n下一步：")
        print("1. 访问 http://127.0.0.1:5173/strategies 查看策略")
        print("2. 使用可视化构建器创建更多策略")
        print("3. 运行回测验证策略")
        print("4. 配置真实API后启动实盘交易")
        
    except Exception as e:
        print(f"❌ 创建策略失败: {e}")

if __name__ == "__main__":
    quick_start()