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 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)
|
||||||
|
|||||||
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user