33 lines
1 KiB
Python
33 lines
1 KiB
Python
from typing import Annotated, Any
|
|
from fastapi import Depends
|
|
from sqlalchemy.ext.asyncio import async_sessionmaker
|
|
from ..db import get_session
|
|
from ..auth import get_current_user, authorize_api_key
|
|
from ..domain.user import User
|
|
|
|
|
|
AsyncSession = Annotated[async_sessionmaker, Depends(get_session)]
|
|
UserDependency = Annotated[User, Depends(get_current_user)]
|
|
ApiKeyDependency = Annotated[bool, Depends(authorize_api_key)]
|
|
|
|
|
|
class BaseController:
|
|
def __init__(self, session: AsyncSession) -> None:
|
|
self.async_session = session
|
|
|
|
async def call(self, *args, **kwargs) -> Any:
|
|
raise NotImplementedError
|
|
|
|
async def __call__(self, *args, **kwargs) -> Any:
|
|
return await self.call(*args, **kwargs)
|
|
|
|
|
|
class AuthorizedController(BaseController):
|
|
def __init__(self, session: AsyncSession, user: UserDependency) -> None:
|
|
super().__init__(session)
|
|
self.user = user
|
|
|
|
|
|
class TasksSessionController(BaseController):
|
|
def __init__(self, session: AsyncSession, api_key: ApiKeyDependency) -> None:
|
|
super().__init__(session)
|