# 是否在禁止交易代码中
|
from db import redis_manager
|
from utils import tool
|
|
__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():
|
BlackListCodeManager.clear()
|
BlackListCodeManager.add_code("000000")
|
|
|
# 移除禁止交易代码
|
def remove_from_forbidden_trade_codes(code):
|
BlackListCodeManager.remove_code(code)
|
|
|
# 添加代码到禁止交易
|
def add_to_forbidden_trade_codes(code):
|
BlackListCodeManager.add_code(code)
|
|
|
# 禁止代码交易
|
def forbidden_trade(code):
|
add_to_forbidden_trade_codes(code)
|
# l2_data_manager.remove_from_l2_fixed_codes(code)
|
# l2_code_operate.L2CodeOperate.get_instance().remove_l2_listen(code, "禁止代码交易")
|
|
|
def is_in_forbidden_trade_codes(code):
|
return BlackListCodeManager.is_in(code)
|
|
|
if __name__ == "__main__":
|
# add_to_forbidden_trade_codes("000977")
|
WhiteListCodeManager.add_code("002977")
|
