#!/usr/bin/env python3
"""自动替换 SDK 调用为内部接口调用"""

replacements = [
    # _base_data 调用
    ("self._base_data.get_code_info(", "self._internal.base.get_code_info("),
    ("self._base_data.get_code_list(", "self._internal.base.get_code_list("),
    ("self._base_data.get_future_code_list(", "self._internal.base.get_future_code_list("),
    ("self._base_data.get_calendar(", "self._internal.base.get_calendar("),
    ("self._base_data.get_adj_factor(", "self._internal.base.get_adj_factor("),
    ("self._base_data.get_backward_factor(", "self._internal.base.get_backward_factor("),
    ("self._base_data.get_etf_pcf(", "self._internal.base.get_etf_pcf("),
    ("self._base_data.get_hist_code_list(", "self._internal.base.get_hist_code_list("),
    
    # _market_data 调用
    ("self._market_data.query_kline(", "self._internal.market.query_kline("),
    ("self._market_data.query_snapshot(", "self._internal.market.query_snapshot("),
    
    # _info_data 调用
    ("self._info_data.get_equity_structure(", "self._internal.info.get_equity_structure("),
    ("self._info_data.get_share_holder(", "self._internal.info.get_share_holder("),
    ("self._info_data.get_holder_num(", "self._internal.info.get_holder_num("),
    ("self._info_data.get_income(", "self._internal.info.get_income("),
    ("self._info_data.get_balance_sheet(", "self._internal.info.get_balance_sheet("),
    ("self._info_data.get_cash_flow(", "self._internal.info.get_cash_flow("),
    ("self._info_data.get_profit_express(", "self._internal.info.get_profit_express("),
    ("self._info_data.get_profit_notice(", "self._internal.info.get_profit_notice("),
    ("self._info_data.get_margin_summary(", "self._internal.info.get_margin_summary("),
    ("self._info_data.get_margin_detail(", "self._internal.info.get_margin_detail("),
    ("self._info_data.get_long_hu_bang(", "self._internal.info.get_long_hu_bang("),
    ("self._info_data.get_block_trading(", "self._internal.info.get_block_trading("),
    ("self._info_data.get_index_constituent(", "self._internal.info.get_index_constituent("),
    ("self._info_data.get_index_weight(", "self._internal.info.get_index_weight("),
    ("self._info_data.get_fund_share(", "self._internal.info.get_fund_share("),
    ("self._info_data.get_kzz_issuance(", "self._internal.info.get_kzz_issuance("),
]

def main():
    with open('app/adapters/amazingdata_adapter.py', 'r', encoding='utf-8') as f:
        content = f.read()
    
    count = 0
    for old, new in replacements:
        if old in content:
            occurrences = content.count(old)
            content = content.replace(old, new)
            count += occurrences
            print(f"Replaced {occurrences} occurrence(s): {old} -> {new}")
    
    with open('app/adapters/amazingdata_adapter.py', 'w', encoding='utf-8') as f:
        f.write(content)
    
    print(f"\nTotal replacements: {count}")

if __name__ == "__main__":
    main()