brinza/telebot_template
Template
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

move pool_recycle and pool_pre_ping in config.py

Browse Source
This commit is contained in:
Ilya Bezrukov 2024-07-30 02:39:47 +03:00
parent f62e05fd06
commit cd5845c180
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mybot/config.py
View File

@ -54,10 +54,14 @@ class StateStorageConfig:
@dataclass @dataclass
class DatabaseConfig: class DatabaseConfig:
url: str url: str
pool_recycle: int
pool_pre_ping: bool
@classmethod @classmethod
def from_env(cls): def from_env(cls):
return cls(os.getenv("DATABASE_URL", "sqlite:///bot.db")) return cls(os.getenv("DB_URL", "sqlite:///bot.db"),
int(os.getenv("DB_POOL_RECYCLE", 3600)),
bool(int(os.getenv("DB_POOL_PRE_PING", True))))
@dataclass @dataclass

4
mybot/database/__init__.py
View File

@ -7,8 +7,8 @@ from .models import User
def get_engine(config: DatabaseConfig): def get_engine(config: DatabaseConfig):
engine = create_engine(config.url, engine = create_engine(config.url,
pool_recycle=3600, pool_recycle=config.pool_recycle,
pool_pre_ping=True) pool_pre_ping=config.pool_pre_ping)
return engine return engine