# 市场行情等级管理
from db import redis_manager_delegate as redis_manager
from db.redis_manager_delegate import RedisUtils
from utils import tool


class MarketSituationManager:
    # 差
    SITUATION_BAD = 0
    # 一般
    SITUATION_COMMON = 1
    # 好
    SITUATION_GOOD = 2

    __db = 2
    redisManager = redis_manager.RedisManager(2)
    __situation_cache = SITUATION_BAD
    __instance = None

    def __new__(cls, *args, **kwargs):
        if not cls.__instance:
            cls.__instance = super(MarketSituationManager, cls).__new__(cls, *args, **kwargs)
            # 初始化设置
            # 获取交易窗口的锁
            cls.__instance.__situation_cache = cls.get_situation()
        return cls.__instance

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

    def set_situation(self, situation):
        if situation not in [self.SITUATION_BAD, self.SITUATION_COMMON, self.SITUATION_GOOD]:
            raise Exception("situation参数值错误")
        self.__situation_cache = situation
        RedisUtils.setex_async(self.__db, "market_situation", tool.get_expire(), situation)

    # 是否可以下单
    @classmethod
    def get_situation(cls):
        # 默认设置为可交易
        val = RedisUtils.get(cls.__get_redis(), "market_situation")
        if val is None:
            return cls.SITUATION_BAD
        return int(val)

    def get_situation_cache(self):
        return self.__situation_cache


class TradeBlockBuyModeManager:
    # 第一位:保留   第二位：保留  第三位：买独苗  第四位：传统
    # 0bxxxx
    # 传统模式
    MODE_TRADITIONAL = 0b0001
    # 独苗模式
    MODE_UNIQUE_BLOCK = 0b0010

    __REDIS_KEY = "trade_block_buy_mode"
    __mode = int(MODE_TRADITIONAL)

    """
    板块买入模式管理
    """
    __instance = None
    __db = 2
    redisManager = redis_manager.RedisManager(2)

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

    @classmethod
    def __load_data(cls):
        val = cls.__get_redis().get(cls.__REDIS_KEY)
        if val is not None:
            cls.__mode = int(val)

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

    def __set_mode(self, mode: int):
        self.__mode = mode
        RedisUtils.setex_async(self.__db, self.__REDIS_KEY, tool.get_expire(), mode)

    def get_mode(self):
        return self.__mode

    def add_unique_block(self):
        mode = int(self.__mode | self.MODE_UNIQUE_BLOCK)
        self.__set_mode(mode)

    def remove_unique_block(self):
        mode = int(self.__mode & ~self.MODE_UNIQUE_BLOCK)
        self.__set_mode(mode)

    def can_buy_unique_block(self):
        r = self.__mode & self.MODE_UNIQUE_BLOCK > 1
        return r > 0