Django FileField，在运行时上传_to

我试图设置我的上传，以便如果用户乔上传一个文件，它去 MEDIA _ ROOT/乔，而不是让每个人的文件去 MEDIA _ ROOT。问题是我不知道如何在模型中定义它。以下是目前的情况:

class Content(models.Model):
name = models.CharField(max_length=200)
user = models.ForeignKey(User)
file = models.FileField(upload_to='.')

所以我想要的不是’.’作为上传 _ to，而是用户的名字。

据我所知，在 Django 1.0中，您可以定义自己的函数来处理 load _ to，但是该函数也不知道用户是谁，所以我有点迷茫。

谢谢你的帮助！

70618 次浏览
小开
最佳答案

你可能读过 文件，所以这里有一个简单的例子来说明它的意义:

def content_file_name(instance, filename):
return '/'.join(['content', instance.user.username, filename])


class Content(models.Model):
name = models.CharField(max_length=200)
user = models.ForeignKey(User)
file = models.FileField(upload_to=content_file_name)

正如您所看到的，您甚至不需要使用给定的文件名——如果您愿意，也可以在 upload_ to 调用中覆盖该文件名。

小开

这真的很有帮助。为了简洁起见，决定在我的例子中使用 lambda:

file = models.FileField(
upload_to=lambda instance, filename: '/'.join(['mymodel', str(instance.pk), filename]),
)
小开

关于使用“ instance”对象的 pk 值的注意事项。根据文档:

在大多数情况下，此对象尚未保存到数据库中，因此如果它使用默认的 AutoField，则可能还没有其主键字段的值。

因此，使用 pk 的有效性取决于定义特定模型的方式。

小开

如果您在迁移方面有问题，那么您可能应该使用 @deconstructible装饰器。

import datetime
import os
import unicodedata


from django.core.files.storage import default_storage
from django.utils.deconstruct import deconstructible
from django.utils.encoding import force_text, force_str




@deconstructible
class UploadToPath(object):
def __init__(self, upload_to):
self.upload_to = upload_to


def __call__(self, instance, filename):
return self.generate_filename(filename)


def get_directory_name(self):
return os.path.normpath(force_text(datetime.datetime.now().strftime(force_str(self.upload_to))))


def get_filename(self, filename):
filename = default_storage.get_valid_name(os.path.basename(filename))
filename = force_text(filename)
filename = unicodedata.normalize('NFKD', filename).encode('ascii', 'ignore').decode('ascii')
return os.path.normpath(filename)


def generate_filename(self, filename):
return os.path.join(self.get_directory_name(), self.get_filename(filename))

用法:

class MyModel(models.Model):
file = models.FileField(upload_to=UploadToPath('files/%Y/%m/%d'), max_length=255)
小开

如果您有一个用户实例，那么应该有一个快速的设置来生成

<model-slug>/<username>-<first_name>-<last_name>/filename-random.png

/medias/content/ft0004-john-doe/filename-lkl9237.png


def upload_directory_name(instance, filename):


user = getattr(instance, 'user', None)
if user:
name = f"{user.username}-{user.get_full_name().replace(' ', '-')}"
else:
name=str(instance)
model_name = instance._meta.verbose_name.replace(' ', '-')
return str(os.path.pathsep).join([model_name, name, filename])




class Content(models.Model):
name = models.CharField(max_length=200)
user = models.ForeignKey(User)
file = models.FileField(upload_to=upload_directory_name)

[@SmileyChris 的修改版]

小开

我想在运行时更改上传路径，但没有一个解决方案适合这种需要。

这就是我所做的:

class Content(models.Model):
name = models.CharField(max_length=200)
user = models.ForeignKey(User)
file = models.FileField(upload_to=DynamicUploadPath.get_file_path)




class ContentSerializer(serializers.ModelSerializer):
class Meta:
model = Content
fields = '__all__'




class UploadDir(models.TextChoices):
PRODUCT = 'PRD', _('Product')
USER_PROFILE = 'UP', _('User Profile')




class DynamicUploadPath:
dir: UploadDir = None


@classmethod
def get_file_path(cls, instance, filename):
return str(cls.dir.name.lower() + '/' + filename)




def set_DynamicUploadPath(dir: UploadDir):
DynamicUploadPath.dir = dir




class UploadFile(APIView):
parser_classes = (MultiPartParser, FormParser)


def post(self, request):
# file save path: MEDIA_ROOT/product/filename
set_DynamicUploadPath(UploadDir.PRODUCT)


# file save path: MEDIA_ROOT/user_profile/filename
# set_DynamicUploadPath(UploadDir.USER_PROFILE)


serializer = ContentSerializer(data=request.data)
serializer.is_valid(raise_exception=True)
serializer.save()


return Response(serializer.data, status=status.HTTP_200_OK)

提交答案