Files
François Voron 53271a32c5 Implement database strategy (#833)
* Implement database strategy

* Export generic AccessToken type

* Apply formatting
2021-12-31 14:43:39 +01:00

39 lines
1.1 KiB
Python

import sys
from datetime import datetime
from typing import Generic, Optional, Type
if sys.version_info < (3, 8):
from typing_extensions import Protocol # pragma: no cover
else:
from typing import Protocol # pragma: no cover
from fastapi_users.authentication.strategy.db.models import A
class AccessTokenDatabase(Protocol, Generic[A]):
"""
Protocol for retrieving, creating and updating access tokens from a database.
:param access_token_model: Pydantic model of an access token.
"""
access_token_model: Type[A]
async def get_by_token(
self, token: str, max_age: Optional[datetime] = None
) -> Optional[A]:
"""Get a single access token by token."""
... # pragma: no cover
async def create(self, access_token: A) -> A:
"""Create an access token."""
... # pragma: no cover
async def update(self, access_token: A) -> A:
"""Update an access token."""
... # pragma: no cover
async def delete(self, access_token: A) -> None:
"""Delete an access token."""
... # pragma: no cover