目前公司的项目把文件图片都是存到腾讯云cos的,其实最佳实践腾讯云文档上都说了,后端签好名,前端直接web上传,省事省力。但是最近有个项目需要用到admin后台,也需要上传文件和图片到腾讯云cos,一开始想到自己重写admin页面增加上传功能,但是这就违背我用admin的初衷(就看中了admin的方便),所以就去翻看django的官方文档。
看了一波后,发现其实还挺简单的,只需要自己自定义storage
类就行了。参考官方文档
自定义storge
只需要几个步骤就行了:
实现一个类,继承
Django.core.files.storage.Storage
。实例化的时候不能传参,但是可以从
settings
中获取。如果要顺利迁移数据库文件的话,再加一个
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())
这样就完事了。