mirror of
https://github.com/MrSedan/neuro-reply-bot-reworked.git
synced 2026-01-14 13:39:42 +03:00
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
import os
|
|
import tomllib
|
|
from typing import List, Optional
|
|
|
|
from attr import dataclass
|
|
from dotenv import load_dotenv
|
|
|
|
from neuroapi.types import Singleton
|
|
|
|
from .types._helpers import *
|
|
|
|
|
|
@dataclass
|
|
class Settings:
|
|
time: List[str]
|
|
|
|
@staticmethod
|
|
def from_dict(obj: Any) -> 'Settings':
|
|
assert isinstance(obj, dict)
|
|
time = from_list(from_str, obj.get("time", []))
|
|
return Settings(time)
|
|
|
|
def to_dict(self) -> dict:
|
|
result: dict = {}
|
|
result['time'] = from_list(from_str, self.time)
|
|
return result
|
|
|
|
class Config(Singleton):
|
|
api_url: str
|
|
settings: Settings
|
|
token: Optional[str]
|
|
proxy_token: Optional[str]
|
|
def __init__(self):
|
|
load_dotenv(os.path.join(os.path.dirname(__file__), '..', '.env'))
|
|
if not os.path.exists(os.path.join(os.path.dirname(__file__), '..', 'settings.toml')): raise Exception('Settings.toml must be in root folder')
|
|
with open(os.path.join(os.path.dirname(__file__), '..', 'settings.toml'), 'rb') as f:
|
|
settings = tomllib.load(f)
|
|
self.settings = Settings.from_dict(settings)
|
|
self.api_url = os.environ.get('API_URL')
|
|
self.token = os.environ.get('TOKEN')
|
|
if self.token == '':
|
|
self.token = None
|
|
self.proxy_token = os.environ.get('PROXY_TOKEN')
|
|
if self.proxy_token == '':
|
|
self.proxy_token = None |