96 lines
2.6 KiB
Python
Executable File
96 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# import sys
|
|
import os
|
|
from PIL import Image
|
|
from resizeimage import resizeimage
|
|
|
|
|
|
# save directory name
|
|
directory_name = 'compress'
|
|
|
|
# save directory name and full path
|
|
final_path = f"{os.getcwd()}/{directory_name}"
|
|
|
|
try:
|
|
# if save directory is exists
|
|
if 'compress' in os.listdir():
|
|
pass
|
|
else:
|
|
# save directory not exist
|
|
# create save directory
|
|
os.mkdir(final_path)
|
|
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
|
|
def compressor(file, quality, resize, size):
|
|
filepath = os.path.join(os.getcwd(), file)
|
|
|
|
with Image.open(filepath) as image:
|
|
if resize and image.width >= size and image.width > image.height:
|
|
print('resize and compress: ' + file)
|
|
cover = resizeimage.resize_width(image, size)
|
|
cover.save(
|
|
str(final_path) + '/' + 'min-' + file,
|
|
image.format,
|
|
optimize=True,
|
|
quality=quality,
|
|
resample=Image.LANCZOS
|
|
)
|
|
elif resize and image.height >= size and image.height > image.width:
|
|
print('resize and compress: ' + file)
|
|
cover = resizeimage.resize_height(image, size)
|
|
cover.save(
|
|
str(final_path) + '/' + 'min-' + file,
|
|
image.format,
|
|
optimize=True,
|
|
quality=quality,
|
|
resample=Image.LANCZOS
|
|
)
|
|
else:
|
|
print('compress: ' + file)
|
|
image.save(
|
|
str(final_path) + "/" + "min-" + file,
|
|
image.format,
|
|
optimize=True,
|
|
quality=quality,
|
|
resample=Image.LANCZOS
|
|
)
|
|
|
|
|
|
def main(quality_img, resize, size):
|
|
# get current directory path
|
|
cwd = os.getcwd()
|
|
|
|
# only support these 4 formats
|
|
formats = ('.jpg', '.jpeg', '.png', '.webp')
|
|
|
|
# get all files
|
|
for file in os.listdir(cwd):
|
|
# check file and compress file
|
|
if os.path.splitext(file)[1].lower() in formats:
|
|
# call compress function
|
|
compressor(file, quality_img, resize, size)
|
|
|
|
print("Done")
|
|
|
|
|
|
# Driver code
|
|
if __name__ == "__main__":
|
|
size = 900
|
|
# get image quality from user
|
|
quality_img = int(input("Enter image quality 1 to 100 (defaults to 80): ") or "80")
|
|
# check for resize
|
|
resize = str(input("Resize images? (N/y): ") or "N").lower()
|
|
if resize == 'n':
|
|
resize = False
|
|
elif resize == 'y':
|
|
resize = True
|
|
# get desired size
|
|
size = int(input("Size of longer side (defaults to 900): ") or "900")
|
|
|
|
main(quality_img, resize, size)
|