Django Admin 上传文件到七牛云

中文圈关于Django Admin 上传文件到七牛云的资料和函数库已经是2年前的了,国外的则都是关于AWS S3、Azure Storage一些国外的服务的。我根据Django的文档里提到的存储系统来实现上传文件到七牛云的简单功能。

在Django Admin的表单是根据数据模型生成的,其中文件上传由FileField和继承FileField的ImageField来决定的,文件上传到静态文件目录,数据库保存相对路径。实现上传文件到七牛云我们是根据FileField的storage参数来实现的。

models.ImageField(storage=MyStorage())

storage参数需要传入一个实现抽象类Storage的对象。

下面引用文档的原句

你自定义的存储系统必须为Django.core.files.storage.Storage的一个子类:

from django.conf import settings
from django.core.files.storage import Storage

class MyStorage(Storage):
    def __init__(self, option=None):
        if not option:
            option = settings.CUSTOM_STORAGE_OPTIONS
        ...

然后重写全部的抽象方法,具体的可以去官网文档看。我们这里先不继承Storage。我在Django的核心里找到了已经继承了的FileSystemStorage,我想只要继承Django标准的文件管理类
FileSystemStorage,重载_save部分变成上传到七牛云不就好了。说干就干。


class MyStorage(FileSystemStorage):
    def _save(self, name, content):
        # 延续原方法的写法
        filename = name.replace('\\', '/')
        # 将文件传入封装好的对象里
        q = Qiniu()
        q.upload_stream(filename, content.file.getvalue())
        return filename

我简单写一下封装的Qiniu上传的类

class Qiniu():
    def __init__(self):
        self.access_key = settings.QINIU_ACCESS_KEY
        self.secret_key = settings.QINIU_SECRET_KEY
        # 要上传的空间
        self.bucket_name = settings.QINIU_BUCKET_NAME
        # 构建鉴权对象
        self.auth = Auth(self.access_key, self.secret_key)
        
   def get_token(self, key):
        """

        :param key: 文件名
        :return: 上传令牌
        """
        policy = {
            'scope': settings.QINIU_BUCKET_NAME,
            'mimeLimit': 'image/jpeg;image/png',
            'deadline': 3600
        }
        # 3600为token过期时间,秒为单位。3600等于一小时
        token = self.auth.upload_token(self.bucket_name, key, 3600, policy)
        return token
        
    def upload_stream(self, filename, stream_data):
        """

        :param filename: 文件名
        :param stream_data: 二进制数据
        :return: 无
        """
        # 上传后保存的文件名
        key = filename
        # 生成上传 Token,可以指定过期时间等
        token = self.auth.upload_token(self.bucket_name, key, 3600)
        # 要上传文件的本地路径
        # localfile = file_path
        ret, info = put_data(up_token=token, key=key, data=stream_data)
        assert ret['key'] == key
        # assert ret['hash'] == etag_stream(stream_data)

参考资料:

  1. https://docs.djangoproject.co...

你可能感兴趣的:(django,python,七牛云存储)