选股宝板块优化

Administrator

2023-03-15 68464c679ae5e1ae35e7e67e3b339ba0f939cbd3

 trade/l2_trade_util.py

@@ -5,30 +5,78 @@
__redis_manager = redis_manager.RedisManager(2)


class WhiteListCodeManager:
    __redis_manager = redis_manager.RedisManager(2)

    @classmethod
    def __get_redis(cls):
        return cls.__redis_manager.getRedis()

    @classmethod
    def add_code(cls, code):
        cls.__get_redis().sadd("white_list_codes", code)
        cls.__get_redis().expire("white_list_codes", tool.get_expire())

    @classmethod
    def remove_code(cls, code):
        cls.__get_redis().srem("white_list_codes", code)

    @classmethod
    def is_in(cls, code):
        return cls.__get_redis().sismember("white_list_codes", code)

    @classmethod
    def list_codes(cls):
        return cls.__get_redis().smembers("white_list_codes")

    @classmethod
    def clear(cls):
        cls.__get_redis().delete("white_list_codes")


class BlackListCodeManager:
    __redis_manager = redis_manager.RedisManager(2)

    @classmethod
    def __get_redis(cls):
        return cls.__redis_manager.getRedis()

    @classmethod
    def add_code(cls, code):
        cls.__get_redis().sadd("forbidden-trade-codes", code)
        cls.__get_redis().expire("forbidden-trade-codes", tool.get_expire())

    @classmethod
    def remove_code(cls, code):
        cls.__get_redis().srem("forbidden-trade-codes", code)

    @classmethod
    def is_in(cls, code):
        return cls.__get_redis().sismember("forbidden-trade-codes", code)

    @classmethod
    def list_codes(cls):
        return cls.__get_redis().smembers("forbidden-trade-codes")

    @classmethod
    def clear(cls):
        cls.__get_redis().delete("forbidden-trade-codes")


#  初始化禁止交易代码库
def init_forbidden_trade_codes():
    key = "forbidden-trade-codes"
    redis = __redis_manager.getRedis()
    count = redis.scard(key)
    if count > 0:
        redis.delete(key)
    redis.sadd(key, "000000")
    redis.expire(key, tool.get_expire())
    BlackListCodeManager.clear()
    BlackListCodeManager.add_code("000000")


# 移除禁止交易代码
def remove_from_forbidden_trade_codes(code):
    key = "forbidden-trade-codes"
    redis = __redis_manager.getRedis()
    redis.srem(key, code)
    BlackListCodeManager.remove_code(code)


# 添加代码到禁止交易
def add_to_forbidden_trade_codes(code):
    key = "forbidden-trade-codes"
    redis = __redis_manager.getRedis()
    redis.sadd(key, code)
    redis.expire(key, tool.get_expire())
    BlackListCodeManager.add_code(code)


# 禁止代码交易
@@ -39,10 +87,9 @@


def is_in_forbidden_trade_codes(code):
    key = "forbidden-trade-codes"
    redis = __redis_manager.getRedis()
    return redis.sismember(key, code)
    return BlackListCodeManager.is_in(code)


if __name__ == "__main__":
    add_to_forbidden_trade_codes("000977")
    # add_to_forbidden_trade_codes("000977")
    WhiteListCodeManager.add_code("002019")