You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

40 lines
1.6 KiB

11 years ago
11 years ago
11 years ago
11 years ago
  1. #!/usr/bin/env python
  2. import os, sys
  3. from PIL import Image
  4. from PIL import ImageEnhance
  5. import argparse
  6. parser = argparse.ArgumentParser()
  7. parser.add_argument("--width", dest="width", type=int, required=True)
  8. parser.add_argument("--height", dest="height", type=int, required=True)
  9. parser.add_argument("--crop", action="store_true", required=False)
  10. parser.add_argument("source", type=str)
  11. parser.add_argument("destination", type=str)
  12. args = parser.parse_args()
  13. size = args.width, args.height
  14. infile = args.source
  15. outfile = args.destination
  16. try:
  17. im = Image.open(infile)
  18. im = im.convert()
  19. if args.crop:
  20. width_crop = im.size[0]
  21. height_crop = im.size[0]*(args.width/args.height)
  22. width_padding = 0
  23. height_padding = (im.size[1]-height_crop)/2
  24. if height_crop > im.size[1]:
  25. width_crop = im.size[1]*(args.height/args.width)
  26. height_crop = im.size[1]
  27. width_padding = (im.size[0]-width_crop)/2
  28. height_padding = 0
  29. im_crop = im.crop((width_padding, height_padding, width_crop+width_padding, height_crop+height_padding))
  30. im_result = im_crop.resize(size, Image.ANTIALIAS)
  31. else:
  32. im.thumbnail(size, Image.ANTIALIAS)
  33. im_result = Image.new('RGB', (args.width, args.height), (255, 255, 255))
  34. im_result.paste(im, ((args.width - im.size[0]) / 2, (args.height - im.size[1]) / 2))
  35. enhancer = ImageEnhance.Contrast(im_result)
  36. im_result = enhancer.enhance(1.05)
  37. im_result.save(outfile, "JPEG", quality=100)
  38. except IOError:
  39. print("cannot create thumbnail for '%s'" % infile)