"""
L2卖管理
"""

from db import redis_manager_delegate as redis_manager
from db.redis_manager_delegate import RedisUtils
from log_module import async_log_util
from log_module.log import logger_l2_market_sell, logger_debug
from utils import tool
from utils.tool import CodeDataCacheUtil


# 卖行情信息
class L2MarketSellManager:
    __db = 0
    __redis_manager = redis_manager.RedisManager(0)
    __instance = None
    __current_total_sell_data_cache = {}
    __last_total_sell_data_cache = {}
    __used_refer_sell_data_cache = {}

    def __new__(cls, *args, **kwargs):
        if not cls.__instance:
            cls.__instance = super(L2MarketSellManager, cls).__new__(cls, *args, **kwargs)
            cls.__load_datas()
        return cls.__instance

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

    @classmethod
    def __load_datas(cls):
        logger_debug.info("L2MarketSellManager __load_datas")
        __redis = cls.__get_redis()
        try:
            keys = RedisUtils.keys(__redis, "fast_buy_used_sell_data-*")
            for k in keys:
                code = k.split("-")[-1]
                val = RedisUtils.smembers(__redis, k)
                CodeDataCacheUtil.set_cache(cls.__used_refer_sell_data_cache, code, val)
        except:
            pass
        finally:
            RedisUtils.realse(__redis)

    def set_sell_time_used(self, code, time_str):
        if code not in self.__used_refer_sell_data_cache:
            self.__used_refer_sell_data_cache[code] = set()
        self.__used_refer_sell_data_cache[code].add(time_str)
        RedisUtils.sadd_async(self.__db, f"fast_buy_used_sell_data-{code}", time_str)
        RedisUtils.expire_async(self.__db, f"fast_buy_used_sell_data-{code}", tool.get_expire())

    def is_refer_sell_time_used(self, code, time_str):
        if code not in self.__used_refer_sell_data_cache:
            return False
        return time_str in self.__used_refer_sell_data_cache[code]

    def clear(self):
        self.__used_refer_sell_data_cache.clear()
        keys = RedisUtils.keys(self.__get_redis(), "fast_buy_used_sell_data-*")
        for k in keys:
            RedisUtils.delete_async(self.__db, k)

    # 设置当前的总卖
    def set_current_total_sell_data(self, code, time_str, money, volume, sell_1_info):
        """
        @param code:
        @param time_str:
        @param money:
        @param volume:
        @param sell_1_info: 格式(卖1价格,卖1量)
        @return:
        """
        # 记录日志
        async_log_util.info(logger_l2_market_sell, f"{code}: {time_str}-{money} {sell_1_info}")
        if code in self.__current_total_sell_data_cache:
            self.__last_total_sell_data_cache[code] = self.__current_total_sell_data_cache.get(code)
        self.__current_total_sell_data_cache[code] = (time_str, round(money), volume, sell_1_info)

    def get_current_total_sell_data(self, code):
        """

        @param code:
        @return:(时间, 总买额, 总量, 卖1信息)
        """
        return self.__current_total_sell_data_cache.get(code)

    # 获取参考卖的数据
    def get_refer_sell_data(self, code, time_str):
        cuurent = self.__current_total_sell_data_cache.get(code)
        if cuurent is None:
            return None
        if int(time_str.replace(":", "")) > int(cuurent[0].replace(":", "")):
            return cuurent
        last = self.__last_total_sell_data_cache.get(code)
        if last and int(time_str.replace(":", "")) > int(last[0].replace(":", "")):
            return last
        return None


# 板上卖统计
class L2LimitUpSellManager:
    __db = 0
    __redis_manager = redis_manager.RedisManager(0)
    __instance = None
    __limit_up_sell_cache = {}

    def __new__(cls, *args, **kwargs):
        if not cls.__instance:
            cls.__instance = super(L2LimitUpSellManager, cls).__new__(cls, *args, **kwargs)
            cls.__load_datas()
        return cls.__instance

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

    @classmethod
    def __load_datas(cls):
        __redis = cls.__get_redis()
        try:
            keys = RedisUtils.keys(__redis, "limit_up_sell_indexes-*")
            for k in keys:
                code = k.split("-")[-1]
                val = RedisUtils.smembers(__redis, k)
                CodeDataCacheUtil.set_cache(cls.__limit_up_sell_cache, code, val)
        except:
            pass
        finally:
            RedisUtils.realse(__redis)

    def add_limit_up_sell(self, code, index):
        if code not in self.__limit_up_sell_cache:
            self.__limit_up_sell_cache[code] = set()
        self.__limit_up_sell_cache[code].add(index)
        RedisUtils.sadd_async(self.__db, f"limit_up_sell_indexes-{code}", index)
        RedisUtils.expire_async(self.__db, f"limit_up_sell_indexes-{code}", tool.get_expire())

    def remove_limit_up_sell(self, code, index):
        if code in self.__limit_up_sell_cache:
            self.__limit_up_sell_cache[code].discard(index)
        RedisUtils.srem_async(self.__db, f"limit_up_sell_indexes-{code}", index)

    def clear(self):
        self.__limit_up_sell_cache.clear()
        keys = RedisUtils.keys(self.__get_redis(), "limit_up_sell_indexes-*")
        for k in keys:
            RedisUtils.delete_async(self.__db, k)

    # 设置当前的总卖
    def get_limit_up_sell_indexes(self, code):
        if code in self.__limit_up_sell_cache:
            return self.__limit_up_sell_cache[code]
        return set()