Python Repository

Python Repository

from abc import ABC, abstractmethod

class AbstractRepository(ABC):
    def __init__(self, model_class):
        self.model_class = model_class

    @abstractmethod
    def get(self, **kwargs):
        pass

    @abstractmethod
    def filter(self, **kwargs):
        pass

    @abstractmethod
    def create(self, **kwargs):
        pass

    @abstractmethod
    def update(self, instance, **kwargs):
        pass

    @abstractmethod
    def delete(self, instance):
        pass


class BaseRepository(AbstractRepository):
    def get(self, **kwargs):
        return self.model_class.objects.get(**kwargs)

    def filter(self, **kwargs):
        return self.model_class.objects.filter(**kwargs)

    def create(self, **kwargs):
        return self.model_class.objects.create(**kwargs)

    def update(self, instance, **kwargs):
        for attr, value in kwargs.items():
            setattr(instance, attr, value)
        instance.save()
        return instance

    def delete(self, instance):
        instance.delete()


class BookRepository(BaseRepository):
    def __init__(self):
        super().__init__(Book)

    def get_books_published_after(self, year):
        return self.model_class.objects.filter(publication_year__gt=year)