add dep user in db, fix typecheck
This commit is contained in:
@@ -1,14 +1,20 @@
|
||||
from typing import Annotated
|
||||
from typing import Annotated, AsyncGenerator
|
||||
|
||||
from fastapi import Depends
|
||||
from fastapi_users_db_sqlalchemy import SQLAlchemyUserDatabase
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
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:
|
||||
yield db
|
||||
|
||||
|
||||
DBDep = Annotated[AsyncSession, Depends(get_db)]
|
||||
|
||||
|
||||
async def get_user_db(session: DBDep):
|
||||
yield SQLAlchemyUserDatabase(session, UsersORM)
|
||||
|
||||
@@ -6,14 +6,14 @@ from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
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")
|
||||
priority_enum = Enum("low", "medium", "high", "critical", name="priority_enum")
|
||||
|
||||
|
||||
class TasksORM(Base):
|
||||
if TYPE_CHECKING:
|
||||
from src.models.users import UsersORM
|
||||
|
||||
__tablename__ = "tasks"
|
||||
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
||||
|
||||
Reference in New Issue
Block a user