Administrator
2023-12-22 f0248f54f90a12f491245f0ee3ccfbe8f477a76b
third_data/data_server.py
@@ -7,6 +7,7 @@
from http.server import BaseHTTPRequestHandler
import dask
from code_attribute.gpcode_manager import BlackListCodeManager
from log_module.log import logger_system, logger_debug
from utils import global_util, tool, data_export_util
from code_attribute import gpcode_manager
@@ -25,7 +26,6 @@
from output import code_info_output, limit_up_data_filter, output_util, kp_client_msg_manager
from trade import bidding_money_manager, trade_manager, l2_trade_util, trade_record_log_util
from trade.l2_trade_util import BlackListCodeManager
import concurrent.futures
# 禁用http.server的日志输出
@@ -62,6 +62,7 @@
            total_datas = KPLLimitUpDataRecordManager.total_datas
        # 通过涨停时间排序
        total_datas = list(total_datas)
        current_reason_codes_dict = KPLLimitUpDataRecordManager.get_current_reason_codes_dict()
        # 统计涨停原因
        limit_up_reason_dict = {}
@@ -84,7 +85,8 @@
            (k, len(limit_up_reason_dict[k]), limit_up_reason_want_count_dict.get(k), limit_up_reason_dict[k][0][5]) for
            k in
            limit_up_reason_dict]
        limit_up_reason_statistic_info.sort(key=lambda x: int(x[1]))
        limit_up_reason_statistic_info.sort(
            key=lambda x: len(current_reason_codes_dict[x[0]]) if x[0] in current_reason_codes_dict else 0)
        limit_up_reason_statistic_info.reverse()
        codes_set = set([d[3] for d in total_datas])
@@ -317,12 +319,25 @@
            # 获取评分信息
            pass
        elif url.path == "/get_kpl_block_info":
            start_time = time.time()
            ps_dict = dict([(k, v[0]) for k, v in parse_qs(url.query).items()])
            code = ps_dict['code']
            try:
                data = code_info_output.get_kpl_block_info(code)
                response_data = json.dumps({"code": 0, "data": data})
                print("get_kpl_block_info 耗时:", time.time() - start_time)
            except Exception as e:
                logger_debug.exception(e)
                logging.exception(e)
        elif url.path == "/get_l2_datas":
            # 获取L2的数据
            ps_dict = dict([(k, v[0]) for k, v in parse_qs(url.query).items()])
            code = ps_dict['code']
            datas = data_export_util.get_l2_datas(code)
            response_data = json.dumps({"code": 0, "data": datas})
            code_name = gpcode_manager.get_code_name(code)
            response_data = json.dumps({"code": 0, "data": {"code": code, "code_name": code_name, "data": datas}})
        elif url.path == "/get_trade_progress":
            # 获取交易进度
            ps_dict = dict([(k, v[0]) for k, v in parse_qs(url.query).items()])
@@ -445,7 +460,8 @@
                    continue
                # 代码,名称,涨停时间,是否炸板,是否想买,是否已经下过单
                codes_info.append(
                    [d[3], d[4], tool.to_time_str(int(d[5])), 1 if d[3] not in now_limit_up_codes else 0, 0, 0])
                    [d[3], d[4], tool.to_time_str(int(d[5])), 1 if d[3] not in now_limit_up_codes else 0, 0, 0, d[12],
                     output_util.money_desc(d[13])])
            codes_info.sort(key=lambda x: x[2])
            # 查询是否为想买单
            want_codes = gpcode_manager.WantBuyCodesManager().list_code_cache()
@@ -499,24 +515,37 @@
            # 获取上个交易日的相同涨停原因的代码信息
            ps_dict = dict([(k, v[0]) for k, v in parse_qs(url.query).items()])
            code = ps_dict["code"]
            day = HistoryKDatasUtils.get_previous_trading_date(tool.get_now_date_str())
            # 获取涨停数据
            # 获取昨日涨停数据
            day = HistoryKDatasUtils.get_previous_trading_date_cache(tool.get_now_date_str())
            limit_up_records = kpl_data_manager.KPLLimitUpDataRecordManager.list_all_cache(day)
            reasons = []
            for d in limit_up_records:
                if d[3] == code:
                    reasons.append(d)
            # 获取代码的原因
            reasons = kpl_data_manager.KPLLimitUpDataRecordManager.list_by_code(code, day)
            if reasons:
                reasons = list(reasons)
                reasons.sort(key=lambda x: x[9])
                reason = reasons[-1][2]
                datas = self.__kplDataManager.get_from_file(kpl_util.KPLDataType.LIMIT_UP, day)
                # 获取涨停数据
                datas = self.__kplDataManager.get_from_file_cache(kpl_util.KPLDataType.LIMIT_UP, day)
                # (代码,名称,首次涨停时间,最近涨停时间,几板,涨停原因,板块,实际流通,主力净额,涨停原因代码,涨停原因代码数量)
                result_list = []
                yesterday_result_list = []
                if datas:
                    for d in datas:
                        if d[5] == reason and d[0] != code:
                            # (代码,名称)
                            result_list.append((d[0], d[1]))
                response_data = json.dumps({"code": 0, "data": {"reason": reason, "data": result_list}})
                            yesterday_result_list.append((d[0], d[1]))
                current_limit_up_list = kpl_data_manager.KPLLimitUpDataRecordManager.latest_origin_datas
                current_result_list = []
                if current_limit_up_list:
                    for c in current_limit_up_list:
                        if c[5] == reason and c[0] != code:
                            current_result_list.append((c[0], c[1]))
                response_data = json.dumps({"code": 0, "data": {"reason": reason,
                                                                "data": {"yesterday": yesterday_result_list,
                                                                         "current": current_result_list}}})
            else:
                response_data = json.dumps({"code": 1, "msg": "昨日未涨停"})