blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
Administrator
2023-08-03 aabf5f98ad26e467645033c2cc80f5ff7e678aa2 
 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)