定制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_
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/'), )