Django: Validate file type of uploaded file

by

Validating files is a common challenge, so I would like to use a validator:

import magic

from django.utils.deconstruct import deconstructible
from django.template.defaultfilters import filesizeformat


@deconstructible
class FileValidator(object):
    error_messages = {
     'max_size': ("Ensure this file size is not greater than %(max_size)s."
                  " Your file size is %(size)s."),
     'min_size': ("Ensure this file size is not less than %(min_size)s. "
                  "Your file size is %(size)s."),
     'content_type': "Files of type %(content_type)s are not supported.",
    }

    def __init__(self, max_size=None, min_size=None, content_types=()):
        self.max_size = max_size
        self.min_size = min_size
        self.content_types = content_types

    def __call__(self, data):
        if self.max_size is not None and data.size > self.max_size:
            params = {
                'max_size': filesizeformat(self.max_size), 
                'size': filesizeformat(data.size),
            }
            raise ValidationError(self.error_messages['max_size'],
                                   'max_size', params)

        if self.min_size is not None and data.size < self.min_size:
            params = {
                'min_size': filesizeformat(self.min_size),
                'size': filesizeformat(data.size)
            }
            raise ValidationError(self.error_messages['min_size'], 
                                   'min_size', params)

        if self.content_types:
            content_type = magic.from_buffer(data.read(), mime=True)
            data.seek(0)

            if content_type not in self.content_types:
                params = { 'content_type': content_type }
                raise ValidationError(self.error_messages['content_type'],
                                   'content_type', params)

    def __eq__(self, other):
        return (
            isinstance(other, FileValidator) and
            self.max_size == other.max_size and
            self.min_size == other.min_size and
            self.content_types == other.content_types
        )

Then you can use FileValidator in your models.FileField or forms.FileField as follows:

validate_file = FileValidator(max_size=1024 * 100, 
                             content_types=('application/xml',))
file = models.FileField(upload_to=settings.XML_ROOT, 
                        validators=[validate_file])

More Related Contents:

  1. When saving, how can you check if a field has changed?
  2. Django: How can I protect against concurrent modification of database entries
  3. How to express a One-To-Many relationship in Django?
  4. Django: Populate user ID when saving a model
  5. Get model’s fields in Django
  6. Can django’s auth_user.username be varchar(75)? How could that be done?
  7. Override default queryset in Django admin
  8. Setting the selected value on a Django forms.ChoiceField
  9. Custom QuerySet and Manager without breaking DRY?
  10. Django: Query using contains each value in a list
  11. Django: Set foreign key using integer?
  12. How can I have two foreign keys to the same model in Django?
  13. How do you serialize a model instance in Django?
  14. Django migrate –fake and –fake-initial explained
  15. How to make email field unique in model User from contrib.auth in Django
  16. Select DISTINCT individual columns in django?
  17. django – get() returned more than one topic
  18. Django ModelForm override widget
  19. Django – how to specify a database for a model?
  20. “Post Image data using POSTMAN”
  21. How to add data into ManyToMany field?
  22. Raw SQL queries in Django views
  23. Django model: delete() not triggered
  24. Django – filtering on foreign key properties
  25. How to register users in Django REST framework?
  26. Django unique_together with nullable ForeignKey
  27. UnicodeEncodeError: ‘ascii’ codec can’t encode character
  28. django most efficient way to count same field values in a query
  29. Sharing models between Django apps
  30. Django: dynamic database file

Leave a Comment