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


@@ -135,7 +144,7 @@




    @classmethod


    def is_in_first_record(cls, code):


        if RedisUtils.sismember( cls.__get_redis(), "first_code_record", code):


        if RedisUtils.sismember(cls.__get_redis(), "first_code_record", code):


            return True


        else:


            return False


@@ -150,7 +159,7 @@


    # 是否涨停过


    @classmethod


    def is_limited_up(cls, code):


        if RedisUtils.sismember( cls.__get_redis(), "first_code_limited_up_record", code):


        if RedisUtils.sismember(cls.__get_redis(), "first_code_limited_up_record", code):


            return True


        else:


            return False


@@ -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):


        RedisUtils.srem( cls.__get_redis(),cls.__redis_key, 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)


        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):


@@ -212,7 +235,7 @@




    @classmethod


    def is_in(cls, code):


        return RedisUtils.sismember( cls.__get_redis(), cls.__redis_key, code)


        return RedisUtils.sismember(cls.__get_redis(), cls.__redis_key, code)




    @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


@@ -281,8 +307,13 @@


    for code in add_codes:


        RedisUtils.sadd(redis_instance, "gp_list_first", code)


    for code in del_set:


        redis_instance.srem("gp_list_first", code)


    RedisUtils.expire(redis_instance, "gp_list_first", tool.get_expire())


        RedisUtils.srem(redis_instance, "gp_list_first", code)


    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:


@@ -297,7 +328,7 @@


def remove_first_gp_code(codes):


    redis_instance = __redisManager.getRedis()


    for code in codes:


        redis_instance.srem("gp_list_first", code)


        RedisUtils.srem(redis_instance, "gp_list_first", code)






# 获取首板代码


@@ -309,7 +340,7 @@


# 是否在首板里面


def is_in_first_gp_codes(code):


    redis_instance = __redisManager.getRedis()


    return redis_instance.sismember("gp_list_first", code)


    return RedisUtils.sismember(redis_instance, "gp_list_first", code)






# 获取名称对应的代码


@@ -366,13 +397,13 @@




def rm_gp(code):


    redis_instance = __redisManager.getRedis()


    redis_instance.srem("gp_list", code)


    RedisUtils.srem(redis_instance, "gp_list", code)


    remove_first_gp_code([code])






def is_in_gp_pool(code):


    redis_instance = __redisManager.getRedis()


    return redis_instance.sismember("gp_list", code) or is_in_first_gp_codes(code)


    return RedisUtils.sismember(redis_instance, "gp_list", code) or is_in_first_gp_codes(code)






def get_gp_list():


@@ -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)




@@ -518,13 +555,13 @@


    keys = RedisUtils.keys(redis_instance, "listen_code-*-*")




    for key in keys:


        result = redis_instance.get(key)


        result = RedisUtils.get(redis_instance, key)


        if result:


            # 移除需要添加的代码


            codes_set.discard(result)


            client_id_, pos_, code_ = int(key.split("-")[1]), int(key.split("-")[2]), result


            key_ = "code_listen_pos-{}".format(code_)


            val = redis_instance.get(key_)


            val = RedisUtils.get(redis_instance, key_)


            if val is None:


                RedisUtils.setex(redis_instance, key_, tool.get_expire(), json.dumps((client_id_, pos_)))


            else:


@@ -541,7 +578,7 @@


def init_listen_code_by_pos(client_id, pos):


    redis_instance = __redisManager.getRedis()


    key = "listen_code-{}-{}".format(client_id, pos)


    redis_instance.setnx(key, "")


    RedisUtils.setnx(redis_instance, key, "")


    RedisUtils.expire(redis_instance, key, tool.get_expire())






@@ -580,7 +617,7 @@


            index = key.split("-")[-1]


            if int(index) + 1 > constant.L2_CODE_COUNT_PER_DEVICE:


                continue


            result = redis_instance.get(key)


            result = RedisUtils.get(redis_instance, key)


            if result is None or len(result) == 0:


                available_positions.append((client_id, int(key.replace("listen_code-{}-".format(client_id), ""))))


            else:


@@ -617,7 +654,7 @@


        k = "listen_code-{}-*".format(client_id)


        keys = RedisUtils.keys(redis_instance, k)


        for key in keys:


            code = redis_instance.get(key)


            code = RedisUtils.get(redis_instance, key)


            if not code:


                free_count += 1


    return free_count


@@ -626,7 +663,7 @@


# 获取正在监听的代码的位置


def get_listen_code_pos(code):


    redis_instance = __redisManager.getRedis()


    val = redis_instance.get("code_listen_pos-{}".format(code))


    val = RedisUtils.get(redis_instance, "code_listen_pos-{}".format(code))


    if val is None:


        return None, None


    val = json.loads(val)


@@ -642,7 +679,7 @@


# 是否正在监听


def is_listen(code):


    redis_instance = __redisManager.getRedis()


    val = redis_instance.get("code_listen_pos-{}".format(code))


    val = RedisUtils.get(redis_instance, "code_listen_pos-{}".format(code))


    if val is None:


        return False


    else:


@@ -666,7 +703,7 @@


# 是否正在操作


def is_operate(code):


    redis_instance = __redisManager.getRedis()


    return redis_instance.get("gp_operate-{}".format(code)) is not None


    return RedisUtils.get(redis_instance, "gp_operate-{}".format(code)) is not None






# 设置正在操作的代码