blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
Administrator
2023-08-02 a3c05bddc4c61d63b804ee541cf6dd01717ba94d 
 code_attribute/gpcode_manager.py


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




import constant


from db import redis_manager


from db.redis_manager import RedisUtils


from utils import tool


import decimal




@@ -36,7 +37,7 @@




    @classmethod


    def list_first_code_name_dict(cls):


        val = cls.__get_redis().get("gp_list_names_first")


        val = RedisUtils.get(cls.__get_redis(), "gp_list_names_first")


        if val is not None:


            val = json.loads(val)


            return val


@@ -44,7 +45,7 @@




    @classmethod


    def list_second_code_name_dict(cls):


        val = cls.__get_redis().get("gp_list_names")


        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")


        if val is not None:


            val = json.loads(val)


            return val


@@ -52,7 +53,7 @@




    @classmethod


    def get_first_code_name(cls, code):


        val = cls.__get_redis().get("gp_list_names_first")


        val = RedisUtils.get(cls.__get_redis(), "gp_list_names_first")


        if not val:


            return None


        val = json.loads(val)


@@ -63,7 +64,7 @@




    @classmethod


    def get_second_code_name(cls, code):


        val = cls.__get_redis().get("gp_list_names")


        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")


        if not val:


            return None


        val = json.loads(val)


@@ -73,7 +74,7 @@




    @classmethod


    def get_first_name_code(cls, name):


        val = cls.__get_redis().get("gp_list_names_first")


        val = RedisUtils.get(cls.__get_redis(), "gp_list_names_first")


        if not val:


            return None


        val = json.loads(val)


@@ -81,7 +82,7 @@




    @classmethod


    def add_first_code_name(cls, code, name):


        val = cls.__get_redis().get("gp_list_names_first")


        val = RedisUtils.get(cls.__get_redis(), "gp_list_names_first")


        if not val:


            return None


        val = json.loads(val)


@@ -90,7 +91,7 @@




    @classmethod


    def get_second_name_code(cls, name):


        val = cls.__get_redis().get("gp_list_names")


        val = RedisUtils.get(cls.__get_redis(), "gp_list_names")


        if not val:


            return None


        val = json.loads(val)


@@ -109,12 +110,12 @@


    # 删除首板代码名称


    @classmethod


    def clear_first_code_names(cls):


        cls.__get_redis().delete("gp_list_names_first")


        RedisUtils.delete(cls.__get_redis(), "gp_list_names_first")




    # 设置二板代码名称


    @classmethod


    def clear_second_code_names(cls):


        cls.__get_redis().delete("gp_list_names")


        RedisUtils.delete(cls.__get_redis(), "gp_list_names")






# 首板代码管理


@@ -129,8 +130,8 @@


    @classmethod


    def add_record(cls, codes):


        for code in codes:


            cls.__get_redis().sadd("first_code_record", code)


        cls.__get_redis().expire("first_code_record", tool.get_expire())


            RedisUtils.sadd(cls.__get_redis(), "first_code_record", code)


        RedisUtils.expire(cls.__get_redis(), "first_code_record", tool.get_expire())




    @classmethod


    def is_in_first_record(cls, code):


@@ -143,8 +144,8 @@


    @classmethod


    def add_limited_up_record(cls, codes):


        for code in codes:


            cls.__get_redis().sadd("first_code_limited_up_record", code)


        cls.__get_redis().expire("first_code_limited_up_record", tool.get_expire())


            RedisUtils.sadd(cls.__get_redis(), "first_code_limited_up_record", code)


        RedisUtils.expire(cls.__get_redis(), "first_code_limited_up_record", tool.get_expire())




    # 是否涨停过


    @classmethod


@@ -166,12 +167,12 @@




    @classmethod


    def clear(cls):


        cls.__get_redis().delete(cls.__redis_key)


        RedisUtils.delete(cls.__get_redis(), cls.__redis_key)




    @classmethod


    def add_code(cls, code):


        cls.__get_redis().sadd(cls.__redis_key, code)


        cls.__get_redis().expire(cls.__redis_key, tool.get_expire())


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


@@ -183,7 +184,7 @@




    @classmethod


    def list_code(cls):


        return cls.__get_redis().smembers(cls.__redis_key)


        return RedisUtils.smembers(cls.__get_redis(), cls.__redis_key)






# 暂停下单代码管理


@@ -198,12 +199,12 @@




    @classmethod


    def clear(cls):


        cls.__get_redis().delete(cls.__redis_key)


        RedisUtils.delete(cls.__get_redis(), cls.__redis_key)




    @classmethod


    def add_code(cls, code):


        cls.__get_redis().sadd(cls.__redis_key, code)


        cls.__get_redis().expire(cls.__redis_key, tool.get_expire())


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


@@ -215,7 +216,7 @@




    @classmethod


    def list_code(cls):


        return cls.__get_redis().smembers(cls.__redis_key)


        return RedisUtils.smembers(cls.__get_redis(), cls.__redis_key)






