You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

57 lines
2.9 KiB

#!/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()