#!/usr/bin/env python3
"""
import_holding_xls.py — 从holding.xls导入持仓到SQLite + portfolio.json

用法：python3 import_holding_xls.py [--cash 现金金额]
"""
import csv, json, sys, subprocess, sqlite3
from datetime import datetime

STOCKS_FILE = "/home/hmo/stocks/holding.xls"
PORTFOLIO_PATH = "/home/hmo/web-dashboard/data/portfolio.json"
DB_PATH = "/home/hmo/web-dashboard/data/mofin.db"
CASH = 80476  # default cash


def clean_cell(v):
    v = v.strip()
    if v.startswith('="') and v.endswith('"'):
        v = v[2:-1]
    elif v.startswith('='):
        v = v[1:]
    return v.strip()


def main():
    # Parse args
    global CASH
    for i, a in enumerate(sys.argv[1:]):
        if a == '--cash' and i + 2 < len(sys.argv):
            CASH = float(sys.argv[i + 2])

    with open(STOCKS_FILE, 'r', encoding='gbk') as f:
        reader = csv.reader(f, delimiter='\t')
        rows = list(reader)

    print(f"读取 {STOCKS_FILE}: {len(rows)-1} 条记录")

    holdings = []
    total_mv_cny = 0

    for r in rows[1:]:
        code = clean_cell(r[0])
        name = r[1].strip()
        shares = int(clean_cell(r[2]))
        price_raw = r[4].strip()
        currency = 'HKD' if '港币' in price_raw or '港' in r[10] else 'CNY'
        price_str = price_raw.replace('港币', '').replace('港元', '').replace('港', '').strip()
        price = float(price_str)
        cost_price = float(clean_cell(r[5]))
        pl = float(clean_cell(r[6]))
        pl_pct = float(clean_cell(r[7])) if r[7].strip() else 0.0
        mkt_val = float(clean_cell(r[11]))
        cost_amount = float(clean_cell(r[15])) if r[15].strip() and r[15].strip() != '--' else 0
        rate_str = clean_cell(r[16])
        rate = float(rate_str) if rate_str and rate_str != '--' else 0.866

        mv_cny = mkt_val if currency == 'CNY' else mkt_val * rate
        total_mv_cny += mv_cny

        holdings.append({
            'code': code, 'name': name, 'shares': shares,
            'price': price, 'cost_price': round(cost_price, 2),
            'currency': currency, 'market_val': mkt_val,
            'cost_amount': cost_amount, 'exchange_rate': rate,
        })

        pfx = 'HK$' if currency == 'HKD' else '￥'
        print(f"  {code} {name} {shares}股 {pfx}{price:.2f} 成本{cost_price:.2f} 盈亏{pl:+,.0f}({pl_pct:+.1f}%)")

    total_assets = total_mv_cny + CASH
    position_pct = round(total_mv_cny / total_assets * 100, 2) if total_assets > 0 else 0

    # Write portfolio.json
    portfolio = {
        'holdings': holdings,
        'cash': CASH,
        'total_market_value': round(total_mv_cny, 2),
        'total_assets': round(total_assets, 2),
        'total_pl': 0,
        'position_pct': position_pct,
        'updated_at': datetime.now().strftime('%Y-%m-%d %H:%M'),
        'source': STOCKS_FILE,
    }
    with open(PORTFOLIO_PATH, 'w') as f:
        json.dump(portfolio, f, indent=2, ensure_ascii=False)
    print(f"  → {PORTFOLIO_PATH}")

    # Write SQLite
    conn = sqlite3.connect(DB_PATH)
    c = conn.cursor()
    c.execute('DELETE FROM holdings')
    c.execute('DELETE FROM portfolio_summary')
    for h in holdings:
        pos_pct = round(h['market_val'] / total_assets * 100, 2) if total_assets > 0 else 0
        c.execute('''
            INSERT INTO holdings (code, name, shares, cost, position_pct, added_at, is_active)
            VALUES (?, ?, ?, ?, ?, ?, 1)
        ''', (h['code'], h['name'], h['shares'], h['cost_price'], pos_pct,
              datetime.now().strftime('%Y-%m-%d')))
    c.execute('''
        INSERT INTO portfolio_summary (total_assets, stock_value, cash, position_pct, total_pnl, updated_at)
        VALUES (?, ?, ?, ?, ?, ?)
    ''', (round(total_assets, 2), round(total_mv_cny, 2), CASH, position_pct, 0,
          datetime.now().strftime('%Y-%m-%d %H:%M')))
    conn.commit()
    conn.close()
    print(f"  → SQLite 已更新")

    print(f"\n统计: {len(holdings)}只持仓 | "
          f"总资产{round(total_assets):,.0f}元 | "
          f"现金{CASH:,.0f}元 | "
          f"仓位{position_pct}%")

    # Trigger full reassessment
    print("\n→ 触发 per_stock_reassess 全量重评...")
    r = subprocess.run(
        ["python3", "/home/hmo/.hermes/profiles/position-analyst/scripts/per_stock_reassess.py"],
        capture_output=True, text=True, timeout=120
    )
    print(r.stdout[-500:] if len(r.stdout) > 500 else r.stdout)

    # Rebuild decision trees
    print("\n→ 重建决策树...")
    sys.path.insert(0, '/home/hmo/web-dashboard')
    from strategy_tree import init_default_branches
    with open('/home/hmo/web-dashboard/data/decisions.json') as f:
        data = json.load(f)
    ok = 0
    for e in data.get('decisions', []):
        branches = init_default_branches(
            e.get('code', ''), e.get('name', ''),
            e.get('entry_low', 0), e.get('entry_high', 0),
            e.get('stop_loss', 0), e.get('take_profit', 0))
        e['strategy_tree'] = {'branches': branches, 'created_at': datetime.now().strftime('%Y-%m-%d')}
        ok += 1
    with open('/home/hmo/web-dashboard/data/decisions.json', 'w') as f:
        json.dump(data, f, indent=2, ensure_ascii=False)
    print(f"决策树重建: {ok}/{len(data.get('decisions',[]))}")


if __name__ == '__main__':
    main()