Commit 8ae87265 authored by Mickaël Desfrênes's avatar Mickaël Desfrênes
Browse files

change download name for files

parent 0546606c
Loading
Loading
Loading
Loading
+13 −1
Original line number Diff line number Diff line
import datetime

from django.utils.text import slugify
from django.db import models, connection
from django.contrib.auth.models import User, Group
import os
@@ -368,6 +368,18 @@ class File(Resource):
        self.ptr_project = self.project
        super(File, self).save(*args, **kwargs)

    @property
    def new_filename(self) -> str:
        """
        Get new filename computed from title.
        Tries to keep original extension.
        """
        _, extension = os.path.splitext(self.original_name)
        if not extension:
            extension = "." + self.file_type.extensions.first().label
        slugged_title = slugify(self.title)
        return slugged_title + extension


class Collection(models.Model):
    title = models.CharField("titre", max_length=256)
+1 −1
Original line number Diff line number Diff line
@@ -333,6 +333,6 @@ def force_download(
    return RangedFileResponse(
        request,
        open(file.local_path(), "rb"),
        filename=file.original_name,
        filename=file.new_filename,
        content_type=file.file_type.mime,
    )