mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-22 15:36:44 +00:00
132 lines
2.9 KiB
Python
132 lines
2.9 KiB
Python
from enum import Enum
|
|
from pathlib import Path
|
|
from typing import (
|
|
List,
|
|
Optional,
|
|
Union,
|
|
)
|
|
|
|
import dotenv
|
|
from pydantic import AnyUrl, BaseModel, Field
|
|
|
|
from utils.const import PROJECT_ROOT
|
|
from utils.models.base import Settings
|
|
from utils.typedefs import NaturalNumber
|
|
|
|
__all__ = ["BotConfig", "config", "JoinGroups"]
|
|
|
|
dotenv.load_dotenv()
|
|
|
|
|
|
class JoinGroups(str, Enum):
|
|
NO_ALLOW = "NO_ALLOW"
|
|
ALLOW_AUTH_USER = "ALLOW_AUTH_USER"
|
|
ALLOW_ALL = "ALLOW_ALL"
|
|
|
|
|
|
class ConfigChannel(BaseModel):
|
|
name: str
|
|
chat_id: int
|
|
|
|
|
|
class ConfigUser(BaseModel):
|
|
username: Optional[str]
|
|
user_id: int
|
|
|
|
|
|
class MySqlConfig(Settings):
|
|
host: str = "127.0.0.1"
|
|
port: int = 3306
|
|
username: str = None
|
|
password: str = None
|
|
database: str = None
|
|
|
|
class Config(Settings.Config):
|
|
env_prefix = "db_"
|
|
|
|
|
|
class RedisConfig(Settings):
|
|
host: str = "127.0.0.1"
|
|
port: int = 6379
|
|
database: int = Field(default=0, env="redis_db")
|
|
password: Optional[str] = None
|
|
|
|
class Config(Settings.Config):
|
|
env_prefix = "redis_"
|
|
|
|
|
|
class LoggerConfig(Settings):
|
|
name: str = "TGPaimon"
|
|
width: int = 180
|
|
time_format: str = "[%Y-%m-%d %X]"
|
|
traceback_max_frames: int = 20
|
|
path: Path = PROJECT_ROOT / "logs"
|
|
render_keywords: List[str] = ["BOT"]
|
|
locals_max_length: int = 10
|
|
locals_max_string: int = 80
|
|
locals_max_depth: Optional[NaturalNumber] = None
|
|
filtered_names: List[str] = ["uvicorn"]
|
|
|
|
class Config(Settings.Config):
|
|
env_prefix = "logger_"
|
|
|
|
|
|
class MTProtoConfig(Settings):
|
|
api_id: Optional[int] = None
|
|
api_hash: Optional[str] = None
|
|
|
|
|
|
class WebServerConfig(Settings):
|
|
url: AnyUrl = "http://localhost:8080"
|
|
host: str = "localhost"
|
|
port: int = 8080
|
|
|
|
class Config(Settings.Config):
|
|
env_prefix = "web_"
|
|
|
|
|
|
class ErrorConfig(Settings):
|
|
pb_url: str = ""
|
|
pb_sunset: int = 43200
|
|
pb_max_lines: int = 1000
|
|
sentry_dsn: str = ""
|
|
notification_chat_id: Optional[str] = None
|
|
|
|
class Config(Settings.Config):
|
|
env_prefix = "error_"
|
|
|
|
|
|
class BotConfig(Settings):
|
|
debug: bool = False
|
|
|
|
bot_token: str = ""
|
|
|
|
channels: List["ConfigChannel"] = []
|
|
admins: List["ConfigUser"] = []
|
|
verify_groups: List[Union[int, str]] = []
|
|
join_groups: Optional[JoinGroups] = JoinGroups.NO_ALLOW
|
|
|
|
timeout: int = 10
|
|
read_timeout: float = 2
|
|
write_timeout: Optional[float] = None
|
|
connect_timeout: Optional[float] = None
|
|
pool_timeout: Optional[float] = None
|
|
|
|
genshin_ttl: Optional[int] = None
|
|
|
|
enka_network_api_agent: str = ""
|
|
pass_challenge_api: str = ""
|
|
pass_challenge_app_key: str = ""
|
|
pass_challenge_user_web: str = ""
|
|
|
|
mysql: MySqlConfig = MySqlConfig()
|
|
logger: LoggerConfig = LoggerConfig()
|
|
webserver: WebServerConfig = WebServerConfig()
|
|
redis: RedisConfig = RedisConfig()
|
|
mtproto: MTProtoConfig = MTProtoConfig()
|
|
error: ErrorConfig = ErrorConfig()
|
|
|
|
|
|
BotConfig.update_forward_refs()
|
|
config = BotConfig()
|