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

# 大单管理
from db import redis_manager
from utils import tool

__redisManager = redis_manager.RedisManager(0)


# 是否为大单
def is_big_num(val):
    if int(val["num"]) >= 8000 or int(val["num"]) * float(val["price"]) >= 30000:
        return True
    else:
        return False


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)


def reset_all():
    redis = __redisManager.getRedis()
    keys = redis.keys("big_money-*")
    for k in keys:
        redis.setex(k, tool.get_expire(), 0)


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