#!/usr/bin/env python3
"""一次性迁移脚本：stock_sector_map.json → stock_sectors 表

运行方式:
    python3 migrate_sectors.py
"""
from mofin_db import get_conn, init_all_tables, migrate_stock_sectors

conn = get_conn()
init_all_tables(conn)
stocks, mappings = migrate_stock_sectors(conn)
print(f"迁移完成: {stocks} 只股票, {mappings} 条板块映射")

# 验证
rows = conn.execute("SELECT COUNT(*) FROM stock_sectors").fetchone()[0]
print(f"stock_sectors 表总行数: {rows}")

# 样例
for r in conn.execute("SELECT * FROM stock_sectors LIMIT 5").fetchall():
    print(f"  {r[0]} → {r[1]}")

conn.close()