codekasten/gallery.py

160 lines
4.9 KiB
Python
Executable File

#!/usr/bin/env python
# see: http://www.amk.ca/python/code/
import sys, os, getopt
try:
from PIL import Image
except ImportError:
print >>sys.stderr, ("Unable to import PIL.Image; "
"is the Python Imaging Library installed?"
"\(e.g.: apt-get install python-image\)")
sys.exit(0)
__version__ = '1.01'
__doc__ = """%s [option] file1 file2
Options:
--help Display this usage message
--html Output HTML to stdout
--resize x y Resize specified images to x,y
--resize-ratio x Resize specified images with correct aspect ratio
--rotate x Rotate the image counter clockwise for x degrees
--size Display the image sizes
--thumbnail x y Make thumbnails of specified images, ignoring existing
thumbnails that happen to be listed among the files.
If a filename is of the form *_thumb.*, it's assumed to be a thumbnail.
""" % sys.argv[0]
def is_thumbnail (filename):
"Returns true if the filename is for a thumbnail"
root, ext = os.path.splitext(filename)
return root.endswith('_thumb')
def thumbnail_name (filename):
"""Return the thumbnail form of a filename,
converting foo.jpg to foo_thumb.jpg.
"""
assert not is_thumbnail(filename)
root, ext = os.path.splitext(filename)
return root + '_thumb' + ext
def output_html (args):
for filename in args:
if is_thumbnail(filename): continue
thumbnail = thumbnail_name(filename)
if not os.path.exists(thumbnail):
print >>sys.stderr, ("%s: thumbnail %s doesn't exist" %
(sys.argv[0], thumbnail) )
im = Image.open(thumbnail)
width, height = im.size
print ('<p><a href="%s"><img src="%s" width="%i" height="%i" '
'alt="XXX" align="XXX"></a>'
'\n\nXXX\n'
'<br clear=all><hr>\n\n'
% (filename, thumbnail, width, height) )
def make_thumbnails (args):
x, y = int(args[0]), int(args[1])
args = args[2:] ; args.sort()
for filename in args:
if is_thumbnail(filename): continue
thumbnail = thumbnail_name(filename)
print >>sys.stderr, filename
im = Image.open(filename)
im.thumbnail((x,y))
im.save(thumbnail)
def rotate_images (args):
x = int(args[0])
args = args[1:] ; args.sort()
for filename in args:
im = Image.open(filename)
im2 = im.rotate(x)
im2.save(filename)
print "rotated %s %i degrees" % (filename, x)
def resize_images (args):
x, y = int(args[0]), int(args[1])
args = args[2:] ; args.sort()
for filename in args:
im = Image.open(filename)
w,h = im.size
im2 = im.resize((x,y))
print >>sys.stderr, ('%s: was %i,%i; resizing to %i,%i'
% (filename, w,h, x,y) )
im2.save(filename)
def resize_images_aspectratio (args):
x = int(args[0])
args = args[1:] ; args.sort()
for filename in args:
im = Image.open(filename)
w,h = im.size
y = x*h/w
im2 = im.resize((x,y))
print >>sys.stderr, ('%s: was %i,%i; resizing to %i,%i'
% (filename, w,h, x,y) )
im2.save(filename)
def main ():
opts, args = getopt.getopt(sys.argv[1:],
'h',
['help',
'html',
'resize',
'resize-ratio',
'size',
'thumbnail',
'rotate'])
# Remove the unused option arguments
opts = [opt for opt,arg in opts]
# Print usage message if requested
if '-h' in opts or '--help' in opts:
print >>sys.stderr, __doc__
sys.exit(0)
# Ensure that exactly one option is supplied
if len(opts) == 0:
print >> sys.stderr, ("%s: must specify one of --size, --help"
"\n --html, --resize, --resize-ratio, --thumbnail --rotate" % sys.argv[0] )
sys.exit(0)
elif len(opts) > 1:
print >> sys.stderr, ("%s: cannot specify multiple options" %
sys.argv[0] )
sys.exit(0)
# Perform each of the possible actions
opt = opts[0]
if opt == '--html':
output_html(args)
elif opt == '--resize':
resize_images(args)
elif opt == '--resize-ratio':
resize_images_aspectratio(args)
elif opt == '--rotate':
rotate_images(args)
elif opt == '--size':
for filename in args:
image = Image.open(filename)
x, y = image.size
print filename, ':', x,y
elif opt == '--thumbnail':
make_thumbnails(args)
else:
print >>sys.stderr, ("%s: unknown option" % sys.argv[0])
if __name__ == '__main__':
main()