-- coding: UTF-8 --
from os import environ
debug = not environ.get("APP_NAME", "") #判断sae环境
from django.utils.translation import ugettext as _
from django.core.files.storage import FileSystemStorage
from django.core.exceptions import ImproperlyConfigured, SuspiciousOperation
from django.conf import settings
import time,os,uuid,random,unicodedata,StringIO
from django.core.files.base import ContentFile
if not debug:
import sae
import tempfile
import sae.storage
from PIL import Image #这里是关键，sae加载Image的方式
else:
import Image
class SaeAndNotSaeStorage(FileSystemStorage):
"""
这是一个支持sae和本地django的FileStorage基类
修改存储文件的路径和基本url
"""
def init(self, location=settings.MEDIA_ROOT, base_url=settings.MEDIA_URL):
super(SaeAndNotSaeStorage, self).init(location, base_url)
def get_valid_name(self, name):
"""
这个方法用于验证文件名,我这里的处理方法是去掉中文，我没有找到支持中文名的方法，欢迎补充
"""
name = unicodedata.normalize('NFKD', name).encode('ascii', 'ignore')
处理中文文件名sae不支持
if not debug:
try:
if 1:
去掉中文
name = unicodedata.normalize('NFKD', name).encode('ascii', 'ignore')
else:
for k in name:
if self.is_chinese(k):
name = "wszw%s"%random.randint(0,100)
except Exception,e:
name = "%s.jpg"%type(name)
end
return super(SaeAndNotSaeStorage, self).get_valid_name(name)
@property
def maxsize(self):
return 1010241024#文件2M--sae限制只能传2M,单个文件，据说是10M,其实只有2M
@property
def filetypes(self):
return []
def makename(self,name):
取一个不重复的名字，sae会把重名覆盖
oname = os.path.basename(name)
path = os.path.dirname(name)
首先判断是否需要重命名---也就是说不想改名字的就加这个前缀
if oname.find("mine")==0:
oname = oname.replace("mine","")
name = os.path.join(path, oname)
return name
end---首先判断是否需要重命名
try:
fname, hk = oname.split(".")
except Exception,e:
fname, hk = oname, ''
if hk:
rname = "%s_%s.%s"%(random.randint(0,10000), fname,hk)
else:
rname = "%s_%s"%(random.randint(0,10000), fname)
name = os.path.join(path, rname)
end
return name
def _save(self, name, content):
"""
可以判断上传哪些文件
"""
hz = name.split(".")[-1]
类型判断
if self.filetypes!='*':
if hz.lower() not in self.filetypes:
raise SuspiciousOperation(u"不支持的文件类型,支持%s"%self.filetypes)
end
name = self.makename(name)
大小判断
if content.size > self.maxsize:
raise SuspiciousOperation(u"文件大小超过限制")
end
保存
if not debug:
s = sae.storage.Client()
if hasattr(content, '_get_file'):#admin入口
ob = sae.storage.Object(content._get_file().read())
else:#view入口（ContentFile）
ob = sae.storage.Object(content.read())
url =s.put('image', name, ob) #注意这里的media,是sae-storage上的domain名
return name
else:
return super(SaeAndNotSaeStorage, self)._save(name, content)
end--保存
def delete(self,name):
"""
sae的存储空间很宝贵，所有我们在删除图片数据库记录的时候也需要删除图片
"""
if not debug:
s = sae.storage.Client()
try:
s.delete('image', name)
except Exception,e:
pass
else:
super(SaeAndNotSaeStorage, self).delete(name)
class ImageStorage(SaeAndNotSaeStorage):
"""
实现一个ImageField的Storage
"""
@property
def maxsize(self):
return 210241024#文件2M
@property
def filetypes(self):
return ['jpg','jpeg','png','gif']
class FileStorage(SaeAndNotSaeStorage):
@property
def maxsize(self):
return 1010241024#文件5M
@property
def filetypes(self):
return "*"
def makename(self, name):
return name
class ThumbStorage(ImageStorage):
"""
缩略图-------这个非常关键，处理后的图片在sae上怎么保存，关键就在StringIO
"""
def _save(self, name, content):
处理
image = Image.open(content)
image = image.convert('RGB')
image.thumbnail((50, 50), Image.ANTIALIAS)
output = StringIO.StringIO()
image.save(output,'JPEG')
co = ContentFile(output.getvalue())
output.close()
end
return super(ThumbStorage, self)._save(name, co)
调用的model.py：
from storage import ImageStorage,FileStorage,ThumbStorageiconUrl = models.ImageField(upload_to=upload_path_handler, storage=ImageStorage(),verbose_name="图片地址")
