import os
import sys

from PIL import Image
from io import BytesIO


def rescale_image_jpeg(image_object, base_width=720):
    img = Image.open(image_object)
    icc_profile = img.info.get('icc_profile')  # remember color profile to not mess up colors
    width_percent = (base_width / float(img.size[0]))
    height = int((float(img.size[1]) * float(width_percent)))

    img = img.resize((base_width, height), Image.ANTIALIAS)
    img_bytes = BytesIO()
    img.save(img_bytes, 'JPEG', quality=75, optimize=True, icc_profile=icc_profile)

    return img_bytes


def rescale_image_png(image_object, base_width=720):
    basewidth = 720
    wpercent = (basewidth / float(image_object.size[0]))
    hsize = int((float(image_object.size[1]) * float(wpercent)))
    img = image_object.resize((basewidth, hsize), Image.ANTIALIAS)

    im_io = BytesIO()
    img.save(im_io, 'PNG', quality=70)
    return img


def get_filetype(name):
    try:
        return os.path.splitext(name)[1]
    except:
        return '.jpeg'


def handle_image(request, image_object, filetype='.jpeg'):
    if sys.getsizeof(image_object) / 8 > 500:
        if filetype == '.jpeg':
            return rescale_image_jpeg(image_object), filetype
        if filetype == '.png':
            return rescale_image_png(image_object), filetype
    return image_object, filetype