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.

42 lines
1.7 KiB

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