"""
买入条件工具
"""
# 获取自由流通市值限制
# 返回:（最大买入,最优自由流通最小,最优自由流通最大）
from settings.trade_setting import MarketSituationManager

# 获取自由流通市值的阈值范围
# （最小流通,最大流通,优秀开始,优秀结束,最优开始,最优结束,最大可买）
from utils import tool


def get_zyltgb_threshold(market_sitation: int):
    if market_sitation == MarketSituationManager.SITUATION_GOOD:
        return 31, 100, 40, 100, 40, 80, 100
    # return 5.9, 41, 8.9, 25, 8.9, 19, 80
    return 5.9, 10000, 8.9, 25, 8.9, 19, 10000


# 获取量比的等级获取量
def get_volume_rate_by_level(level: int):
    volume_rates = [0, 0.19, 0.49, 0.89]
    return volume_rates[level]


def get_cancel_and_buy_space_time(code):
    return 50 if tool.is_sz_code(code) else 250