mbs-panel/settings.py

import config
import legal

PRICE_ENV_KEYS = {"7d": "PRICE_7D", "1m": "PRICE_1M", "3m": "PRICE_3M", "6m": "PRICE_6M", "1y": "PRICE_1Y"}


def _bool(raw: str, default: bool) -> bool:
    if raw is None or raw == "":
        return default
    return raw.strip().lower() == "true"


def _positive_int(raw: str, default: int) -> int:
    if raw and raw.strip().lstrip("-").isdigit():
        parsed = int(raw)
        if parsed >= 0:
            return parsed
    return default


def get_plans() -> list:
    raw = legal.read_env_vars(list(PRICE_ENV_KEYS.values()))
    plans = []
    for p in config.PLANS:
        env_key = PRICE_ENV_KEYS[p["code"]]
        plans.append({
            "code": p["code"],
            "label": p["label"],
            "days": p["days"],
            "price": _positive_int(raw.get(env_key), p["price"]),
        })
    return plans


def get_plans_by_code() -> dict:
    return {p["code"]: p for p in get_plans()}


def set_plan_prices(prices: dict):
    for code, price in prices.items():
        if code in PRICE_ENV_KEYS:
            legal.update_env_var(PRICE_ENV_KEYS[code], str(int(price)))


def get_payment_settings() -> dict:
    raw = legal.read_env_vars(["PAYMENTS_ENABLED", "YOOKASSA_ENABLED", "PLATEGA_ENABLED"])
    return {
        "payments_enabled": _bool(raw.get("PAYMENTS_ENABLED"), config.PAYMENTS_ENABLED),
        "yookassa_enabled": _bool(raw.get("YOOKASSA_ENABLED"), config.YOOKASSA_ENABLED),
        "platega_enabled": _bool(raw.get("PLATEGA_ENABLED"), config.PLATEGA_ENABLED),
    }


def yookassa_credentials():
    raw = legal.read_env_vars(["YOOKASSA_SHOP_ID", "YOOKASSA_SECRET_KEY"])
    return (
        raw.get("YOOKASSA_SHOP_ID") or config.YOOKASSA_SHOP_ID,
        raw.get("YOOKASSA_SECRET_KEY") or config.YOOKASSA_SECRET_KEY,
    )


def platega_credentials():
    raw = legal.read_env_vars(["PLATEGA_MERCHANT_ID", "PLATEGA_SECRET"])
    return (
        raw.get("PLATEGA_MERCHANT_ID") or config.PLATEGA_MERCHANT_ID,
        raw.get("PLATEGA_SECRET") or config.PLATEGA_SECRET,
    )


def get_brand_name() -> str:
    raw = legal.read_env_var("BRAND_NAME", "")
    return raw.strip() if raw.strip() else config.BRAND_NAME


def bot_credentials():
    raw = legal.read_env_vars(["BOT_TOKEN", "BOT_USERNAME"])
    return (
        raw.get("BOT_TOKEN") or config.BOT_TOKEN,
        raw.get("BOT_USERNAME") or config.BOT_USERNAME,
    )


def get_hwid_settings() -> dict:
    raw = legal.read_env_vars(["HWID_LIMIT_ENABLED", "HWID_FALLBACK_LIMIT"])
    limit = _positive_int(raw.get("HWID_FALLBACK_LIMIT"), config.HWID_FALLBACK_LIMIT)
    return {
        "enabled": _bool(raw.get("HWID_LIMIT_ENABLED"), config.HWID_LIMIT_ENABLED),
        "fallback_limit": limit if limit > 0 else config.HWID_FALLBACK_LIMIT,
    }


def get_referral_settings() -> dict:
    raw = legal.read_env_vars(["REFERRAL_ENABLED", "REFERRAL_BONUS_DAYS"])
    days = _positive_int(raw.get("REFERRAL_BONUS_DAYS"), config.REFERRAL_BONUS_DAYS)
    return {
        "enabled": _bool(raw.get("REFERRAL_ENABLED"), config.REFERRAL_ENABLED),
        "bonus_days": days if days > 0 else config.REFERRAL_BONUS_DAYS,
    }


def set_referral_settings(enabled: bool, bonus_days: int):
    legal.update_env_var("REFERRAL_ENABLED", "true" if enabled else "false")
    legal.update_env_var("REFERRAL_BONUS_DAYS", str(int(bonus_days)))