定制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

  

 

posted @ 2017-08-22 16:24  再见紫罗兰  阅读(1615)  评论(0编辑  收藏  举报