"""Collect one real sector into an isolated PostgreSQL snapshot store.

Run with the server's uv environment from zhixing-server. No production
publication is created; facts must be read back from this store for validation.
"""
from datetime import date
from pathlib import Path
from urllib.parse import urlsplit, urlunsplit
import json
import time
import tushare as ts
from zhixing_server.bootstrap.config import Settings
from zhixing_server.modules.sector_radar.domain.source import build_source_snapshot
from zhixing_server.modules.sector_radar.infrastructure.postgres import PostgresSectorRadarRepository


def main():
    """Persist provider responses before inspecting their values; redact errors."""
    settings = Settings(_env_file='../.env')
    url = urlsplit(settings.database_url)
    host = url.netloc.rsplit('@', 1)[0] + '@127.0.0.1:5433'
    database = urlunsplit((url.scheme, host, '/radar_detail_selftest_0906', url.query, ''))
    repository = PostgresSectorRadarRepository(database, max_connections=2)
    client = ts.pro_api(settings.tushare_token)
    target = date(2026, 9, 4)
    counts = []

    def collect(api, fields, **params):
        """Store each raw response atomically and return persisted rows."""
        time.sleep(0.25)
        frame = client.query(api, fields=fields, **params)
        snapshot = build_source_snapshot(api_name=api, params=params,
            rows=frame.to_dict('records'), target_trade_date=target)
        repository.save_source_snapshots((snapshot,))
        counts.append({'api':api, 'rows':snapshot.row_count, 'snapshot':snapshot.snapshot_id})
        return snapshot.rows

    try:
        collect('dc_index','ts_code,trade_date,name,idx_type,level,pct_change,leading_code',
            trade_date='20260904', ts_code='BK1147.DC')
        members = collect('dc_member','trade_date,ts_code,con_code,name',
            trade_date='20260904', ts_code='BK1147.DC')
        collect('stock_basic','ts_code,symbol,name,market,exchange,list_status,list_date,delist_date',list_status='L')
        collect('suspend_d','ts_code,trade_date,suspend_timing,suspend_type',trade_date='20260904')
        collect('trade_cal','exchange,cal_date,is_open,pretrade_date',exchange='SSE',start_date='20260720',end_date='20260904')
        for member in members:
            code = member['con_code']
            collect('daily','ts_code,trade_date,close,pre_close,pct_chg,vol,amount',trade_date='20260904',ts_code=code)
            collect('moneyflow_dc','trade_date,ts_code,name,net_amount,net_amount_rate,pct_change,close',trade_date='20260904',ts_code=code)
            collect('moneyflow','trade_date,ts_code,net_mf_amount',trade_date='20260904',ts_code=code)
        result={'status':'collected','sector':'BK1147.DC','trade_date':str(target),'members':len(members),'snapshots':counts}
    except Exception as exc:
        result={'status':'partial','snapshots':counts,'error_type':type(exc).__name__,
            'message':str(exc).replace(settings.tushare_token,'[redacted]')[:300]}
    finally:
        repository.close()
    Path('../.trellis/tasks/09-06-capital-radar-daily-detail/research/collection-result.json').write_text(json.dumps(result,ensure_ascii=False,indent=2))
    print(json.dumps({k:v for k,v in result.items() if k!='snapshots'},ensure_ascii=False))
    print('Stored snapshots:',len(counts))

if __name__ == '__main__':
    main()