app/database.py - master - durovcode/backend

from typing import Annotated, AsyncIterator

# import redis.asyncio as aioredis
from fastapi import Depends
from sqlalchemy.ext.asyncio import AsyncEngine, async_sessionmaker, create_async_engine

from app.env import SQLALCHEMY_DB_URI


alchemy_engine: AsyncEngine = create_async_engine(
    url=SQLALCHEMY_DB_URI, pool_pre_ping=True
)


AlchemySessionLocal = async_sessionmaker(
    bind=alchemy_engine, autoflush=False, future=True
)


async def pull_session() -> AsyncIterator[async_sessionmaker]:
    # todo! try except SQLAlchemyError & log
    yield AlchemySessionLocal


AlchemySession = Annotated[async_sessionmaker, Depends(dependency=pull_session)]


# redis = aioredis.Redis()