Часть прямо из проекта !!!
__author__ = 'vadim7j7'
# ----------------------------------------------------------------------------------------------------------------------
import multiprocessing
from os.path import join, isfile
from os import unlink
from io import BytesIO
import tornado.gen
from concurrent.futures import ThreadPoolExecutor
from tornado.concurrent import run_on_executor
from PIL import Image
from app import BaseHandler
from app.modules.helpers import gen_ran_name, crop_img
from app.sql_raw_methods.User import get_user_info, update_user_photo
# ----------------------------------------------------------------------------------------------------------------------
class UpFile(BaseHandler):
executor = ThreadPoolExecutor(max_workers=multiprocessing.cpu_count())
def __init__(self, application, request, **kwargs):
super().__init__(application, request, **kwargs)
self.path_constants = {
'1': self.settings['tmp_path'],
'2': self.settings['tmp_path'],
'3': self.settings['user_avatars']}
@tornado.gen.coroutine
def post(self):
self._data['dateType'] = 'json'
if self.request.files['file']:
if len(self.request.files['file'][0]['body']) <= 12000000:
try:
thumbnail = yield self.make_thumbnail(self.request.files['file'][0]['body'])
except IOError as er:
self._data['status'] = 500
else:
self._data['body'] = thumbnail
if self.get_argument("dataTypeObject") == '3':
yield self.rewrite_user_avatar(thumbnail)
self.render(None)
def delete(self):
self._data['dateType'] = 'json'
name = self.get_argument('file', None)
from_remove = self.get_argument('from_remove', 'tmp')
if name is not None:
patch_file_tmp = join(self.settings['static_path'], from_remove, name)
if isfile(patch_file_tmp):
unlink(patch_file_tmp)
self._data['body'] = 'Ok'
else:
self._data['status'] = 404
self._data['body'] = 'Not file is tmp'
else:
self._data['status'] = 301
self._data['body'] = 'Not correct request'
self.render(None)
@run_on_executor
def make_thumbnail(self, content):
im = Image.open(BytesIO(content))
path_out = self.path_constants[self.get_argument('dataTypeObject')]
with BytesIO() as output:
new_name = '%s.JPEG' % gen_ran_name()
patch_file_tmp = join(path_out, new_name)
if self.get_argument("dataTypeObject") == '3':
src_width, src_height = im.size
im = crop_img(im, src_width, src_height, 128, 128)
im.convert('RGB').save(patch_file_tmp, 'JPEG', quality=100, optimize=True)
del im
return new_name
@tornado.gen.coroutine
def rewrite_user_avatar(self, photo):
cursor = yield self.db.execute(get_user_info(self.current_user['id'], 'photo'))
user = cursor.fetchone()
if user[0]:
old_avatar_path = join(self.path_constants.get('3'), user[0])
if isfile(old_avatar_path):
unlink(old_avatar_path)
yield self.db.execute(update_user_photo(self.current_user['id'], photo))
# ----------------------------------------------------------------------------------------------------------------------
def crop_img(original_img, src_width, src_height, max_width, max_height):
if max_width <= 0:
max_width = max_height * src_width / src_height
if max_height <= 0:
max_height = max_width * src_height / src_width
src_ratio = float(src_width) / float(src_height)
dst_width, dst_height = max_width, max_height
dst_ratio = float(dst_width) / float(dst_height)
if dst_ratio < src_ratio:
crop_height = src_height
crop_width = crop_height * dst_ratio
x_offset = float(src_width - crop_width) / 2
y_offset = 0
else:
crop_width = src_width
crop_height = crop_width / dst_ratio
x_offset = 0
y_offset = float(src_height - crop_height) / 3
preview_img = original_img.crop((int(x_offset), int(y_offset),
int(x_offset + int(crop_width)),
int(y_offset) + int(crop_height))).resize((int(dst_width), int(dst_height)), 1)
return preview_img
