imgpx.py

from PIL import Image
import os

class ImgPx:

    def __init__(self, path = os.getcwd()):
        self.path = path

    def createFolder(self, folder):
        if not os.path.exists(folder):
            os.makedirs(folder)

    def destinationFile(self, source_img, des_folder, des_format = None):
        base_name = os.path.basename(source_img)
        filename = os.path.splitext(base_name)[0]
        extention = os.path.splitext(base_name)[1]
        if des_format is None:
            file_ext = filename + extention
        else:
            file_ext = filename + "." + des_format
        return os.path.join(des_folder, file_ext)

    def saveImage(self, source_img, des_format, des_folder):
        try:
            image = Image.open(source_img)
            new_image = self.destinationFile(source_img, des_folder, des_format)
            image.save(new_image)
        except Exception:
            pass

    def typeConvert(self, des_format, des_folder = "Converted"):
        self.createFolder(des_folder)
        for file in os.listdir(self.path):
            file_path = os.path.join(self.path, file)
            self.saveImage(file_path, des_format, des_folder)

    def resizeImage(self, source_img, width, height, des_folder, aspect_ratio, des_format ):
        try:
            size = width, height
            image = Image.open(source_img)
            if aspect_ratio:
                image.thumbnail((width, height), Image.ANTIALIAS)
            else:
                image = image.resize(size)
            new_image = self.destinationFile(source_img, des_folder, des_format)
            image.save(new_image)
        except Exception:
            pass

    def resize(self, width, height, aspect_ratio = True, des_format = None, des_folder = "Resized"):
        self.createFolder(des_folder)
        for file in os.listdir(self.path):
            file_path = os.path.join(self.path, file)
            self.resizeImage(file_path, width, height, des_folder, aspect_ratio, des_format)

    def resizedDimension(self, original_size, target_size, dimension):
        aspect = dimension[0] / dimension[1]
        size_deviation = original_size / target_size
        width = int(dimension[0] / size_deviation ** 0.5)
        height = int(width / aspect)
        dim = (width, height)
        return dim

    def resizeValidation(self, source_img, destination, width, height, size):
        image = Image.open(source_img)
        image.thumbnail((width, height), Image.ANTIALIAS)
        image.save(destination)
        if os.path.getsize(destination) <= size:
            return
        else:
            self.resizeValidation(source_img, destination, width-10, height, size)

    def resizeBySizeImage(self, source_img, expected_size, des_folder):
        try:
            image = Image.open(source_img)
            original_size = os.path.getsize(source_img)
            dimension = self.resizedDimension(original_size, expected_size, image.size)
            width = dimension[0]
            height = dimension[1]
            destination_image = self.destinationFile(source_img, des_folder)
            self.resizeValidation(source_img, destination_image, width, height, expected_size)
        except Exception:
            pass

    def resizeBySize(self, size, des_folder = "Resized By Size"):
        self.createFolder(des_folder)
        for file in os.listdir(self.path):
            file_path = os.path.join(self.path, file)
            self.resizeBySizeImage(file_path, size, des_folder)