72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
import os
|
|
from wand.image import Image
|
|
from wand.color import Color
|
|
|
|
|
|
class ImageProcessor:
|
|
def __init__(
|
|
self, canvas_width=900, canvas_height=900, background_color="transparent"
|
|
):
|
|
"""
|
|
Initialize the ImageProcessor with default values.
|
|
"""
|
|
self.canvas_width = canvas_width
|
|
self.canvas_height = canvas_height
|
|
self.background_color = background_color
|
|
|
|
def set_canvas_size(self, width, height):
|
|
"""
|
|
Set the canvas size.
|
|
"""
|
|
self.canvas_width = int(width)
|
|
self.canvas_height = int(height)
|
|
|
|
def set_background_color(self, color):
|
|
"""
|
|
Set the background color.
|
|
"""
|
|
self.background_color = Color(color)
|
|
|
|
def resize_image(self, image_path, output_path, additional_name=None):
|
|
"""
|
|
Resize and process the image.
|
|
"""
|
|
# 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}>")
|
|
|
|
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:
|
|
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(output_path)[1]
|
|
# Construct the final output path
|
|
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}")
|
|
|
|
|
|
# Example usage
|
|
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")
|