roberto/siti-image-convertor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
QuBerto
2024-07-14 04:07:50 +02:00
parent cc2aca0f4d
commit 0702b93405
6 changed files with 282 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
utils/file_operations.py
View File

@@ -2,7 +2,7 @@ import os
import shutil
from tkinter import filedialog, messagebox
from utils.image_processing import ImageProcessor
from pprint import pprint
class FileProcessor:
"""
@@ -134,17 +134,44 @@ class FileProcessor:
log (function): The log function to use.
"""
image = ImageProcessor()
image.set_background_color(options.get("background_color", "transparent"))
image.set_image_size(options.get("image_size", "contain"))
for file_path in image_paths:
output_path = os.path.join(
output_directory, os.path.relpath(
file_path, self.selected_directory)
)
# output_path = os.path.join(
# output_directory, os.path.relpath(
# file_path, self.selected_directory)
# )
output_path = self.generate_output_path(output_directory, file_path, options)
os.makedirs(os.path.dirname(output_path), exist_ok=True)
self.log_message(f"Running: {file_path}", log)
image.resize_image(
file_path, output_path, options.get("additional_name", "")
file_path, output_path, options
)
if os.path.exists(file_path) and options.get("is_checked", False):
if os.path.exists(file_path) and options.get("delete_images", False):
self.log_message(f"Removing: {file_path}", log)
os.remove(file_path)
self.log_message(f"Processed: {file_path}", log)
def generate_output_path(self, output_directory, file_path, options, product = None):
"""
Generate the output path for resized images based on a template.
Returns:
str: The generated output path.
"""
sku = slug = title = ""
name, ext = os.path.splitext(os.path.basename(file_path))
width = options.get("canvas_width")
height = options.get("canvas_height")
if product:
sku = product.get("sku", "")
slug = product.get("name", "")
title = product.get("slug", "")
new_filename = options.get('template', '{name}').format(
name=name, sku=sku, width=width, height=height, slug=slug, title=title
)
pprint(new_filename)
return os.path.join(output_directory, new_filename + ext)

120
utils/image_processing.py
View File

@@ -4,15 +4,14 @@ from wand.color import Color
class ImageProcessor:
def __init__(
self, canvas_width=900, canvas_height=900, background_color="transparent"
):
def __init__(self, canvas_width=900, canvas_height=900, background_color="transparent", image_size="fit"):
"""
Initialize the ImageProcessor with default values.
"""
self.canvas_width = canvas_width
self.canvas_height = canvas_height
self.background_color = background_color
self.background_color = Color(background_color)
self.image_size = image_size
def set_canvas_size(self, width, height):
"""
@@ -27,40 +26,106 @@ class ImageProcessor:
"""
self.background_color = Color(color)
def resize_image(self, image_path, output_path, additional_name=None):
def set_image_size(self, size):
"""
Set the image size mode.
"""
self.image_size = size
def resize_image(self, image_path, output_path, options):
"""
Resize and process the image.
Args:
image_path (str): The path to the input image.
output_path (str): The path to the output image.
additional_name (str, optional): Additional name to append to the output filename.
mode (str, optional): The resizing mode ("contain", "cover", "fit"). Default is "contain".
"""
log = options.get("log_message", None)
# Normalize the paths to ensure consistency
image_path = os.path.normpath(image_path)
output_path = os.path.normpath(output_path)
print(image_path)
print(output_path)
with Image(filename=image_path) as img:
img.transform(resize=f"{self.canvas_width}x{self.canvas_height}>")
self.log_message(f"Original image size: {img.width}x{img.height}", log)
if self.image_size == "contain":
self._contain(img)
elif self.image_size == "cover":
self._cover(img)
# elif self.image_size == "fit":
# self._fit(img)
x_offset = int((self.canvas_width - img.width) / 2)
y_offset = int((self.canvas_height - img.height) / 2)
with Image(
width=self.canvas_width,
height=self.canvas_height,
background=self.background_color,
) as canvas:
with Image(width=self.canvas_width, height=self.canvas_height, background=self.background_color) as canvas:
canvas.composite(img, left=x_offset, top=y_offset)
# Create a new filename
new_filename = os.path.splitext(
os.path.basename(output_path))[0]
if additional_name:
new_filename += " - " + additional_name.strip()
new_filename = os.path.splitext(os.path.basename(output_path))[0]
new_filename += os.path.splitext(output_path)[1]
# Construct the final output path
final_output_path = os.path.join(
os.path.dirname(output_path), new_filename
)
final_output_path = os.path.join(os.path.dirname(output_path), new_filename)
# Save the image to the final output path
canvas.save(filename=final_output_path)
print(f"Saved to: {final_output_path}")
self.log_message(f"Saved to: {final_output_path}", log)
def _cover(self, img:Image):
"""
Resize the image to cover the entire canvas.
"""
base_width = self.canvas_width
base_heigth = self.canvas_height
wpercent = (base_width / float(img.size[0]))
hpercent = (base_heigth / float(img.size[1]))
hsize = int((float(img.size[1]) * float(wpercent)))
wsize = int((float(img.size[0]) * float(hpercent)))
img.resize(wsize, base_heigth)
aspect_ratio_img = img.width / img.height
aspect_ratio_canvas = self.canvas_width / self.canvas_height
print(f"Image aspect ratio: {aspect_ratio_img}, Canvas aspect ratio: {aspect_ratio_canvas}")
print(f"Cover resized image size: {img.width}x{img.height}")
def _contain(self, img):
"""
Resize the image to cover the entire canvas.
"""
aspect_ratio_img = img.width / img.height
aspect_ratio_canvas = self.canvas_width / self.canvas_height
print(f"Image aspect ratio: {aspect_ratio_img}, Canvas aspect ratio: {aspect_ratio_canvas}")
if aspect_ratio_img > aspect_ratio_canvas:
img.transform(resize=f"{self.canvas_width}x")
else:
img.transform(resize=f"x{self.canvas_height}")
print(f"Cover resized image size: {img.width}x{img.height}")
# def _fit(self, img):
# """
# Fit the image within the canvas without scaling up if it's smaller.
# """
# if img.width > self.canvas_width or img.height > self.canvas_height:
# img.transform(resize=f"{self.canvas_width}x{self.canvas_height}>")
# print(f"Fit resized image size: {img.width}x{img.height}")
def log_message(self, message, log=None):
"""
Log a message or print it if no log function is provided.
Args:
message (str): The message to log or print.
log (function, optional): The log function to use. Defaults to None.
"""
if log:
log(message)
else:
print(message)
# Example usage
@@ -68,4 +133,15 @@ if __name__ == "__main__":
processor = ImageProcessor()
processor.set_canvas_size(900, 900)
processor.set_background_color("white")
processor.resize_image("input_image.jpg", "output_image.jpg", "example")
# Contain mode
processor.set_image_size("contain")
processor.resize_image("input_image.jpg", "output_image_contain.jpg", "example")
# Cover mode
processor.set_image_size("cover")
processor.resize_image("input_image.jpg", "output_image_cover.jpg", "example")
# Fit mode
processor.set_image_size("fit")
processor.resize_image("input_image.jpg", "output_image_fit.jpg", "example")