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


# 获取自由流通市值的阈值范围
# （最小流通,最大流通,优秀开始,优秀结束,最优开始,最优结束,最大可买）
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
    # TODO 大规模测试
    return 5.9, 2000, 8.9, 25, 8.9, 19, 2000


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