from typing import Callable

from starlette.responses import Response

from fastapi_users.db import BaseUserDatabase
from fastapi_users.models import UserDB


class BaseAuthentication:

    userDB: BaseUserDatabase

    def __init__(self, userDB: BaseUserDatabase):
        self.userDB = userDB

    async def get_login_response(self, user: UserDB, response: Response):
        raise NotImplementedError()

    def get_authentication_method(self) -> Callable[..., UserDB]:
        raise NotImplementedError()