flask-demo-api/flask_demo_api/protocols/repository.py

from abc import abstractmethod
from typing import Protocol

from protocols.models import KeyDTO


class Repository(Protocol):
    @abstractmethod
    def get_key(self, obj: KeyDTO) -> KeyDTO | None:
        raise NotImplementedError

    @abstractmethod
    def add_key(self, obj: KeyDTO) -> KeyDTO:
        raise NotImplementedError

    @abstractmethod
    def put_key(self, obj: KeyDTO) -> KeyDTO:
        raise NotImplementedError