django admin把图片，文件都上传到腾讯云cos服务器

目前公司的项目把文件图片都是存到腾讯云cos的，其实最佳实践腾讯云文档上都说了，后端签好名，前端直接web上传，省事省力。但是最近有个项目需要用到admin后台，也需要上传文件和图片到腾讯云cos，一开始想到自己重写admin页面增加上传功能，但是这就违背我用admin的初衷（就看中了admin的方便），所以就去翻看django的官方文档。

看了一波后，发现其实还挺简单的，只需要自己自定义storage类就行了。参考官方文档

自定义storge只需要几个步骤就行了：

1. 实现一个类，继承Django.core.files.storage.Storage。

2. 实例化的时候不能传参，但是可以从settings中获取。

3. 实现 _open() 以及 _save() 方法。

4. 如果要顺利迁移数据库文件的话，再加一个django.utils.deconstruct.deconstructible类装饰器就ok了。

具体项目代码如下：

import random
import time
from hashlib import sha1

from django.conf import settings
from django.core.files.storage import Storage
from django.utils.deconstruct import deconstructible
from qcloud_cos import CosConfig, CosS3Client

secret_id = settings.COS_SECRET_ID
secret_key = settings.COS_SECRET_KEY
region = settings.REGION
bucket = settings.BUCKET
config = CosConfig(Region=region, Secret_id=secret_id, Secret_key=secret_key)
client = CosS3Client(config)
host = 'https://' + bucket + '.cos.' + region + '.myqcloud.com/'


@deconstructible
class CosStorage(Storage):

    def save(self, name, content, max_length=None):
        """
        我没有实现_save()方法，而是直接重写了save()方法，因为save()其实是调用了_save()方法，
        所以这样简单粗暴，不知有没有坑。
        """
        suffix = name.split('.')[-1]
        key = self.generate_key(suffix)
        try:
            response = client.put_object(
                Bucket=bucket,
                Body=content.read(),
                Key=key,
                EnableMD5=False
            )
        except Exception as e:
            raise
        return host + key

    def generate_key(self, suffix):
        """
          给文件重命名
        """
        file_name = str(int(time.time() * 10000000)) + ''.join([str(random.randint(1, 9)) for i in range(3)])
        s = sha1()
        s.update(file_name.encode('utf-8'))
        file_name = str(s.hexdigest())
        key = file_name + '.' + suffix
        return key

    def url(self, name):
        return name

定义完storage，然后在定义字段的时候直接引用它就行了。
例如：

file = models.FileField(verbose_name='文件', storage=CosStorage())

这样就完事了。

大家如果喜欢，欢迎点赞分享评论哦！😘