avatar.py

#!/usr/bin/python

import cgi, cgitb
import random
from PIL import Image, ImageOps
from StringIO import StringIO

SQUARESIZE = 150
SIZERANGE = (70, 130)

images = (("../img/classy-small.png", True), ("../img/drugface.png", True),
          ("../img/dewmgui.png", True))

cgitb.enable()

avatar  = Image.new("RGBA", (SQUARESIZE, SQUARESIZE), (0, 0, 0, 0))

imgName, fullRotate = random.choice(images)

avatarI = Image.open(imgName)

if random.randint(0, 1):
    avatarI = ImageOps.mirror(avatarI)

randScale = random.randint(*SIZERANGE) / 100.0

avatarI = avatarI.resize((avatarI.size[0] * randScale, avatarI.size[1] * randScale), Image.BICUBIC)

avatar.paste(avatarI, ((SQUARESIZE-avatarI.size[0]) // 2, (SQUARESIZE-avatarI.size[1]) // 2))

if fullRotate:
    avatar = avatar.rotate(random.randint(1, 360), resample=Image.BICUBIC)
else:
    avatar = avatar.rotate(90 * random.randint(1,4))


avaMsg  = StringIO()
avatar.save(avaMsg, "PNG")

print("Content-type: image/png\n")
print(avaMsg.getvalue())