Source code for common.fields

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Tue Oct 30 15:32:58 2018

@author: Paolo Cozzi <cozzi@ibba.cnr.it>
"""

from django.db.models.fields.files import FileField, ImageField

from .storage import ProtectedFileSystemStorage


# https://gist.github.com/cobusc/ea1d01611ef05dacb0f33307e292abf4
[docs]class ProtectedFileField(FileField):
[docs]    def __init__(self, *args, **kwargs):
        # override storage parameter
        kwargs["storage"] = ProtectedFileSystemStorage()

        super(ProtectedFileField, self).__init__(
            *args, **kwargs)


[docs]class ProtectedImageField(ImageField):
[docs]    def __init__(self, *args, **kwargs):
        # override storage parameter
        kwargs["storage"] = ProtectedFileSystemStorage()

        super(ProtectedFileField, self).__init__(
            *args, **kwargs)