svthalia/concrexit

from django.core.exceptions import ValidationError
from django.utils.deconstruct import deconstructible
from django.utils.translation import gettext_lazy as _

import puremagic as magic


@deconstructible
class ArchiveFileTypeValidator:
    """Validator class for archive files."""

    types = ["application/gzip", "application/zip", "application/x-gzip"]
    message = _("Only zip and tar files are allowed.")

    def __init__(self, types=None, message=None):
        """Initialize ArchiveFileTypeValidator with allowed types and warning message."""
        if types is not None:
            self.types = types
        if message is not None:
            self.message = message

    def __call__(self, value):
        """Validate that the input contains (or does *not* contain, if inverse_match is True) a match for the regular expression."""
        if (
            magic.from_stream(
                value.temporary_upload.file,
                mime=True,
                filename=value.temporary_upload.upload_name,
            )
            not in self.types
        ):
            raise ValidationError(self.message)

    def __eq__(self, other):
        """Check whether the current object and other are equal."""
        return (
            isinstance(other, ArchiveFileTypeValidator)
            and self.types == other.types
            and (self.message == other.message)
        )