import hashlib
import six
from django import forms
from django.conf import settings
from django.core.exceptions import ValidationError
from django.core.files.base import ContentFile
from django.core.validators import FileExtensionValidator
from django.forms.models import modelform_factory
from django.utils.translation import gettext_lazy as _
from filer.models import File, Folder, Image
from shuup.admin.utils.permissions import has_permission
from shuup.core.models import MediaFile, MediaFolder
[docs]
def file_size_validator(value):
size = getattr(value, "size", None)
if size and settings.SHUUP_MAX_UPLOAD_SIZE and settings.SHUUP_MAX_UPLOAD_SIZE < size:
raise ValidationError(
_("Maximum file size reached (%(size)s MB).") % {"size": settings.SHUUP_MAX_UPLOAD_SIZE / 1000 / 1000},
code="file_max_size_reached",
)
return value
[docs]
def filer_folder_from_path(path):
"""
Split `path` by slashes and create a hierarchy of Filer Folder objects accordingly.
Blank path components are ignored, so "/////foo//////bar///" is the same as "foo/bar".
The empty string (and `None`) are handled as "no folder", i.e. root folder.
:param path: Pathname or None.
:type path: str|None
:return: Folder.
:rtype: filer.models.Folder
"""
if path is None:
return None
folder = None
for component in six.text_type(path).split("/"):
if component:
folder = Folder.objects.get_or_create(name=component, parent=folder)[0]
return folder
def _filer_file_from_upload(model, request, path, upload_data, sha1=None):
"""
Create some sort of Filer file (either File or Image, really) from the given
upload data (ContentFile or UploadFile).
:param model: Model class.
:param request: Request, to figure out the owner for this file.
:type request: django.http.request.HttpRequest|None
:param path: Pathname string (see `filer_folder_from_path`) or a Filer Folder.
:type path: basestring|filer.models.Folder
:param upload_data: Upload data.
:type upload_data: django.core.files.base.File
:param sha1: SHA1 checksum. If given and a matching `model` with the SHA1 is found, it is returned instead.
:type sha1: basestring
:return: Filer file.
"""
if sha1:
upload = model.objects.filter(sha1=sha1).first()
if upload:
return upload
file_form_cls = modelform_factory(model=model, fields=("original_filename", "owner", "file"))
upload_form = file_form_cls(
data={
"original_filename": upload_data.name,
"owner": (request.user.pk if (request and not request.user.is_anonymous) else None),
},
files={"file": upload_data},
)
upload = upload_form.save(commit=False)
upload.is_public = True
if isinstance(path, Folder):
upload.folder = path
else:
upload.folder = filer_folder_from_path(path)
upload.save()
return upload
[docs]
def filer_file_from_upload(request, path, upload_data, sha1=None):
"""
Create a filer.models.filemodels.File from an upload (UploadedFile or such).
If the `sha1` parameter is passed and a file with said SHA1 is found, it will be returned instead.
:param request: Request, to figure out the owner for this file.
:type request: django.http.request.HttpRequest|None
:param path: Pathname string (see `filer_folder_from_path`) or a Filer Folder.
:type path: basestring|filer.models.Folder
:param upload_data: Upload data.
:type upload_data: django.core.files.base.File
:param sha1: SHA1 checksum. If given and a matching `model` with the SHA1 is found, it is returned instead.
:type sha1: basestring
:rtype: filer.models.filemodels.File
"""
return _filer_file_from_upload(model=File, request=request, path=path, upload_data=upload_data, sha1=sha1)
[docs]
def filer_image_from_upload(request, path, upload_data, sha1=None):
"""
Create a Filer Image from an upload (UploadedFile or such).
If the `sha1` parameter is passed and an Image with said SHA1 is found, it will be returned instead.
:param request: Request, to figure out the owner for this file.
:type request: django.http.request.HttpRequest|None
:param path: Pathname string (see `filer_folder_from_path`) or a Filer Folder.
:type path: basestring|filer.models.Folder
:param upload_data: Upload data.
:type upload_data: django.core.files.base.File
:param sha1: SHA-1 checksum of the data, if available, to do deduplication.
:type sha1: basestring
:rtype: filer.models.imagemodels.Image
"""
return _filer_file_from_upload(model=Image, request=request, path=path, upload_data=upload_data, sha1=sha1)
[docs]
def filer_image_from_data(request, path, file_name, file_data, sha1=None):
"""
Create a Filer Image from the given data string.
If the `sha1` parameter is passed and True (the value True, not a truey value), the SHA-1 of the data string
is calculated and passed to the underlying creation function.
If the `sha1` parameter is truthy (generally the SHA-1 hex string), it's passed directly to the creation function.
:param request: Request, to figure out the owner for this file.
:type request: django.http.request.HttpRequest|None
:param path: Pathname string (see `filer_folder_from_path`) or a Filer Folder.
:type path: basestring|filer.models.Folder
:param file_name: File name.
:type file_data: basestring
:param file_data: Upload data.
:type file_data: bytes
:param sha1: SHA-1 checksum of the data, if available, to do deduplication.
May also be `True` to calculate the SHA-1 first.
:type sha1: basestring|bool
:rtype: filer.models.imagemodels.Image
"""
if sha1 is True:
sha1 = hashlib.sha1(file_data).hexdigest()
upload_data = ContentFile(file_data, file_name)
return _filer_file_from_upload(model=Image, request=request, path=path, upload_data=upload_data, sha1=sha1)
[docs]
def filer_file_to_json_dict(file, user=None):
"""
:type file: filer.models.File
:rtype: dict
"""
assert file.is_public
try:
thumbnail = file.easy_thumbnails_thumbnailer.get_thumbnail(
{
"size": (128, 128),
"crop": False,
"upscale": True,
"background": "#ffffff",
"subject_location": file.subject_location,
}
)
except Exception:
thumbnail = None
extra_permissions = {}
if user:
if has_permission(user, "media.delete-file"):
extra_permissions["delete-file"] = True
if has_permission(user, "media.rename-file"):
extra_permissions["rename-file"] = True
return {
"id": file.id,
"name": file.label,
"size": file.size,
"url": file.url,
"owner": True if user == file.owner else False,
"thumbnail": (thumbnail.url if thumbnail else None),
"date": file.uploaded_at.isoformat(),
**extra_permissions,
}
[docs]
def filer_folder_to_json_dict(folder, children=None, user=None):
"""
:type folder: filer.models.Folder|None
:type children: list(filer.models.Folder)
:rtype: dict
"""
if folder and children is None:
# This allows us to pass `None` as a pseudo root folder
children = folder.get_children()
is_owned = subfolder_of_users_root(user=user, folder=folder)
extra_permissions = {"folder-edit": True if has_permission(user, "media.edit-access") else False}
if user and not is_owned:
if has_permission(user, "media.create-folder"):
extra_permissions["folder-new"] = True
if has_permission(user, "media.rename-folder"):
extra_permissions["folder-rename"] = True
if has_permission(user, "media.delete-folder"):
extra_permissions["folder-delete"] = True
if has_permission(user, "media.upload-to-folder"):
extra_permissions["upload-media"] = True
return {
"id": folder.pk if folder else 0,
"name": folder.name if folder else _("Root"),
"owner": is_owned,
"children": [filer_folder_to_json_dict(child, user=user) for child in children],
"canSeeRoot": can_see_root_folder(user),
**extra_permissions,
}
[docs]
def subfolder_of_users_root(user, folder):
if not folder or not user:
return False
if user.id in list(folder.media_folder.all().values_list("owners", flat=True)):
return True
parents = folder.logical_path
for parent in parents:
if user.id in list(parent.media_folder.all().values_list("owners", flat=True)):
return True
return False
[docs]
def get_or_create_folder(shop, path, user=None):
folders = path.split("/")
parent = None
child = None
for folder in folders:
if folder != "":
child = Folder.objects.get_or_create(parent=parent, name=folder)[0]
ensure_media_folder(shop, child)
parent = child
return child
[docs]
def can_see_root_folder(user):
"""
Return True if the user is allowed to see files that exists in the root folder.
This means all files that have `folder=None`.
"""
return bool(user and has_permission(user, "media.view-all"))