"""
ЛЕГКИЙ РЕЖИМ БАЛАНСА
Для новичков
"""

from game_balance.test_balance.standard_mode import StandardBalance


class EasyBalance(StandardBalance):
    """Облегченный режим"""

    @classmethod
    def get_assets_config(cls):
        config = super().get_assets_config()

        # Делаем активы дешевле и стабильнее
        for asset_id, asset_data in config.items():
            if asset_data.get('base_price'):
                asset_data['base_price'] *= 0.5
            if asset_data.get('volatility'):
                asset_data['volatility'] *= 0.7
            if asset_data.get('income_per_month'):
                asset_data['income_per_month'] *= 1.3

        return config

    @classmethod
    def get_players_config(cls):
        config = super().get_players_config()
        config['STARTING_CAPITAL'] = 200000
        config['TARGET_CAPITAL'] = 25000000

        # Более мягкие ограничения
        config['MAX_ASSETS_PER_TYPE'] = {
            'bonds': None,
            'stocks': 0.6,
            'real_estate': 0.7,
            'crypto': None,
            'commodities': 0.5,
            'business': 0.6,
            'unique': 1.0,
        }

        # Быстрое снятие лимитов
        config['PURCHASE_LIMITS_BY_MONTH'] = {
            1: 0.3,
            2: 0.6,
            3: 1.0,
        }

        return config

    @classmethod
    def get_economy_config(cls):
        config = super().get_economy_config()

        # Меньше налогов
        config['TAX_SYSTEM']['income_tax'] = [
            {'threshold': 0, 'rate': 0.00},
            {'threshold': 100000, 'rate': 0.05},
            {'threshold': 500000, 'rate': 0.10},
            {'threshold': 2000000, 'rate': 0.15},
            {'threshold': 5000000, 'rate': 0.20},
        ]
        config['TAX_SYSTEM']['wealth_tax']['threshold'] = 10000000

        # Ниже ставки по кредитам
        config['LOAN_SYSTEM']['interest_rates']['standard'] = 0.03

        # Меньше инфляции
        config['MACROECONOMICS']['base_inflation'] = 0.002

        return config