def __parse_codes_data(code_datas):


@@ -239,10 +240,10 @@


    codes, name_codes = __parse_codes_data(code_datas)


    redis_instance = __redisManager.getRedis()


    # 删除之前的


    redis_instance.delete("gp_list")


    RedisUtils.delete(redis_instance, "gp_list")


    CodesNameManager.clear_second_code_names()


    for d in codes:


        redis_instance.sadd("gp_list", d)


        RedisUtils.sadd(redis_instance, "gp_list", d)


    CodesNameManager.set_second_code_names(name_codes)






@@ -253,7 +254,7 @@


    redis_instance = __redisManager.getRedis()


    codes, name_codes = __parse_codes_data(code_datas)


    for d in codes:


        redis_instance.sadd("gp_list", d)


        RedisUtils.sadd(redis_instance, "gp_list", d)


    old_name_codes = CodesNameManager.list_second_code_name_dict()


    if old_name_codes is None:


        old_name_codes = name_codes


@@ -272,16 +273,16 @@


    codes_set = set()


    for code in codes:


        codes_set.add(code)


    old_codes_set = redis_instance.smembers("gp_list_first")


    old_codes_set = RedisUtils.smembers(redis_instance, "gp_list_first")


    if old_codes_set is None:


        old_codes_set = set()


    del_set = old_codes_set - codes_set


    add_codes = codes_set - old_codes_set


    for code in add_codes:


        redis_instance.sadd("gp_list_first", code)


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


    for code in del_set:


        redis_instance.srem("gp_list_first", code)


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


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




    old_name_codes = CodesNameManager.list_first_code_name_dict()


    if old_name_codes is None:


@@ -302,7 +303,7 @@


# 获取首板代码


def get_first_gp_codes():


    redis_instance = __redisManager.getRedis()


    return redis_instance.smembers("gp_list_first")


    return RedisUtils.smembers(redis_instance, "gp_list_first")






# 是否在首板里面


@@ -349,17 +350,18 @@


    # 获取基本信息


    redis_instance = __redisManager.getRedis()


    # 删除之前的


    redis_instance.delete("gp_limit_up_list")


    RedisUtils.delete(redis_instance, "gp_limit_up_list")


    for d in gpset:


        redis_instance.sadd("gp_limit_up_list", json.dumps(d))


    redis_instance.expire("gp_limit_up_list", tool.get_expire())


    redis_instance.setex("gp_limit_up_list_update_time", tool.get_expire(), round(time.time() * 1000))


        RedisUtils.sadd(redis_instance, "gp_limit_up_list", json.dumps(d))


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


    RedisUtils.setex(redis_instance, "gp_limit_up_list_update_time", tool.get_expire(), round(time.time() * 1000))






# 获取涨停列表


def get_limit_up_list():


    redis_instance = __redisManager.getRedis()


    return redis_instance.get("gp_limit_up_list_update_time"), redis_instance.smembers("gp_limit_up_list")


    return RedisUtils.get(redis_instance, "gp_limit_up_list_update_time"), RedisUtils.smembers(redis_instance,


                                                                                               "gp_limit_up_list")






def rm_gp(code):


@@ -375,7 +377,7 @@




def get_gp_list():


    redis_instance = __redisManager.getRedis()


    codes = redis_instance.smembers("gp_list")


    codes = RedisUtils.smembers(redis_instance, "gp_list")


    first_codes = get_first_gp_codes()


    return set.union(codes, first_codes)




@@ -383,7 +385,7 @@


# 获取二板代码


def get_second_gp_list():


    redis_instance = __redisManager.getRedis()


    codes = redis_instance.smembers("gp_list")


    codes = RedisUtils.smembers(redis_instance, "gp_list")


    return codes






@@ -402,7 +404,7 @@


# 获取收盘价


def get_price_pre(code):


    redis_instance = __redisManager.getRedis()


    result = redis_instance.get("price-pre-{}".format(code))


    result = RedisUtils.get(redis_instance, "price-pre-{}".format(code))


    if result is not None:


        return float(result)


    return None


@@ -427,7 +429,7 @@


    if code not in codes and not FirstCodeManager.is_in_first_record(code) and not force:


        return


    redis_instance = __redisManager.getRedis()


    redis_instance.setex("price-pre-{}".format(code), tool.get_expire(), str(price))


    RedisUtils.setex(redis_instance, "price-pre-{}".format(code), tool.get_expire(), str(price))






__limit_up_price_dict = {}


@@ -463,7 +465,7 @@


# 获取现价


def get_price(code):


    redis_instance = __redisManager.getRedis()


    result = redis_instance.get("price-{}".format(code))


    result = RedisUtils.get(redis_instance, "price-{}".format(code))


    if result is not None:


        return float(result)


    return None


