redis封装

Administrator

2023-08-03 2984f2b9363959b4147cdb669eeaa67060500d72

redis封装

 output/code_info_output.py

@@ -80,7 +80,7 @@
    day = tool.get_now_date_str()
    is_target_code = gpcode_manager.FirstCodeManager.is_in_first_record(code)
    code_extra_infos = []
    if l2_trade_util.BlackListCodeManager.is_in(code):
    if l2_trade_util.BlackListCodeManager.is_in_cache(code):
        code_extra_infos.append("黑名单")
    if l2_trade_util.WhiteListCodeManager.is_in(code):
        code_extra_infos.append("白名单")

 trade/l2_trade_util.py

@@ -24,7 +24,7 @@

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

    @classmethod
    def list_codes(cls):
@@ -32,11 +32,12 @@

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


class BlackListCodeManager:
    __redis_manager = redis_manager.RedisManager(2)
    __forbidden_trade_codes_cache = set()

    @classmethod
    def __get_redis(cls):
@@ -44,23 +45,31 @@

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

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

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

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

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

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


@@ -88,7 +97,7 @@


def is_in_forbidden_trade_codes(code):
    return BlackListCodeManager.is_in(code)
    return BlackListCodeManager.is_in_cache(code)


if __name__ == "__main__":