microblog/app/database.py

36 lines
1.1 KiB
Python
Raw Normal View History

2022-06-22 18:11:22 +00:00
from typing import Any
2022-06-29 18:43:17 +00:00
from typing import AsyncGenerator
2022-06-22 18:11:22 +00:00
from sqlalchemy import MetaData
2022-06-22 18:11:22 +00:00
from sqlalchemy import create_engine
2022-06-29 18:43:17 +00:00
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.ext.asyncio import create_async_engine
2022-06-22 18:11:22 +00:00
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
2022-06-29 18:43:17 +00:00
from app.config import DB_PATH
2022-08-28 09:24:46 +00:00
from app.config import DEBUG
2022-06-22 18:11:22 +00:00
from app.config import SQLALCHEMY_DATABASE_URL
engine = create_engine(
2022-08-14 20:44:30 +00:00
SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False, "timeout": 15}
2022-06-22 18:11:22 +00:00
)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
2022-06-29 18:43:17 +00:00
DATABASE_URL = f"sqlite+aiosqlite:///{DB_PATH}"
2022-08-14 20:44:30 +00:00
async_engine = create_async_engine(
2022-08-28 09:24:46 +00:00
DATABASE_URL, future=True, echo=DEBUG, connect_args={"timeout": 15}
2022-08-14 20:44:30 +00:00
)
2022-06-29 18:43:17 +00:00
async_session = sessionmaker(async_engine, class_=AsyncSession, expire_on_commit=False)
2022-06-22 18:11:22 +00:00
Base: Any = declarative_base()
metadata_obj = MetaData()
2022-06-22 18:11:22 +00:00
2022-06-29 18:43:17 +00:00
async def get_db_session() -> AsyncGenerator[AsyncSession, None]:
async with async_session() as session:
2022-07-14 10:13:23 +00:00
try:
yield session
finally:
await session.close()