27 lines
790 B
Python
27 lines
790 B
Python
from motor.motor_asyncio import AsyncIOMotorClient
|
|
from sqlalchemy import NullPool
|
|
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
|
|
from sqlalchemy.orm import DeclarativeBase, sessionmaker
|
|
|
|
from app.config import settings
|
|
|
|
DATABASE_URL = f"""postgresql+asyncpg://{settings.DB_USER}:
|
|
{settings.DB_PASS}@{settings.DB_HOST}:
|
|
{settings.DB_PORT}/{settings.DB_NAME}"""
|
|
DATABASE_PARAMS = {"pool_size": 50, "pool_pre_ping": True}
|
|
|
|
engine = create_async_engine(DATABASE_URL, **DATABASE_PARAMS)
|
|
|
|
async_session_maker = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
|
|
|
|
|
|
class Base(DeclarativeBase):
|
|
pass
|
|
|
|
|
|
MONGO_URL = f"mongodb://{settings.MONGO_HOST}:{settings.MONGO_PORT}"
|
|
|
|
mongo_client = AsyncIOMotorClient(MONGO_URL)
|
|
|
|
mongo_db = mongo_client.test_db
|
|
|