Update log messages and set a more dynamic image resize up to a limit

This commit is contained in:
Sheldon Lee 2023-11-18 15:57:36 +08:00
parent 737821fa6a
commit 3c1f996342

View File

@ -4,15 +4,13 @@ import sys
def main(): def main():
target_width = 2048
argn = len(sys.argv) argn = len(sys.argv)
if argn == 1: if argn == 1:
print("You must specify an input file") print("Error: You must specify an input file")
exit(1) exit(1)
elif argn == 2: elif argn == 2:
print("You must specify an output file") print("Error: You must specify an output file")
exit(1) exit(1)
input_image_path = sys.argv[1] input_image_path = sys.argv[1]
@ -21,10 +19,12 @@ def main():
try: try:
image = Image.open(input_image_path) image = Image.open(input_image_path)
except IOError: except IOError:
print(f"Cannot open image file {input_image_path}") print(f"Error: Cannot open image file {input_image_path}")
image.close()
exit(1) exit(1)
width, height = image.size width, height = image.size
target_width = int(max(width/4, 1280))
new_height = int(target_width * height / width) new_height = int(target_width * height / width)
image = image.resize((target_width, new_height), Image.Resampling.BICUBIC) image = image.resize((target_width, new_height), Image.Resampling.BICUBIC)
@ -37,7 +37,8 @@ def main():
try: try:
image.save(output_image_path, **kwargs) image.save(output_image_path, **kwargs)
except IOError: except IOError:
print(f"Cannot save file {output_image_path}") print(f"Error: Cannot save file {output_image_path}")
image.close()
exit(1) exit(1)