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

2023-12-04

当我将文件提交到 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',)

Django 的默认行为Storageclass 的作用是当文件名已经存在时在文件名末尾附加一系列随机字符。如果您想控制附加哪些字符,您可以创建自己的字符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().

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

Django 在文件末尾添加一些奇怪的字符 的相关文章

随机推荐