@@ -472,16 +474,16 @@


# 设置现价


def set_price(code, price):


    redis_instance = __redisManager.getRedis()


    redis_instance.setex("price-{}".format(code), tool.get_expire(), price)


    RedisUtils.setex(redis_instance, "price-{}".format(code), tool.get_expire(), price)






# 获取正在监听的代码


def get_listen_codes():


    redis_instance = __redisManager.getRedis()


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


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


    codes = set()


    for k in keys:


        code = redis_instance.get(k)


        code = RedisUtils.get(redis_instance, k)


        if code is not None and len(code) > 0:


            codes.add(code)


    return codes


@@ -491,7 +493,7 @@


def get_listen_code_by_pos(client_id, pos):


    redis_instance = __redisManager.getRedis()


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


    value = redis_instance.get(key)


    value = RedisUtils.get(redis_instance, key)


    # print("redis:", key,value)


    return value




@@ -499,7 +501,7 @@


# 设置位置的监听代码


def set_listen_code_by_pos(client_id, pos, code):


    redis_instance = __redisManager.getRedis()


    redis_instance.setex("listen_code-{}-{}".format(client_id, pos), tool.get_expire(), code)


    RedisUtils.setex(redis_instance, "listen_code-{}-{}".format(client_id, pos), tool.get_expire(), code)


    # 同步监听的代码集合


    __sync_listen_codes_pos()




@@ -508,12 +510,12 @@


def __sync_listen_codes_pos():


    redis_instance = __redisManager.getRedis()


    # 获取已经正在监听的代码


    keys = redis_instance.keys("code_listen_pos-*")


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


    codes_set = set()


    for key in keys:


        codes_set.add(key.replace("code_listen_pos-", ""))




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


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




    for key in keys:


        result = redis_instance.get(key)


@@ -524,15 +526,15 @@


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


            val = redis_instance.get(key_)


            if val is None:


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


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


            else:


                val = json.loads(val)


                if val[0] != client_id_ or val[1] != pos_:


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


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




    # 移除没有监听的代码


    for code_ in codes_set:


        redis_instance.delete(code_)


        RedisUtils.delete(redis_instance, code_)






# 初始化位置


@@ -540,23 +542,23 @@


    redis_instance = __redisManager.getRedis()


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


    redis_instance.setnx(key, "")


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


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






# 清除所有监听代码


def clear_listen_codes():


    redis_instance = __redisManager.getRedis()


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


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


    for key in keys:


        redis_instance.setex(key, tool.get_expire(), "")


        RedisUtils.setex(redis_instance, key, tool.get_expire(), "")






def clear_first_codes():


    redis_instance = __redisManager.getRedis()


    redis_instance.delete("gp_list_first")


    redis_instance.delete("gp_list_names_first")


    redis_instance.delete("first_code_record")


    redis_instance.delete("first_code_limited_up_record")


    RedisUtils.delete(redis_instance, "gp_list_first")


    RedisUtils.delete(redis_instance, "gp_list_names_first")


    RedisUtils.delete(redis_instance, "first_code_record")


    RedisUtils.delete(redis_instance, "first_code_limited_up_record")






# 获取可以操作的位置


@@ -571,7 +573,7 @@


    for client_id in client_ids:


        redis_instance = __redisManager.getRedis()


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


        keys = redis_instance.keys(k)


        keys = RedisUtils.keys(redis_instance, k)


        # random.shuffle(keys)


        codes = []


        for key in keys:


@@ -613,7 +615,7 @@


    for client_id in client_ids:


        redis_instance = __redisManager.getRedis()


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


        keys = redis_instance.keys(k)


        keys = RedisUtils.keys(redis_instance, k)


        for key in keys:


            code = redis_instance.get(key)


            if not code:


@@ -670,27 +672,29 @@


# 设置正在操作的代码


def set_operate(code):


    redis_instance = __redisManager.getRedis()


    redis_instance.setex("gp_operate-{}".format(code), 30, "1")


    RedisUtils.setex(


        redis_instance, "gp_operate-{}".format(code), 30, "1")






# 批量设置正在操作的代码


def set_operates(codes):


    redis_instance = __redisManager.getRedis()


    for code in codes:


        redis_instance.setex("gp_operate-{}".format(code), 30, "1")


        RedisUtils.setex(


            redis_instance, "gp_operate-{}".format(code), 30, "1")






# 移除正在操作的代码


def rm_operate(code):


    redis_instance = __redisManager.getRedis()


    redis_instance.delete("gp_operate-{}".format(code))


    RedisUtils.delete(redis_instance, "gp_operate-{}".format(code))






# 批量移除正在操作的代码


def rm_operates(codes):


    redis_instance = __redisManager.getRedis()


    for code in codes:


        redis_instance.delete("gp_operate-{}".format(code))


        RedisUtils.delete(redis_instance, "gp_operate-{}".format(code))






if __name__ == '__main__':