33 lines
1,018 B
Python
33 lines
1,018 B
Python
from typing import Annotated, Any
|
|
|
|
from fastapi import Depends
|
|
from sqlalchemy.ext.asyncio import async_sessionmaker
|
|
|
|
from ..auth import authorize_api_key, get_current_user
|
|
from ..db import get_db_session, AsyncSession
|
|
from ..domain.user import User
|
|
|
|
UserDependency = Annotated[User, Depends(get_current_user)]
|
|
ApiKeyDependency = Annotated[None, Depends(authorize_api_key)]
|
|
|
|
|
|
class BaseController:
|
|
def __init__(self, session: AsyncSession) -> None:
|
|
self.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)
|