Source code for medmij_oauth.server.data_store

import uuid
import datetime
from abc import ABC, abstractmethod

class DataStore(ABC):
    """Abstract Class that handles interaction instantiation, persisting and lookups of OAuthSessions."""
    @abstractmethod
    async def create_oauth_session(self, response_type, client_id, redirect_uri, scope, state, **kwargs):
        """Create a new oauth_session, persist the oauth_session and return it."""

    @abstractmethod
    async def get_oauth_session_by_id(self, oauth_session_id, **kwargs):
        """Get a oauth_session based on its id and return it, else return None"""

    @abstractmethod
    async def get_oauth_session_by_authorization_code(self, authorization_code, **kwargs):
        """Get a oauth_session based on its authorization_code and return it, else return None"""

    @abstractmethod
    async def save_oauth_session(self, oauth_session, **kwargs):
        """Persist the current state of the oauth_session and return it"""