bug修复

Administrator

2024-06-11 870c02bbc2302171afe54bebc29fa9a0ae597256

bug修复

 huaxin_client/l1_client.py

@@ -200,6 +200,7 @@
    codes_sz = []
    for i in range(15):
        try:
            logger_local_huaxin_l1.info("开始获取目标代码")
            codes_sh, codes_sz = l1_subscript_codes_manager.get_codes()
            logger_local_huaxin_l1.info(f"获取上证，深证代码数量：sh-{len(codes_sh)} sz-{len(codes_sz)}")
            break

 huaxin_client/l1_subscript_codes_manager.py

@@ -75,25 +75,25 @@
    codes_sh, codes_sz = [], []
    if os.path.exists(__CODE_SH_PATH):
        with open(__CODE_SH_PATH, 'r') as f:
            line = f.readline()
            while line:
                if line.strip():
                    code = line.strip()
                    if not tool.is_can_buy_code(code):
                        continue
                    codes_sh.append(code.encode('utf-8'))
                line = f.readline()
            lines = f.readlines()
            for line in lines:
                if line:
                    if line.strip():
                        code = line.strip()
                        if not tool.is_can_buy_code(code):
                            continue
                        codes_sh.append(code.encode('utf-8'))

    if os.path.exists(__CODE_SZ_PATH):
        with open(__CODE_SZ_PATH, 'r') as f:
            line = f.readline()
            while line:
                if line.strip():
                    code = line.strip()
                    if not tool.is_can_buy_code(code):
                        continue
                    codes_sz.append(code.encode('utf-8'))
                line = f.readline()
            lines = f.readlines()
            for line in lines:
                if line:
                    if line.strip():
                        code = line.strip()
                        if not tool.is_can_buy_code(code):
                            continue
                        codes_sz.append(code.encode('utf-8'))
    return codes_sh, codes_sz


@@ -105,4 +105,4 @@


if __name__ == '__main__':
    print( request_l1_subscript_target_codes())
    print( get_codes())

 test/test.py

@@ -21,6 +21,5 @@


if __name__ == "__main__":
    CodesNameManager.add_code_name("000333", "美的集团")
    CodesNameManager.add_code_name("000333", "美的集团")
    print(CodesNameManager.list_code_name_dict())
    codes_sh, codes_sz = l1_subscript_codes_manager.get_codes()
    print(len(codes_sh), len(codes_sz))