redis封装

Administrator

2023-08-03 aabf5f98ad26e467645033c2cc80f5ff7e678aa2

redis封装

 code_attribute/code_data_util.py

@@ -19,7 +19,7 @@
    if code == '600066':
        print('进入调试')
    # 昨日收盘价
    price_close = gpcode_manager.get_price_pre(code)
    price_close = gpcode_manager.get_price_pre_cache(code)
    max_price = tool.to_price(decimal.Decimal(str(price_close)) * decimal.Decimal("1.1"))
    min_price = tool.to_price(decimal.Decimal(str(price_close)) * decimal.Decimal("0.9"))
    if min_price <= decimal.Decimal(str(price)) <= max_price:

 code_attribute/first_target_code_data_processor.py

@@ -105,7 +105,7 @@
        need_get_volumn = False
        if code not in global_util.max60_volumn or global_util.max60_volumn.get(code) is None:
            need_get_volumn = True
        if not need_get_volumn and code_nature_analyse.CodeNatureRecordManager.get_nature(
        if not need_get_volumn and code_nature_analyse.CodeNatureRecordManager.get_nature_cache(
                code) is None:
            need_get_volumn = True
        if need_get_volumn:
@@ -191,7 +191,7 @@
        if is_limit_up:
            # 加入首板涨停
            gpcode_manager.FirstCodeManager.add_limited_up_record([code])
        pricePre = gpcode_manager.get_price_pre(code)
        pricePre = gpcode_manager.get_price_pre_cache(code)
        if pricePre is None:
            inited_data.re_set_price_pres([code])


 code_attribute/gpcode_manager.py

@@ -17,6 +17,7 @@

class CodesNameManager:
    redisManager = redis_manager.RedisManager(0)
    __gp_list_names_first_cache = []

    @classmethod
    def __get_redis(cls):
@@ -44,8 +45,8 @@
        return None

    @classmethod
    def list_second_code_name_dict(cls):
        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")
    def list_first_code_name_dict_cache(cls):
        val = RedisUtils.get(cls.__get_redis(), "gp_list_names_first")
        if val is not None:
            val = json.loads(val)
            return val
@@ -61,16 +62,6 @@
            if val[k] == code:
                return k
        return None

    @classmethod
    def get_second_code_name(cls, code):
        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")
        if not val:
            return None
        val = json.loads(val)
        for k in val:
            if val[k] == code:
                return k

    @classmethod
    def get_first_name_code(cls, name):
@@ -89,6 +80,34 @@
        val[name] = code
        cls.set_first_code_names(val)

    # 设置首板代码名称
    @classmethod
    def set_first_code_names(cls, datas):
        RedisUtils.set(cls.__get_redis(), "gp_list_names_first", json.dumps(datas))

    # 删除首板代码名称
    @classmethod
    def clear_first_code_names(cls):
        RedisUtils.delete(cls.__get_redis(), "gp_list_names_first")

    @classmethod
    def list_second_code_name_dict(cls):
        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")
        if val is not None:
            val = json.loads(val)
            return val
        return None

    @classmethod
    def get_second_code_name(cls, code):
        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")
        if not val:
            return None
        val = json.loads(val)
        for k in val:
            if val[k] == code:
                return k

    @classmethod
    def get_second_name_code(cls, name):
        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")
@@ -97,20 +116,10 @@
        val = json.loads(val)
        return val.get(name)

    # 设置首板代码名称
    @classmethod
    def set_first_code_names(cls, datas):
        RedisUtils.set(cls.__get_redis(), "gp_list_names_first", json.dumps(datas))

    # 设置二板代码名称
    @classmethod
    def set_second_code_names(cls, datas):
        RedisUtils.set(cls.__get_redis(), "gp_list_names", json.dumps(datas))

    # 删除首板代码名称
    @classmethod
    def clear_first_code_names(cls):
        RedisUtils.delete(cls.__get_redis(), "gp_list_names_first")

    # 设置二板代码名称
    @classmethod
@@ -160,6 +169,7 @@
class WantBuyCodesManager:
    redisManager = redis_manager.RedisManager(0)
    __redis_key = "want_buy_codes"
    __want_buy_codes_cache = set()

    @classmethod
    def __get_redis(cls):
@@ -171,16 +181,29 @@

    @classmethod
    def add_code(cls, code):
        cls.__want_buy_codes_cache.add(code)
        RedisUtils.sadd(cls.__get_redis(), cls.__redis_key, code)
        RedisUtils.expire(cls.__get_redis(), cls.__redis_key, tool.get_expire())

    @classmethod
    def remove_code(cls, code):
        cls.__want_buy_codes_cache.discard(code)
        RedisUtils.srem(cls.__get_redis(), cls.__redis_key, code)

    @classmethod
    def sync(cls):
        cls.__want_buy_codes_cache.clear()
        codes = RedisUtils.smembers(cls.__get_redis(), cls.__redis_key)
        if codes:
            cls.__want_buy_codes_cache += set(codes)

    @classmethod
    def is_in(cls, code):
        return RedisUtils.sismember(cls.__get_redis(), cls.__redis_key, code)

    @classmethod
    def is_in_cache(cls, code):
        return code in cls.__want_buy_codes_cache

    @classmethod
    def list_code(cls):
@@ -264,6 +287,9 @@
    CodesNameManager.set_second_code_names(old_name_codes)


__gp_list_first_codes_cache = set()


# -------------------------------首板代码管理-------------------------------
# 添加首板代码
# code_datas 掘金返回的数据
@@ -273,7 +299,7 @@
    codes_set = set()
    for code in codes:
        codes_set.add(code)
    old_codes_set = RedisUtils.smembers(redis_instance, "gp_list_first")
    old_codes_set = __gp_list_first_codes_cache
    if old_codes_set is None:
        old_codes_set = set()
    del_set = old_codes_set - codes_set
@@ -282,7 +308,12 @@
        RedisUtils.sadd(redis_instance, "gp_list_first", code)
    for code in del_set:
        RedisUtils.srem(redis_instance, "gp_list_first", code)
    RedisUtils.expire(redis_instance, "gp_list_first", tool.get_expire())
    if add_codes or del_set:
        RedisUtils.expire(redis_instance, "gp_list_first", tool.get_expire())
    # 更新缓存
    __gp_list_first_codes_cache.clear()
    for code in codes_set:
        __gp_list_first_codes_cache.add(code)

    old_name_codes = CodesNameManager.list_first_code_name_dict()
    if old_name_codes is None:
@@ -440,7 +471,7 @@
    # 读取内存中的值
    if code in __limit_up_price_dict:
        return __limit_up_price_dict[code]
    price = get_price_pre(code)
    price = get_price_pre_cache(code)
    if price is None:
        return None
    limit_up_price = tool.to_price(decimal.Decimal(str(price)) * decimal.Decimal("1.1"))
@@ -456,7 +487,7 @@

# 获取跌停价
def get_limit_down_price(code):
    price = get_price_pre(code)
    price = get_price_pre_cache(code)
    if price is None:
        return None
    return tool.to_price(decimal.Decimal(str(price)) * decimal.Decimal("0.9"))
@@ -471,8 +502,14 @@
    return None


__current_price_cache = {}


# 设置现价
def set_price(code, price):
    if code in __current_price_cache and __current_price_cache[code] == price:
        return
    __current_price_cache[code] = price
    redis_instance = __redisManager.getRedis()
    RedisUtils.setex(redis_instance, "price-{}".format(code), tool.get_expire(), price)


 l2/l2_data_manager_new.py

@@ -830,7 +830,7 @@
                    return False, True, f"自由流通200亿以上，买1剩余档数大于10档，买一（{buy1_price}）涨停（{limit_up_price}）"

        open_limit_up_lowest_price = code_price_manager.Buy1PriceManager.get_open_limit_up_lowest_price(code)
        price_pre_close = gpcode_manager.get_price_pre(code)
        price_pre_close = gpcode_manager.get_price_pre_cache(code)
        if open_limit_up_lowest_price and (
                float(open_limit_up_lowest_price) - price_pre_close) / price_pre_close < 0.05:
            return False, True, f"炸板后最低价跌至5%以下"
@@ -863,7 +863,7 @@
                                      cls.__l2PlaceOrderParamsManagerDict[code].score_index,
                                      cls.__l2PlaceOrderParamsManagerDict[code].score_info)

        if not gpcode_manager.WantBuyCodesManager.is_in(code):
        if not gpcode_manager.WantBuyCodesManager.is_in_cache(code):
            if TradeTargetCodeModeManager.get_mode() == TradeTargetCodeModeManager.MODE_ONLY_BUY_WANT_CODES:
                return False, True, f"只买想买单中的代码"
            score_index = cls.__l2PlaceOrderParamsManagerDict[code].score_index

 output/code_info_output.py

