"""
L2大单管理器
"""

# 大单管理
import redis_manager
import tool

__redisManager = redis_manager.RedisManager(0)


def add_num(code, num):
    redis = __redisManager.getRedis()
    redis.incrby("big_money-{}".format(code), num)


# 设置过期时间
def expire(code):
    redis = __redisManager.getRedis()
    redis.expire("big_money-{}".format(code), tool.get_expire())


def reset(code):
    redis = __redisManager.getRedis()
    redis.set("big_money-{}".format(code), 0)


def get_num(code):
    redis = __redisManager.getRedis()
    num = redis.get("big_money-{}".format(code))
    if num is None:
        return 0
    return round(int(num)/1000/4)


if __name__ == "__main__":
    add_num("000332", 0)
    expire("000332")