""" 接受客户端数据的服务器 """ import decimal import json import logging import random import socketserver import socket import threading import time from utils import alert_util, data_process, global_util, ths_industry_util, tool, import_util, socket_util from code_attribute import code_volumn_manager, global_data_loader, gpcode_manager, first_target_code_data_processor import constant from user import authority from l2 import l2_data_manager_new, l2_data_manager, l2_data_log, l2_log, code_price_manager import l2_data_util from l2.cancel_buy_strategy import HourCancelBigNumComputer, LCancelBigNumComputer import l2.l2_data_util from output import code_info_output from third_data import block_info, kpl_api from third_data.code_plate_key_manager import CodesHisReasonAndBlocksManager from third_data.history_k_data_util import HistoryKDatasUtils from third_data.kpl_data_manager import KPLCodeLimitUpReasonManager from ths import l2_listen_pos_health_manager, l2_code_operate, client_manager from trade import trade_data_manager, trade_manager, l2_trade_util, \ current_price_process_manager, trade_juejin from code_attribute.code_data_util import ZYLTGBUtil import l2.transaction_progress from log_module.log import logger_l2_error, logger_device, logger_trade_delegate, logger_buy_1_volumn_record, \ logger_l2_trade_queue, logger_l2_latest_data, logger_l2_trade_buy_queue, logger_debug from trade.huaxin import huaxin_trade_record_manager from trade.trade_manager import TradeTargetCodeModeManager from trade.trade_queue_manager import THSBuy1VolumnManager, thsl2tradequeuemanager ths_util = import_util.import_lib("ths.ths_util") trade_gui = import_util.import_lib("trade.trade_gui") class MyTCPServer(socketserver.TCPServer): def __init__(self, server_address, RequestHandlerClass, bind_and_activate=True, pipe_trade=None, pipe_ui=None): self.pipe_trade = pipe_trade # 增加的参数 self.pipe_ui = pipe_ui # 初始化数据 block_info.init() socketserver.TCPServer.__init__(self, server_address, RequestHandlerClass, bind_and_activate=bind_and_activate) # 如果使用异步的形式则需要再重写ThreadingTCPServer class MyThreadingTCPServer(socketserver.ThreadingMixIn, MyTCPServer): pass # 首板tick级数据 class MyBaseRequestHandle(socketserver.BaseRequestHandler): l2_data_error_dict = {} last_trade_delegate_data = None buy1_volumn_manager = THSBuy1VolumnManager() ths_l2_trade_queue_manager = thsl2tradequeuemanager() latest_buy1_volumn_dict = {} l2_trade_queue_time_dict = {} l2_save_time_dict = {} l2_trade_buy_queue_dict = {} tradeBuyQueue = l2.transaction_progress.TradeBuyQueue() last_time = {} first_tick_datas = [] latest_oringin_data = {} last_l2_listen_health_time = {} __KPLCodeLimitUpReasonManager = KPLCodeLimitUpReasonManager() __CodesPlateKeysManager = CodesHisReasonAndBlocksManager() # 在L2监控上采集的现价 __l2_current_price_data = {} def setup(self): super().setup() # 可以不调用父类的setup()方法，父类的setup方法什么都没做 # print("----setup方法被执行-----") # print("打印传入的参数：", self.server.pipe_trade) self.l2CodeOperate = l2_code_operate.L2CodeOperate.get_instance() def __notify_trade(self, type_): if self.server.pipe_trade: self.server.pipe_trade.send(json.dumps({"type": type_})) def handle(self): host = self.client_address[0] super().handle() # 可以不调用父类的handler(),方法，父类的handler方法什么都没做 # print("-------handler方法被执行----") # print(self.server) # print(self.request) # 服务 # print("客户端地址：", self.client_address) # 客户端地址 # print(self.__dict__) # print("- " * 30) # print(self.server.__dict__) # print("- " * 30) sk: socket.socket = self.request while True: data = sk.recv(1024 * 100) if len(data) == 0: # print("客户端断开连接") break _str = str(data, encoding="gbk") if len(_str) > 0: # print("结果：",_str) type = -1 try: # 如果带有头 if _str.startswith("##"): total_length = int(_str[2:10]) _str = _str[10:] # 防止socket数据发生粘连 while total_length > len(_str): d = sk.recv(1024 * 100) if d: _str += d.decode(encoding='gbk') type = data_process.parseType(_str) except Exception as e: print("接受到的异常数据：", f"{_str[:10]}...{_str[-10:]}") if str(e).find("Unterminated string starting") > -1: _str = _str.replace("\n", "") type = data_process.parseType(_str) else: print(_str) return_str = "OK" if type == 0: try: origin_start_time = round(time.time() * 1000) __start_time = round(time.time() * 1000) # level2盘口数据 day, client, channel, code, capture_time, process_time, origin_datas, origin_datas_count = l2.l2_data_util.parseL2Data( _str) last_health_time = self.last_l2_listen_health_time.get((client, channel)) # --------------------------------设置L2健康状态-------------------------------- if last_health_time is None or __start_time - last_health_time > 1000: self.last_l2_listen_health_time[(client, channel)] = __start_time # 更新监听位健康状态 if origin_datas_count == 0: l2_listen_pos_health_manager.set_unhealthy(client, channel) else: l2_listen_pos_health_manager.set_healthy(client, channel) l2_log.threadIds[code] = random.randint(0, 100000) if True: # 间隔1s保存一条l2的最后一条数据 if code not in self.l2_save_time_dict or origin_start_time - self.l2_save_time_dict[ code] >= 1000 and len(origin_datas) > 0: self.l2_save_time_dict[code] = origin_start_time logger_l2_latest_data.info("{}#{}#{}", code, capture_time, origin_datas[-1]) # 10ms的网络传输延时 capture_timestamp = __start_time - process_time - 10 # print("截图时间：", process_time) __start_time = l2_data_log.l2_time(code, round(time.time() * 1000) - __start_time, "截图时间：{} 数据解析时间".format(process_time)) cid, pid = gpcode_manager.get_listen_code_pos(code) __start_time = l2_data_log.l2_time(code, round(time.time() * 1000) - __start_time, "l2获取代码位置耗时") # 判断目标代码位置是否与上传数据位置一致 if cid is not None and pid is not None and client == int(cid) and channel == int(pid): # l2.l2_data_util.set_l2_data_latest_count(code, len(origin_datas)) l2_data_util.save_l2_latest_data_number(code, origin_datas_count) # 保存l2数据条数 if not origin_datas: # or not l2.l2_data_util.is_origin_data_diffrent(origin_datas,self.latest_oringin_data.get(code)): raise Exception("无新增数据") # 保存最近的数据 self.latest_oringin_data[code] = origin_datas limit_up_price = gpcode_manager.get_limit_up_price(code) datas = l2.l2_data_util.L2DataUtil.format_l2_data(origin_datas, code, limit_up_price) try: # 校验客户端代码 l2_code_operate.verify_with_l2_data_pos_info(code, client, channel) __start_time = round(time.time() * 1000) if gpcode_manager.is_listen(code): __start_time = l2_data_log.l2_time(code, round(time.time() * 1000) - __start_time, "l2外部数据预处理耗时") l2_data_manager_new.L2TradeDataProcessor.process(code, datas, capture_timestamp) __start_time = l2_data_log.l2_time(code, round(time.time() * 1000) - __start_time, "l2数据有效处理外部耗时", False) # 保存原始数据数量 # l2_data_util.save_l2_latest_data_number(code, len(origin_datas)) # if round(time.time() * 1000) - __start_time > 20: # l2_data_log.l2_time(code, round(time.time() * 1000) - __start_time, # "异步保存原始数据条数耗时", # False) except l2_data_manager.L2DataException as l: # 单价不符 if l.get_code() == l2_data_manager.L2DataException.CODE_PRICE_ERROR: key = "{}-{}-{}".format(client, channel, code) if key not in self.l2_data_error_dict or round( time.time() * 1000) - self.l2_data_error_dict[key] > 10000: # self.l2CodeOperate.repaire_l2_data(code) logger_l2_error.warning("code-{} l2单价错误:{}", code, l.msg) # 单价不一致时需要移除代码重新添加 l2_code_operate.L2CodeOperate().remove_l2_listen(code, "l2监听单价错误") self.l2_data_error_dict[key] = round(time.time() * 1000) except Exception as e: print("异常", str(e), code) logging.exception(e) logger_l2_error.error("出错：{}".format(str(e))) logger_l2_error.error("内容：{}".format(_str)) finally: __end_time = round(time.time() * 1000) # 只记录大于40ms的数据 if __end_time - origin_start_time > 100: l2_data_log.l2_time(code, round(time.time() * 1000) - origin_start_time, "l2数据处理总耗时", True) except Exception as e: if str(e).find("新增数据"): pass else: logger_l2_error.exception(e) elif type == 1: # 设置股票代码 data_list, is_add = data_process.parseGPCode(_str) ZYLTGBUtil.save_list(data_list) code_list = [] for data in data_list: code_list.append(data["code"]) # 获取基本信息 code_datas = HistoryKDatasUtils.get_gp_latest_info(code_list) # if is_add: # gpcode_manager.add_gp_list(code_datas) # else: # gpcode_manager.set_gp_list(code_datas) if not is_add: # 同步同花顺目标代码 t1 = threading.Thread(target=lambda: sync_target_codes_to_ths()) t1.setDaemon(True) t1.start() elif type == 2: # 涨停代码 dataList, is_add = data_process.parseGPCode(_str) # 设置涨停时间 gpcode_manager.set_limit_up_list(dataList) # 保存到内存中 if dataList: global_data_loader.add_limit_up_codes(dataList) ths_industry_util.set_industry_hot_num(dataList) # 保存涨停时间 gp_list = gpcode_manager.get_gp_list() gp_code_set = set(gp_list) now_str = tool.get_now_time_str() if dataList: for d in dataList: if d["time"] == "00:00:00" or tool.get_time_as_second(now_str) < tool.get_time_as_second( d["time"]): continue if d["code"] not in gp_code_set: continue # 获取是否有涨停时间 # if limit_up_time_manager.get_limit_up_time(d["code"]) is None: # limit_up_time_manager.save_limit_up_time(d["code"], d["time"]) elif type == 22: print("---接受到首板代码") try: if int(tool.get_now_time_str().replace(":", "")) < int("092500"): raise Exception('未到接受时间') # 首板代码 dataList, is_add = data_process.parseGPCode(_str) tick_datas = first_target_code_data_processor.process_first_codes_datas(dataList) # 保存现价 self.first_tick_datas.clear() self.first_tick_datas.extend(tick_datas) except Exception as e: logging.exception(e) finally: print("首板代码处理完毕：") return_str = socket_util.load_header(json.dumps({"code": 0}).encode("utf-8")).decode("utf-8") elif type == 3: # 交易成功信息 dataList = data_process.parseList(_str) try: trade_manager.process_trade_success_data(dataList) except Exception as e: logging.exception(e) trade_manager.save_trade_success_data(dataList) elif type == 5: logger_trade_delegate.debug("接收到委托信息") __start_time = round(time.time() * 1000) try: # 交易委托信息 dataList = data_process.parseList(_str) if self.last_trade_delegate_data != _str: self.last_trade_delegate_data = _str # 保存委托信息 logger_trade_delegate.info(dataList) try: # 设置申报时间 for item in dataList: apply_time = item["apply_time"] if apply_time and len(apply_time) >= 8: code = item["code"] trade_state = trade_manager.CodesTradeStateManager().get_trade_state(code) # 设置下单状态的代码为已委托 if trade_state == trade_manager.TRADE_STATE_BUY_PLACE_ORDER: origin_apply_time = apply_time apply_time = apply_time[0:6] apply_time = "{}:{}:{}".format(apply_time[0:2], apply_time[2:4], apply_time[4:6]) ms = origin_apply_time[6:9] if int(ms) > 500: # 时间+1s apply_time = tool.trade_time_add_second(apply_time, 1) print(apply_time) except Exception as e: logging.exception(e) try: trade_manager.process_trade_delegate_data(dataList) except Exception as e: logging.exception(e) trade_manager.save_trade_delegate_data(dataList) # 刷新交易界面 if trade_gui is not None: trade_gui.THSGuiTrade().refresh_data() finally: pass elif type == 4: # 行业代码信息 dataList = data_process.parseList(_str) codes = [] for datas in dataList: for d in datas: name = ths_industry_util.get_name_by_code(d['code']) if not name or name == 'None': codes.append(d["code"]) # 根据代码获取代码名称 codes_name = {} if codes: codes_name = HistoryKDatasUtils.get_gp_codes_names(codes) ths_industry_util.save_industry_code(dataList, codes_name) elif type == 6: # 可用金额 datas = data_process.parseData(_str) client = datas["client"] money = datas["money"] # TODO存入缓存文件 trade_manager.AccountAvailableMoneyManager().set_available_money(client, money) # l2交易队列 elif type == 10: # 可用金额 __start_time = time.time() datas = data_process.parseData(_str) channel = datas["channel"] code = datas["code"] msg = "" try: if not gpcode_manager.is_in_gp_pool( code) and not gpcode_manager.FirstGPCodesManager().is_in_first_gp_codes_cache(code): # 没在目标代码中且没有在首板今日历史代码中 raise Exception("代码没在监听中") data = datas["data"] buy_time = data["buyTime"] buy_one_price = data["buyOnePrice"] buy_one_volumn = data["buyOneVolumn"] sell_one_price = data["sellOnePrice"] sell_one_volumn = data["sellOneVolumn"] buy_queue = data["buyQueue"] if buy_one_price is None: print('买1价没有，', code) limit_up_price = gpcode_manager.get_limit_up_price(code) if limit_up_price is not None: code_price_manager.Buy1PriceManager().process(code, buy_one_price, buy_one_volumn, buy_time, limit_up_price, sell_one_price, sell_one_volumn) _start_time = time.time() msg += "买1价格处理：" + f"{_start_time - __start_time} " buy_queue_result_list = self.tradeBuyQueue.save(code, limit_up_price, buy_one_price, buy_time, buy_queue) msg += "买队列保存：" + f"{time.time() - _start_time} " _start_time = time.time() if buy_queue_result_list: # 有数据 try: buy_one_price_ = decimal.Decimal(round(float(buy_one_price), 2)).quantize( decimal.Decimal("0.00")) # 获取执行位时间 order_begin_pos = l2_data_manager.TradePointManager().get_buy_compute_start_data_cache(code) if True: # 只有下单过后才获取交易进度 exec_time = None try: if order_begin_pos.buy_exec_index and order_begin_pos.buy_exec_index > -1: exec_time = \ l2.l2_data_util.local_today_datas.get(code)[order_begin_pos.buy_exec_index]["val"]["time"] except: pass buy_progress_index = self.tradeBuyQueue.compute_traded_index(code, buy_one_price_, buy_queue_result_list, exec_time) if buy_progress_index is not None: LCancelBigNumComputer().set_trade_progress(code, order_begin_pos.buy_single_index, buy_progress_index, l2.l2_data_util.local_today_datas.get( code)) logger_l2_trade_buy_queue.info("获取成交位置成功： code-{} index-{} 数据-{}", code, buy_progress_index, json.dumps(buy_queue_result_list)) else: raise Exception("暂未获取到交易进度") msg += "计算成交进度：" + f"{time.time() - _start_time} " _start_time = time.time() except Exception as e: logging.exception(e) print("买入队列", code, buy_queue_result_list) logger_l2_trade_buy_queue.warning("获取成交位置失败： code-{} 原因-{} 数据-{}", code, str(e), json.dumps(buy_queue_result_list)) # buy_queue是否有变化 if self.l2_trade_buy_queue_dict.get( code) is None or buy_queue != self.l2_trade_buy_queue_dict.get( code): self.l2_trade_buy_queue_dict[code] = buy_queue logger_l2_trade_buy_queue.info("{}-{}", code, buy_queue) msg += "保存记录日志：" + f"{time.time() - _start_time} " _start_time = time.time() # 保存最近的记录 if self.ths_l2_trade_queue_manager.save_recod(code, data): if buy_time != "00:00:00": logger_l2_trade_queue.info("{}-{}", code, data) need_sync, need_cancel, cancel_msg = self.buy1_volumn_manager.save(code, buy_time, int(buy_one_volumn), buy_one_price) # if need_sync: # # 同步数据 # s = time.time() # L2LimitUpMoneyStatisticUtil.verify_num(code, int(buy_one_volumn), buy_time) # msg += "量校验："+f"{time.time()-s} " # print(buy_time, buy_one_price, buy_one_volumn) # print("L2买卖队列",datas) msg += "买1处理：" + f"{time.time() - _start_time} " _start_time = time.time() except: pass finally: space = time.time() - __start_time if space > 0.1: logger_debug.info("{}成交队列处理时间：{},{}", code, space, msg) elif type == 20: # 登录 data = data_process.parse(_str)["data"] try: client_id, _authoritys = authority.login(data["account"], data["pwd"]) return_str = data_process.toJson( {"code": 0, "data": {"client": int(client_id), "authoritys": json.loads(_authoritys)}}) except Exception as e: return_str = data_process.toJson({"code": 1, "msg": str(e)}) # 现价更新 elif type == 40: datas = data_process.parse(_str)["data"] if datas is None: datas = [] print("二板现价") # 获取暂存的二版现价数据 if self.first_tick_datas: datas.extend(self.first_tick_datas) if datas is not None: print("二板现价数量", len(datas)) for item in datas: volumn = item["volume"] volumnUnit = item["volumeUnit"] code_volumn_manager.save_today_volumn(item["code"], volumn, volumnUnit) current_price_process_manager.accept_prices(datas) # L2现价更新 elif type == 41: datas = data_process.parse(_str)["data"] if datas: for d in datas: code = d["code"] self.__l2_current_price_data[code] = d elif type == 50: data = data_process.parse(_str)["data"] if data is not None: index = data["index"] code_name = data["codeName"].replace(" ", "") volumn = data["volumn"] price = data["price"] time_ = data["time"] code = global_util.name_codes.get(code_name) if code is None: global_data_loader.load_name_codes() code = global_util.name_codes.get(code_name) if code is not None: # 记录日志 if self.latest_buy1_volumn_dict.get(code) != "{}-{}".format(volumn, price): # 记录数据 logger_buy_1_volumn_record.info("{}-{}", code, data) self.latest_buy1_volumn_dict[code] = "{}-{}".format(volumn, price) # 校正时间 time_ = tool.compute_buy1_real_time(time_) # 保存数据 need_sync, need_cancel, cancel_msg = self.buy1_volumn_manager.save(code, time_, volumn, price) # if need_cancel: # l2_data_manager_new.L2TradeDataProcessor.cancel_buy(code, cancel_msg, "trade_queue") # if need_sync: # # 同步数据 # L2LimitUpMoneyStatisticUtil.verify_num(code, volumn, time_) elif type == 30: # 心跳信息 data = data_process.parse(_str)["data"] client_id = data["client"] thsDead = data.get("thsDead") logger_device.info("（{}）客户端信息：{}".format(client_id, json.dumps(data))) client_manager.saveClientActive(int(client_id), host, thsDead) if constant.is_windows(): # 动态导入 if ths_util.is_ths_dead(client_id): # TODO 重启同花顺 # 报警 l2_clients = authority.get_l2_clients() if client_id in l2_clients: alert_util.alarm() elif type == 60: # L2自启动成功 data = data_process.parse(_str)["data"] client_id = data["client"] print("L2自启动成功", client_id) now_str = tool.get_now_time_str() ts = tool.get_time_as_second(now_str) # 9点25到9点28之间的自启动就需要批量设置代码,目前永远不执行 if tool.get_time_as_second("09:24:50") <= ts <= tool.get_time_as_second("09:28:00") and False: # 准备批量设置代码 return_json = {"code": 1, "msg": "等待批量设置代码"} return_str = json.dumps(return_json) # 获取排名前16位的代码 codes = trade_data_manager.CodeActualPriceProcessor().get_top_rate_codes(16) codes = sorted(codes) if client_id == 2: codes = codes[:constant.L2_CODE_COUNT_PER_DEVICE] else: codes = codes[constant.L2_CODE_COUNT_PER_DEVICE:] codes_datas = [] for i in range(0, constant.L2_CODE_COUNT_PER_DEVICE): if i >= len(codes): break codes_datas.append((i, codes[i])) # 如果设置失败需要重试2次 for i in range(0, 3): set_success = l2_code_operate.betch_set_client_codes(client_id, codes_datas) if set_success: break else: time.sleep(3) else: return_json = {"code": 0, "msg": "开启在线状态"} return_str = json.dumps(return_json) elif type == 70: # 选股宝热门概念 data_json = data_process.parse(_str) day = data_json["day"] datas = data_json["data"] # if datas: # hot_block_data_process.save_datas(day, datas) print(datas) # 获取最近2个交易日涨停代码 elif type == 72: day = tool.get_now_date_str() data_dict = {} for i in range(0, 2): day = HistoryKDatasUtils.get_previous_trading_date(day) data_list = list(block_info.KPLLimitUpDataRecordManager.list_all(day)) codes_set = set() if data_list: for d in data_list: if len(d[4]) > 6: codes_set.add(d[3]) data_dict[day] = list(codes_set) return_str = json.dumps({"code": 0, "data": data_dict}) elif type == 80: # 撤单 data = json.loads(_str) code = data["data"]["code"] if code: state = trade_manager.CodesTradeStateManager().get_trade_state(code) if state == trade_manager.TRADE_STATE_BUY_PLACE_ORDER or state == trade_manager.TRADE_STATE_BUY_DELEGATED or state == trade_manager.TRADE_STATE_BUY_CANCEL_ING: try: l2_data_manager_new.L2TradeDataProcessor.cancel_buy(code, "手动撤销") return_str = json.dumps({"code": 0}) except Exception as e: return_str = json.dumps({"code": 2, "msg": str(e)}) else: return_str = json.dumps({"code": 1, "msg": "未处于可撤单状态"}) else: return_str = json.dumps({"code": 1, "msg": "请上传代码"}) elif type == 82: # 获取委托列表 data = json.loads(_str) update_time = data["data"]["update_time"] results, update_time = huaxin_trade_record_manager.DelegateRecordManager.list_by_day( tool.get_now_date_str("%Y%m%d"), update_time) return_str = json.dumps( {"code": 0, "data": {"list": results, "updateTime": update_time}, "msg": "请上传代码"}) elif type == 201: # 加入黑名单 data = json.loads(_str) codes = data["data"]["codes"] for code in codes: l2_trade_util.forbidden_trade(code, msg="手动加入") name = gpcode_manager.get_code_name(code) if not name: results = HistoryKDatasUtils.get_gp_codes_names([code]) if results: gpcode_manager.CodesNameManager.add_first_code_name(code, results[code]) return_str = json.dumps({"code": 0}) self.__notify_trade("black_list") elif type == 202: # 加入白名单 data = json.loads(_str) codes = data["data"]["codes"] try: for code in codes: # 自由流通市值>50亿,股价高于30块的不能加白名单 # limit_up_price = gpcode_manager.get_limit_up_price(code) # if float(limit_up_price) > 30: # raise Exception("股价高于30元") # zyltgb = global_util.zyltgb_map.get(code) # if zyltgb is None: # global_data_loader.load_zyltgb() # zyltgb = global_util.zyltgb_map.get(code) # if zyltgb > 50 * 100000000: # raise Exception("自由流通股本大于50亿") gpcode_manager.WhiteListCodeManager().add_code(code) name = gpcode_manager.get_code_name(code) if not name: results = HistoryKDatasUtils.get_gp_codes_names([code]) if results: gpcode_manager.CodesNameManager.add_first_code_name(code, results[code]) return_str = json.dumps({"code": 0}) except Exception as e: return_str = json.dumps({"code": 1, "msg": str(e)}) self.__notify_trade("white_list") elif type == 203: # 移除黑名单 data = json.loads(_str) codes = data["data"]["codes"] for code in codes: l2_trade_util.remove_from_forbidden_trade_codes(code) return_str = json.dumps({"code": 0}) self.__notify_trade("black_list") elif type == 204: # 移除白名单 data = json.loads(_str) codes = data["data"]["codes"] for code in codes: gpcode_manager.WhiteListCodeManager().remove_code(code) return_str = json.dumps({"code": 0}) self.__notify_trade("white_list") elif type == 301: # 黑名单列表 codes = gpcode_manager.BlackListCodeManager().list_codes() datas = [] for code in codes: name = gpcode_manager.get_code_name(code) datas.append(f"{name}:{code}") return_str = json.dumps({"code": 0, "data": datas}) elif type == 302: # 黑名单列表 codes = gpcode_manager.WhiteListCodeManager().list_codes() datas = [] for code in codes: name = gpcode_manager.get_code_name(code) datas.append(f"{name}:{code}") return_str = json.dumps({"code": 0, "data": datas}) elif type == 401: # 加入想要买 data = json.loads(_str) codes = data["data"]["codes"] for code in codes: gpcode_manager.WantBuyCodesManager().add_code(code) name = gpcode_manager.get_code_name(code) if not name: results = HistoryKDatasUtils.get_gp_codes_names([code]) if results: gpcode_manager.CodesNameManager.add_first_code_name(code, results[code]) if "plates" in data["data"]: for i in range(len(data["data"]["plates"])): self.__KPLCodeLimitUpReasonManager.save_reason(codes[i], data["data"]["plates"][i]) return_str = json.dumps({"code": 0}) self.__notify_trade("want_list") elif type == 402: data = json.loads(_str) codes = data["data"]["codes"] for code in codes: gpcode_manager.WantBuyCodesManager().remove_code(code) return_str = json.dumps({"code": 0}) self.__notify_trade("want_list") elif type == 403: plate = None include_codes = set() if _str: data = json.loads(_str) plate = data.get("plate") if plate: code_map = self.__KPLCodeLimitUpReasonManager.list_all() for k in code_map: if code_map[k] == plate: include_codes.add(k) codes = gpcode_manager.WantBuyCodesManager().list_code_cache() datas = [] for code in codes: if plate and plate != '其他' and code not in include_codes: continue name = gpcode_manager.get_code_name(code) datas.append(f"{name}:{code}") return_str = json.dumps({"code": 0, "data": datas}) elif type == 411: data = json.loads(_str) codes = data["data"]["codes"] for code in codes: gpcode_manager.PauseBuyCodesManager().add_code(code) name = gpcode_manager.get_code_name(code) if not name: results = HistoryKDatasUtils.get_gp_codes_names([code]) if results: gpcode_manager.CodesNameManager.add_first_code_name(code, results[code]) return_str = json.dumps({"code": 0}) self.__notify_trade("pause_buy_list") # 加入暂停买入列表 elif type == 412: # 移除暂停买入列表 data = json.loads(_str) codes = data["data"]["codes"] for code in codes: gpcode_manager.PauseBuyCodesManager().remove_code(code) return_str = json.dumps({"code": 0}) self.__notify_trade("pause_buy_list") elif type == 413: # 暂停买入列表 codes = gpcode_manager.PauseBuyCodesManager().list_code() datas = [] for code in codes: name = gpcode_manager.get_code_name(code) datas.append(f"{name}:{code}") return_str = json.dumps({"code": 0, "data": datas}) elif type == 420: # 是否可以撤单 data = json.loads(_str) codes = data["data"]["codes"] code = codes[0] state = trade_manager.CodesTradeStateManager().get_trade_state(code) if state != trade_manager.TRADE_STATE_BUY_CANCEL_SUCCESS and state != trade_manager.TRADE_STATE_BUY_SUCCESS: return_str = json.dumps({"code": 0, "msg": "可以取消"}) else: return_str = json.dumps({"code": 1, "msg": "不可以取消"}) elif type == 430: # 查询代码属性 data = json.loads(_str) code = data["data"]["code"] # 查询是否想买单/白名单/黑名单/暂不买 code_name = gpcode_manager.get_code_name(code) want = gpcode_manager.WantBuyCodesManager().is_in_cache(code) white = gpcode_manager.WhiteListCodeManager().is_in_cache(code) black = l2_trade_util.is_in_forbidden_trade_codes(code) pause_buy = gpcode_manager.PauseBuyCodesManager().is_in_cache(code) desc_list = [] if want: desc_list.append("【想买单】") if white: desc_list.append("【白名单】") if black: desc_list.append("【黑名单】") if pause_buy: desc_list.append("【暂不买】") return_str = json.dumps( {"code": 0, "data": {"code_info": (code, code_name), "desc": "".join(desc_list)}}) elif type == 501: data = json.loads(_str) is_open = data["data"]["open"] if is_open: trade_manager.TradeStateManager().open_buy() else: trade_manager.TradeStateManager().close_buy() return_str = json.dumps({"code": 0, "msg": ("开启成功" if is_open else "关闭成功")}) self.__notify_trade("trade_state") elif type == 502: can_buy = trade_manager.TradeStateManager().is_can_buy_cache() return_str = json.dumps({"code": 0, "data": {"can_buy": can_buy}}) elif type == 503: # 设置交易目标代码的模式 data = json.loads(_str) mode = data["data"]["mode"] try: TradeTargetCodeModeManager().set_mode(mode) return_str = json.dumps({"code": 0, "data": {"mode": mode}}) except Exception as e: return_str = json.dumps({"code": 1, "msg": str(e)}) self.__notify_trade("trade_mode") elif type == 504: # 获取交易目标代码模式 mode = TradeTargetCodeModeManager().get_mode_cache() return_str = json.dumps({"code": 0, "data": {"mode": mode}}) elif type == 601: pass # 加自选 elif type == 602: pass # 移除自选 sk.send(return_str.encode()) # print("----------handler end ----------") def finish(self): super().finish() # 可以不调用父类的finish(),方法，父类的finish方法什么都没做 # print("--------finish方法被执行---") def send_msg(client_id, data): _ip = client_manager.getActiveClientIP(client_id) # print("ip", client_id, _ip) if _ip is None or len(_ip) <= 0: raise Exception("客户端IP为空") socketClient = socket.socket(socket.AF_INET, socket.SOCK_STREAM) socketClient.connect((_ip, 9006)) # 连接socket try: socketClient.send(json.dumps(data).encode()) recv = socketClient.recv(1024) result = str(recv, encoding="gbk") return result finally: socketClient.close() # 客户端心跳机制 def test_client_server(): while True: clients = authority.get_l2_clients() for client in clients: # print("心跳", client) try: send_msg(client, {"action": "test"}) except: pass # 矫正客户端代码 l2_code_operate.correct_client_codes() time.sleep(5) # 获取采集客户端的状态 def get_client_env_state(client): result = send_msg(client, {"action": "getEnvState"}) result = json.loads(result) if result["code"] == 0: return json.loads(result["data"]) else: raise Exception(result["msg"]) # 修复采集客户端 def repair_client_env(client): result = send_msg(client, {"action": "repairEnv"}) result = json.loads(result) if result["code"] != 0: raise Exception(result["msg"]) # 同步目标标的到同花顺 def sync_target_codes_to_ths(): codes = gpcode_manager.get_second_gp_list() code_list = [] for code in codes: code_list.append(code) client = authority._get_client_ids_by_rule("data-maintain") result = send_msg(client[0], {"action": "syncTargetCodes", "data": code_list}) return result # 修复同花顺主站 def repair_ths_main_site(client): result = send_msg(client, {"action": "updateTHSSite"}) result = json.loads(result) if result["code"] != 0: raise Exception(result["msg"]) else: # 测速成功 client_infos = [] for index in range(0, constant.L2_CODE_COUNT_PER_DEVICE): client_infos.append((client, index)) l2_listen_pos_health_manager.init_all(client_infos) if __name__ == "__main__1": # 交易成功无法读取时备用 while True: try: datas = trade_juejin.get_execution_reports() # 上传数据 fdatas = [] for d in datas: fdatas.append( {"code": d[0], "money": d[4], "num": d[2], "price": d[3], "time": d[7], "trade_num": d[5], "type": d[1] - 1}) print(fdatas) if fdatas: try: trade_manager.process_trade_success_data(fdatas) except Exception as e: logging.exception(e) trade_manager.save_trade_success_data(fdatas) except: pass time.sleep(1.5)