7j7

__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