add dep user in db, fix typecheck

This commit is contained in:
IluaAir
2025-06-22 12:01:04 +03:00
parent 1951260788
commit 05121b3d06
2 changed files with 10 additions and 4 deletions

View File

@@ -1,14 +1,20 @@
from typing import Annotated from typing import Annotated, AsyncGenerator
from fastapi import Depends from fastapi import Depends
from fastapi_users_db_sqlalchemy import SQLAlchemyUserDatabase
from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy.ext.asyncio import AsyncSession
from src.db.database import async_session_maker from src.db.database import async_session_maker
from src.models import UsersORM
async def get_db(): async def get_db() -> AsyncGenerator[AsyncSession, None]:
async with async_session_maker as db: async with async_session_maker as db:
yield db yield db
DBDep = Annotated[AsyncSession, Depends(get_db)] DBDep = Annotated[AsyncSession, Depends(get_db)]
async def get_user_db(session: DBDep):
yield SQLAlchemyUserDatabase(session, UsersORM)

View File

@@ -6,14 +6,14 @@ from sqlalchemy.orm import Mapped, mapped_column, relationship
from src.db.database import Base from src.db.database import Base
if TYPE_CHECKING:
from src.models.users import UsersORM
status_enum = Enum("open", "closed", "in_progress", "todo", name="status_enum") status_enum = Enum("open", "closed", "in_progress", "todo", name="status_enum")
priority_enum = Enum("low", "medium", "high", "critical", name="priority_enum") priority_enum = Enum("low", "medium", "high", "critical", name="priority_enum")
class TasksORM(Base): class TasksORM(Base):
if TYPE_CHECKING:
from src.models.users import UsersORM
__tablename__ = "tasks" __tablename__ = "tasks"
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True) id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)