Django добавляет несколько странных символов в конец файла - PullRequest
Новая
       9

Django добавляет несколько странных символов в конец файла

0 голосов
/ 26 февраля 2019

У меня есть проблем, когда я отправляю файл на сервер django, и имя файла существует в каталоге.Django не переопределяет файл, но добавляет некоторые символы в конце имени.Как я могу добавить символы вместо Django.

Мой model.py выглядит так 

class UpLoadFile(models.Model):
      user = models.ForeignKey(User, on_delete=models.CASCADE)
      file = models.FileField()
      uploaded_at = models.DateTimeField(default=timezone.now)

      def __str__(self):
          return self.file.name + ' (' + self.user.username + ')'

views.py выглядит так: 

@login_required
def home(request):
    if request.method == 'POST':

        if 'upload_file' in request.POST:
            form = UploadFileForm(request.POST, request.FILES)
            form.instance.user = request.user
            if form.is_valid():
                messages.success(request, 'Datoteka {} je bila uspešno dodana'.format(str(request.FILES["file"])))
                form.save()
                return HttpResponseRedirect('/')

        else:
            messages.error(request, 'Datoteke {} ni bilo mogoče dodati. Preveriti če datoteka vsebuje podatke.'
                           .format(str(request.FILES["file"])))
            return HttpResponseRedirect('/')


    elif request.method == 'GET':
        upload_file = UploadFileForm()
        return render(request, 'update_file/home.html', {'page_title': 'Home', 'upload_form': upload_file})

forms.pyвыглядит так: 

class UploadFileForm(forms.ModelForm):
    file = forms.FileField(required=True)

    class Meta:
        model = UpLoadFile
        fields = ('file',)

1 Ответ

0 голосов
/ 28 февраля 2019

Поведение по умолчанию класса Storage Джанго состоит в добавлении серии случайных символов в конец имени файла, когда имя файла уже существует.Если вы хотите контролировать, какие символы добавляются, вы можете создать свой собственный подкласс Storage и переопределить функцию get_available_name().

Очень простой пример, который добавляет фиксированный набор символов в конецимя (вам может потребоваться адаптировать его к различным символам): 

import os
from django.core.files.storage import FileSystemStorage

class MyStorage(FileSystemStorage):

    def get_available_name(self, name, max_length=None):
        if self.exists(name):
            dir_name, file_name = os.path.split(name)
            file_root, file_ext = os.path.splitext(file_name)            

            my_chars = 'abcde'  # The characters you want to append

            name = os.path.join(dir_name, '{}_{}{}'.format(file_root, my_chars, file_ext))
        return name

Затем вы можете указать пользовательский класс хранения для модели: 

class UpLoadFile(models.Model):
      user = models.ForeignKey(User, on_delete=models.CASCADE)
      file = models.FileField(storage=MyStorage())  # Your custom storage
      ...

Также стоит посмотреть на существующиекод для Джанго get_available_name().

...