from pathlib import Path from pydantic import model_validator from pydantic_settings import BaseSettings, SettingsConfigDict from sqlalchemy.engine.url import make_url class Settings(BaseSettings): model_config = SettingsConfigDict(env_file=".env", env_file_encoding="utf-8", extra="ignore") max_token: str max_device_id: str tg_bot_token: str tg_forum_channel_id: int fallback_user_id: int database_url: str = "sqlite+aiosqlite:///data/bridge.db" redis_url: str = "redis://redis:6379/0" tg_rate_limit_delay_sec: float = 3.5 max_rate_limit_delay_sec: float = 1.0 ls_topic_prefix: str = "👤 " max_reconnect_fetch_limit: int = 50 log_level: str = "INFO" data_dir: str = "" max_session_name: str = "max_session.db" @property def sqlite_path(self) -> str: db = make_url(self.database_url).database if not db: return str(Path("data/bridge.db").resolve()) path = Path(db) return str(path if path.is_absolute() else path.resolve()) @model_validator(mode="after") def _default_data_dir(self) -> "Settings": if not self.data_dir: object.__setattr__(self, "data_dir", str(Path(self.sqlite_path).parent)) return self