This commit is contained in:
2024-07-03 08:11:08 +03:00
parent 7ba3426989
commit c9ddfaf8b4
17 changed files with 751 additions and 42 deletions

View File

@@ -3,11 +3,17 @@ import os
from dotenv import load_dotenv, find_dotenv
load_dotenv(find_dotenv())
# Database
PG_LOGIN = os.environ.get('PG_LOGIN')
PG_PASSWORD = os.environ.get('PG_PASSWORD')
PG_PORT = os.environ.get('PG_PORT')
PG_HOST = os.environ.get('PG_HOST')
PG_DATABASE = os.environ.get('PG_DATABASE')
# Celery
CELERY_BROKER_URL = os.environ.get('CELERY_BROKER_URL')
CELERY_RESULT_BACKEND = os.environ.get('CELERY_RESULT_BACKEND')
# Yandex
YANDEX_CLIENT_ID = os.environ.get('YANDEX_CLIENT_ID')

View File

@@ -15,8 +15,16 @@ database_url = (
f"{PG_LOGIN}:{PG_PASSWORD}@"
f"{PG_HOST}:{PG_PORT}/{PG_DATABASE}"
)
engine = create_async_engine(database_url)
session_factory = async_sessionmaker(engine, expire_on_commit=False, autoflush=False, autocommit=False)
engine = create_async_engine(database_url,
pool_size=20,
max_overflow=10
)
session_factory = async_sessionmaker(
engine,
expire_on_commit=False,
autoflush=False,
autocommit=False
)
async def get_session() -> AsyncGenerator[AsyncSession, None]: