link_shortener/app/database.py

20 lines
600 B
Python

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 = {"poolclass": NullPool}
engine = create_async_engine(DATABASE_URL, **DATABASE_PARAMS)
async_session_maker = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
class Base(DeclarativeBase):
pass