@@ -85,7 +85,7 @@
    if l2_trade_util.WhiteListCodeManager.is_in(code):
        code_extra_infos.append("白名单")
    # 获取白名单,黑名单
    if code_attribute.gpcode_manager.WantBuyCodesManager.is_in(code):
    if code_attribute.gpcode_manager.WantBuyCodesManager.is_in_cache(code):
        code_extra_infos.append("想买单")
    if code_attribute.gpcode_manager.PauseBuyCodesManager.is_in(code):
        code_extra_infos.append("暂不买")

 server.py

@@ -810,7 +810,7 @@
                    code = data["data"]["code"]
                    # 查询是否想买单/白名单/黑名单/暂不买
                    code_name = gpcode_manager.get_code_name(code)
                    want = gpcode_manager.WantBuyCodesManager.is_in(code)
                    want = gpcode_manager.WantBuyCodesManager.is_in_cache(code)
                    white = l2_trade_util.WhiteListCodeManager.is_in(code)
                    black = l2_trade_util.is_in_forbidden_trade_codes(code)
                    pause_buy = gpcode_manager.PauseBuyCodesManager.is_in(code)

 trade/current_price_process_manager.py

@@ -43,7 +43,7 @@
            code, price = d["code"], float(d["price"])
            gpcode_manager.set_price(code, price)
            # 获取收盘价
            pricePre = gpcode_manager.get_price_pre(code)
            pricePre = gpcode_manager.get_price_pre_cache(code)
            if pricePre is not None:
                rate = round((price - pricePre) * 100 / pricePre, 2)
                if first_codes and code in first_codes:

 trade/l2_trade_factor.py

@@ -99,7 +99,7 @@
                break
        self.score_index = score_index
        # 只要加入想买单的，全部执行主动买入二星方案
        if gpcode_manager.WantBuyCodesManager.is_in(code):
        if gpcode_manager.WantBuyCodesManager.is_in_cache(code):
            self.score_index = 1
            self.is_want_buy = True
        else: