from outbo.apps.core.models import SoftDeletableManager
from outbo.apps.products.enums import FileType


class DraftFileManager(SoftDeletableManager):
    def get_queryset(self):
        return super().get_queryset().filter(file_type=FileType.DRAFT)


class FirstFinalFileManager(SoftDeletableManager):
    def get_queryset(self):
        return super().get_queryset().filter(file_type=FileType.FIRST_FINAL)


class PrintingFileManager(SoftDeletableManager):
    def get_queryset(self):
        return super().get_queryset().filter(file_type=FileType.PRINTING_FILE)