Django在文件末尾添加了一些奇怪的字符

问题描述

我在向Django服务器提交文件时遇到问题，并且文件名存在于目录中。Django不会覆盖该文件，但它会在名称末尾添加一些字符。如何添加字符而不是Django。

我的mod.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',)

---

解决方案

django的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
      ...

还值得查看Django的get_available_name()的现有代码。