Как изменять имя файла при загрузки в Django?

Question

[Дмитрий Иванов]

Как изменить имя файла при загрузки в Django на свое, желательно на UUID?

Answer 1

[Сергей Горностаев]

from os.path import splitext
from uuid import uuid4
from django.core.files.storage import FileSystemStorage

class UUIDFileStorage(FileSystemStorage):
    def get_available_name(self, name, max_length=None):
        _, ext = path.splitext(name)
        return uuid4().hex + ext

class SomeModel(models.Model):
    some_filed = models.ImageField('Какое-то изображение', upload_to='someimages', storage=UUIDFileStorage())

Comments

[Дмитрий Иванов]

Большое спасибо. Но если несложно как сохранить вместе с расширением файла и сохранением возможности установки upload_to у FileField.

[Сергей Горностаев]

Дмитрий Иванов, дополнил ответ.

[un1t]

Сергей Горностаев, на самом деле не нужно определять свой сторадж. Параметр upload_to может принимать функцию в качестве аргумента. В ней делаете имя какое нужно.

Пример из документации

def user_directory_path(instance, filename):
    # file will be uploaded to MEDIA_ROOT/user_<id>/<filename>
    return 'user_{0}/{1}'.format(instance.user.id, filename)

class MyModel(models.Model):
    upload = models.FileField(upload_to=user_directory_path)

https://docs.djangoproject.com/en/2.0/ref/models/f...

[neatsoft]

Совет по поводу этой строки:
_, ext = path.splitext(name)

Underscore в Django традиционно используется для gettext или gettext_lazy, поэтому для ненужных аргументов лучше использовать что-нибудь другое, например, двойное подчёркивание (__).

А в данном случае проще написать так:
ext = os.path.splitext(filename)[1].lower()

[Сергей Горностаев]

neatsoft, обращение по индексу может привести к исключению, если у загружаемого файла не будет расширения.

[neatsoft]

Сергей Горностаев, проверил, всё должно быть ок:

>>> import os
>>> os.path.splitext('file.ext')[1].lower()
'.ext'
>>> os.path.splitext('file')[1].lower()
''

Это с обычным split может быть такая проблема:

>>> filename, ext = 'file.ext'.split('.')
>>> print(filename, ext)
file ext
>>> filename, ext = 'file'.split('.')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: not enough values to unpack (expected 2, got 1)
>>> filename, ext = 'file.ext.ext'.split('.')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ValueError: too many values to unpack (expected 2)

Answer 2

[neatsoft]

Для именования файлов лучше использовать хэш, а не uuid. Это гарантирует уникальность имён, и обеспечивает дедупликацию данных: сколько бы раз пользователи не заливали файл с одинаковым содержимым, он будет храниться в единственном экземпляре (исключение - файлы с одинаковым содержимым и разными расширениями, но это разумный компромисс - работать с файлами без расширений менее удобно на многих этапах).

Можно реализовать соответствующую функциональность самостоятельно:

app_name/models.py:

import hashlib
import os

from django.conf import settings
from django.core.files.storage import FileSystemStorage
from django.db import models


def upload_to(instance, filename, fieldname):
    ext = os.path.splitext(filename)[1].lower()
    class_name = instance.__class__.__name__.lower()

    h = hashlib.sha256()
    field = getattr(instance, fieldname)
    for chunk in field.chunks():
        h.update(chunk)
    name = h.hexdigest()

    return os.path.join(
        class_name,
        name + ext,
    )


class OverwriteStorage(FileSystemStorage):
    def get_available_name(self, name, max_length=None):
        if self.exists(name):
            os.remove(os.path.join(settings.MEDIA_ROOT, name))
        return name


class Article(models.Model):
    image = models.ImageField(
        storage=OverwriteStorage(),
        upload_to=lambda inst, fn: upload_to(inst, fn, 'image'),
    )

Но гораздо разумнее воспользоваться готовой библиотекой:

Установить hashedfilenamestorage:
pip install django-hashedfilenamestorage

и добавить следующую строку в settings.py:

DEFAULT_FILE_STORAGE = 'django_hashedfilenamestorage.storage.HashedFilenameFileSystemStorage'

Comments

[neatsoft]

Посмотрел как storage реализован в hashedfilenamestorage. В собственной реализации лучше не get_available_name переопределять, а save, и если файл существует, то просто скипать стадию сохранения.

Если не заменять стандартный storage, то при повторной записи файла с идентичным содержимым джанга добавит к имени файла последовательность случайных символов.