‘功能完善

Administrator

5 天以前 09f84c4ecae2a71e07f6dd64e6066563599a62b6

 api/outside_api_callback.py

@@ -1,10 +1,19 @@
import json
import logging
import os
import threading

import constant

from api.outside_api_command_manager import ActionCallback
from huaxin_client import l1_subscript_codes_manager
from huaxin_client.client_network import SendResponseSkManager
from strategy import strategy_params_settings
from utils import socket_util, middle_api_protocol
from strategy import strategy_params_settings, env_info, strategy_manager
from strategy.env_info import RealTimeEnvInfo
from strategy.strategy_variable_factory import DataLoader
from third_data.history_k_data_manager import TradeDateManager
from third_data.kpl_block_manager import KPLCodeJXBlocksManager
from utils import socket_util, middle_api_protocol, tool


class MyAPICallback(ActionCallback):
@@ -39,22 +48,122 @@
        获取交易参数
        @return:
        """
        result = strategy_params_settings.settings.to_json_str()
        return json.dumps({"code": 0, "data": result})
        result = strategy_params_settings.StrategyParamsSettingsManager().get_settings().to_json_str()
        return {"code": 0, "data": json.loads(result)}

    def __on_get_env(self):
    def __on_set_settings(self, data):
        """
        获取环境信息
        设置交易参数
        @return:
        """
        return json.dumps({"code": 0, "data": {}, "msg": "测试结果"})
        settings = strategy_params_settings.StrategyParamsSettingsManager().get_settings()
        for k in data:
            settings.__setattr__(k, data[k])
        strategy_params_settings.StrategyParamsSettingsManager().set_settings(settings)
        # 同步参数设置
        return {"code": 0, "data": {}}

    def __on_get_env(self, need_hsitory_data):
        """
        获取环境信息
        @param need_hsitory_data: 是否需要历史数据
        @return:
        """
        fdata = {}
        # 实时数据
        fdata["real_time_data"] = RealTimeEnvInfo().to_dict()
        # 历史数据
        fdata["history_data"] = {}
        print("获取环境", os.getpid())
        if need_hsitory_data:
            if tool.get_now_time_str() < '16:00:00':
                # 如果在16:00之前采用当前日期
                day = tool.get_now_date_str()
            else:
                # 如果在16:00之后采用下一个交易日
                day = TradeDateManager().get_next_trade_day(tool.get_now_date_str())
            fdata["history_data"][
                "leading_limit_up_block_codes_count"] = env_info.get_leading_limit_up_block_codes_count(
                day)

            if tool.get_now_time_str() < '16:00:00':
                # 如果在16:00之前采用当前日期
                day = TradeDateManager().get_previous_trade_day(tool.get_now_date_str())
            else:
                # 如果在16:00之后采用下一个交易日
                day = tool.get_now_date_str()
            fdata["history_data"]["k_bars_count"] = env_info.get_history_k_bars(day)

            day = tool.get_now_date_str()
            fdata["history_data"]["kpl_code_jx_blocks_count"] = env_info.get_kpl_code_jx_blocks(day)


        return {"code": 0, "data": fdata, "msg": "测试结果"}

    def __on_update_leading_limit_up_datas(self):
        """
        更新领涨代码信息
        @return:
        """

        def update():
            plates = __DataLoader.get_limit_up_reasons_with_plate_code()
            for p in plates:
                print(p)
                __DataLoader.load_plate_codes(p[0], p[1])

        if tool.get_now_time_str() < '16:00:00':
            # 如果在16:00之前采用当前日期
            day = tool.get_now_date_str()
        else:
            # 如果在16:00之后采用下一个交易日
            day = TradeDateManager().get_next_trade_day(tool.get_now_date_str())
        __DataLoader = DataLoader(day)
        threading.Thread(target=lambda: update(), daemon=True).start()
        return {"code": 0}

    def __on_update_kpl_code_jx_blocks_datas(self):
        """
        更新开盘啦精选板块数据
        @return:
        """

        def update():
            codes = set()
            codes_sh, codes_sz = l1_subscript_codes_manager.get_codes()
            codes |= set([x.decode() for x in codes_sh])
            codes |= set([x.decode() for x in codes_sz])
            KPLCodeJXBlocksManager(day, codes).start_download_blocks()
            # 如果在16:00之前采用当前日期
        day = tool.get_now_date_str()
        threading.Thread(target=lambda: update(), daemon=True).start()
        return {"code": 0}

    def __on_init_data(self):
        try:
            strategy_manager.low_suction_strtegy.load_data()
            return {"code": 0}
        except Exception as e:
            return {"code": 1, "msg": str(e)}

    def OnCommonRequest(self, client_id, request_id, data):
        ctype = data["ctype"]
        result_str = ''
        result_json = {}
        if ctype == "get_settings":
            result_str = self.__on_get_settings()
            result_json = self.__on_get_settings()
        elif ctype == 'set_settings':
            del data["ctype"]
            result_json = self.__on_set_settings(data)
        elif ctype == 'get_env':
            result_str = self.__on_get_env()

        self.send_response(result_str, client_id, request_id)
            # 获取环境数据
            result_json = self.__on_get_env(data.get("history"))
        elif ctype == 'update_leading_limit_up_datas':
            # 更新领涨数据
            result_json = self.__on_update_leading_limit_up_datas()
        elif ctype == 'update_kpl_code_jx_blocks_datas':
            # 更新开盘啦精选数据
            result_json = self.__on_update_kpl_code_jx_blocks_datas()
        elif ctype == 'init_datas':
            # 初始化数据
            result_json = self.__on_init_data()
        self.send_response(result_json, client_id, request_id)