From d0fd153d07f9bd41cf0a30ddce30402eaef83da3 Mon Sep 17 00:00:00 2001 From: Administrator <admin@example.com> Date: 星期二, 14 十一月 2023 09:26:46 +0800 Subject: [PATCH] A类/B类一键切换 --- utils/data_export_util.py | 26 ++++++++++++++++---------- 1 files changed, 16 insertions(+), 10 deletions(-) diff --git a/utils/data_export_util.py b/utils/data_export_util.py index cd55100..75578cf 100644 --- a/utils/data_export_util.py +++ b/utils/data_export_util.py @@ -24,7 +24,8 @@ # 鑾峰彇L2澶勭悊浣嶇疆淇℃伅 process_indexs = log_export.get_l2_process_position(code, date) trade_indexs = log_export.get_l2_trade_position(code, date) - fdatas = export_l2_data(code, datas, process_indexs, trade_indexs) + real_position_indexes = log_export.get_real_place_order_positions(code, date) + fdatas = export_l2_data(code, datas, process_indexs, trade_indexs, real_position_indexes) __save_l2_datas(code, fdatas) @@ -34,11 +35,12 @@ datas = local_today_datas[code] process_indexs = log_export.get_l2_process_position(code, date) trade_indexs = log_export.get_l2_trade_position(code, date) - fdatas = export_l2_data(code, datas, process_indexs, trade_indexs) + real_position_indexes = log_export.get_real_place_order_positions(code, date) + fdatas = export_l2_data(code, datas, process_indexs, trade_indexs, real_position_indexes) return fdatas -def export_l2_data(code, datas, process_indexs, trade_indexs): +def export_l2_data(code, datas, process_indexs, trade_indexs, real_position_indexes): def find_process_index(index): for i in range(0, len(process_indexs)): if process_indexs[i][0] <= index <= process_indexs[i][1]: @@ -70,6 +72,10 @@ for data in datas: index += 1 trade_info = find_trade_index(data["index"]) + if not trade_info: + # 鑾峰彇鐪熷疄涓嬪崟浣嶇疆 + if data["index"] in real_position_indexes: + trade_info = [3] style_int = None if find_process_index(data["index"]) % 2 == 0: style_int = 0 @@ -108,31 +114,31 @@ limit_price = "" if int(data["val"]["limitPrice"]) == 1: - limit_price = "娑ㄥ仠" + limit_price = "T" elif int(data["val"]["limitPrice"]) == 2: - limit_price = "璺屽仠" + limit_price = "D" operateDesc = "" if int(data["val"]["operateType"]) == 0: if len(limit_price) > 0: - operateDesc = '涔� ({})'.format(limit_price) + operateDesc = '涔皗}'.format(limit_price) else: operateDesc = '涔�' elif int(data["val"]["operateType"]) == 1: if len(limit_price) > 0: - operateDesc = '涔版挙 ({})'.format(limit_price) + operateDesc = '涔版挙{}'.format(limit_price) else: operateDesc = '涔版挙' elif int(data["val"]["operateType"]) == 2: if len(limit_price) > 0: - operateDesc = '鍗� ({})'.format(limit_price) + operateDesc = '鍗杮}'.format(limit_price) else: operateDesc = '鍗�' elif int(data["val"]["operateType"]) == 3: if len(limit_price) > 0: - operateDesc = '鍗栨挙 ({})'.format(limit_price) + operateDesc = '鍗栨挙{}'.format(limit_price) else: operateDesc = '鍗栨挙' format_data.append(operateDesc) @@ -265,6 +271,6 @@ if __name__ == "__main__": try: - export_l2_excel("000981") + export_l2_excel("605218") except Exception as e: logging.exception(e) -- Gitblit v1.8.0