Implement thumbnail generation
This commit is contained in:
Executable
+62
@@ -0,0 +1,62 @@
|
||||
#!/usr/bin/python
|
||||
from PIL import Image
|
||||
import sys
|
||||
|
||||
|
||||
def main():
|
||||
target_width = 2048
|
||||
|
||||
argn = len(sys.argv)
|
||||
|
||||
if argn == 1:
|
||||
print("You must specify an input file")
|
||||
exit(1)
|
||||
elif argn == 2:
|
||||
print("You must specify an output file")
|
||||
exit(1)
|
||||
|
||||
input_image_path = sys.argv[1]
|
||||
output_image_path = sys.argv[2]
|
||||
|
||||
try:
|
||||
image = Image.open(input_image_path)
|
||||
except IOError:
|
||||
print(f"Cannot open image file {input_image_path}")
|
||||
exit(1)
|
||||
|
||||
width, height = image.size
|
||||
new_height = int(target_width * height / width)
|
||||
|
||||
image = image.resize((target_width, new_height), Image.Resampling.BICUBIC)
|
||||
image = rotate_image(image)
|
||||
|
||||
kwargs = {"quality": 50}
|
||||
if "exif" in image.info.keys():
|
||||
kwargs["exif"] = image.info["exif"]
|
||||
|
||||
try:
|
||||
image.save(output_image_path, **kwargs)
|
||||
except IOError:
|
||||
print(f"Cannot save file {output_image_path}")
|
||||
exit(1)
|
||||
|
||||
|
||||
def rotate_image(image):
|
||||
exif = image.getexif()
|
||||
if exif is None:
|
||||
return image
|
||||
|
||||
orientation = exif.get(0x0012, 1) # 0x0112 is the EXIF code for orientation
|
||||
if orientation == 1:
|
||||
return image
|
||||
elif orientation == 3:
|
||||
image = image.rotate(180, expand=True)
|
||||
elif orientation == 6:
|
||||
image = image.rotate(270, expand=True)
|
||||
elif orientation == 8:
|
||||
image = image.rotate(90, expand=True)
|
||||
return image
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user