定制FileField中的上传文件名称

FileField中的upload_to属性可以设定上传文件的存储目录和名称，它可以是个字符串，也可以是个callable，比如一个方法。

当upload_to的值设为一个方法时，就可以对上传文件的名称进行修改了。方法需要两个参数，instance与filename，instance为此FileField所属的Model实例，filename为上传文件的名称。

举例：

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)

　当一个类定义了__call__方法时，也可以像func一样被调用，所以，upload_to的值也可以是一个定义了__call__方法的类。

比如，要根据上传时间为文件名加时间戳：

import hashlib

import os

import time

from django.utils.deconstruct import deconstructible

@deconstructible

class TimeStampFileName(object):

    def __init__(self, path):

        self.path = os.path.join(path, "%s%s")

    def __call__(self, instance, filename):

        extension = os.path.splitext(filename)[1]

        data = "%s_%d"%(filename,int(time.time()))

        file_hash = hashlib.sha1(data).hexdigest()

        return self.path % (file_hash, extension)

　Model中的FileField可以如下定义：

class MyModel(models.Model):

    upload = models.FileField(upload_to=TimeStampFileName('media/'), )

当访问一个FileField字段时，得到的是一个FieldFile对象，如果要为Model设置默认文件或图片，可以设置一个默认FieldFile值：

from django.db import models

from django.db.models.fields.files import ImageFieldFile, FileField

class MyModel(models.Model):

    _img = models.ImageField(upload_to=TimeStampFileName('media/'), )

    @property

    def img(self):

        if self._img:

            return self._img

        else:

            return ImageFieldFile(instance=None, field=FileField(),

                                  name='default.png', )

    @img.setter

    def img(self, value):

        self._img = value

秒客网

定制FileField中的上传文件名称

相关文章