summaryrefslogtreecommitdiff
path: root/imgrid.py
diff options
context:
space:
mode:
Diffstat (limited to 'imgrid.py')
-rwxr-xr-ximgrid.py356
1 files changed, 356 insertions, 0 deletions
diff --git a/imgrid.py b/imgrid.py
new file mode 100755
index 0000000..a751b6b
--- /dev/null
+++ b/imgrid.py
@@ -0,0 +1,356 @@
+#!/usr/bin/python2.7
+
+import sys
+import re
+import os
+import string
+import simplejson as json
+import random
+import time
+import urllib
+import urllib2
+from subprocess import Popen,PIPE,call
+urlencode = urllib.urlencode
+urlopen = urllib2.urlopen
+Request = urllib2.Request
+
+MAX_SIZE = 1024 * 1024 * 1.2 * 1.5
+WORKING_DIR = "/tmp"
+
+BIN_CONVERT = "/usr/bin/convert"
+BIN_COMPOSITE = "/usr/bin/composite"
+BIN_IDENTIFY = "/usr/bin/identify"
+THREEDROTATE = "./3Drotate"
+GRID = "./grid"
+DB_TAG = "grid";
+
+DEFAULT_HEIGHT = 400
+DEFAULT_WIDTH = 600
+DEFAULT_LINE_COLOR = "silver"
+DEFAULT_FINALFORMAT = "png"
+
+#{{{Utility functions
+def bool_correct(s):
+ if re.match(r'^false$', s, re.IGNORECASE):
+ return False
+ elif re.match(r'^true$', s, re.IGNORECASE):
+ return True
+ else:
+ return s
+
+class dotdict(dict):
+ """dot.notation access to dictionary attributes"""
+ def __getattr__(self, attr):
+ return self.get(attr)
+ __setattr__= dict.__setitem__
+ __delattr__= dict.__delitem__
+
+def get_mimetype(f):
+ try:
+ mimetype = Popen(
+ [BIN_IDENTIFY, f], stdout=PIPE
+ ).communicate()[0].split(" ")[1].lower()
+ return mimetype
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise;
+
+def sanitize (str):
+ return re.sub(r'\W+', '', str)
+
+def now():
+ return int(time.time())
+
+def browser_request (url, data=None):
+ headers = {
+ 'User-Agent': 'Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)',
+ 'Accept': '*/*',
+ }
+ try:
+ req = Request(url, data, headers)
+ response = urlopen(req)
+ except IOError, e:
+ if hasattr(e, 'code'):
+ sys.stderr.write( '%s - ERROR %s' % (url, e.code) )
+ raise;
+ return None
+ else:
+ return response
+
+def download(url, destination, max_size=MAX_SIZE):
+ response = browser_request(url, None)
+ rawimg = response.read()
+ if len(rawimg) == 0:
+ sys.stderr.write("got zero-length file")
+ raise;
+ if len(rawimg) > max_size:
+ sys.stderr.write("file too big: max size {} KB / {} is {} KB".format(
+ str(MAX_SIZE/1024),
+ destination,
+ str(len(rawimg)/1024)
+ ))
+ raise;
+ f = open(destination, "w")
+ f.write(rawimg)
+ f.close()
+
+def file_size (filepath):
+ try:
+ return os.stat(file)[6]
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise;
+
+def gif_frames(filepath):
+ try:
+ info = Popen([BIN_IDENTIFY,filepath], stdout=PIPE).communicate()[0]
+ frames = filter((lambda x: x), map(
+ (lambda x: x.split(" ")[0]),
+ (info).split('\n')
+ ))
+ return frames
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise;
+#}}}
+
+class Imgrid():
+ def __init__(self, params):
+ self.params = {}
+ self.tag = "imGrid"
+ self.now = now()
+ self.files_created = []
+ self.commands = [];
+ self.required_keys = [
+#{{{ required_keys
+ "width",
+ "height",
+ "linethickness",
+ "opacity",
+ "linecolor",
+ "spacing",
+ "vlines",
+ "hlines",
+ "shadow",
+ "bgimage",
+ "bgcolor",
+ "imageinstead",
+ "planebgcolor",
+ "planebgimage",
+ "swing",
+ "tilt",
+ "roll",
+ "zoom",
+ "skycolor",
+ "transition",
+ "trim",
+ "format",
+ "username"
+#}}}
+ ]
+ for k in self.required_keys:
+ if k in params:
+ if k in [ 'bgimage', 'planebgimage', 'imageinstead' ] and bool_correct(params[k]):
+ self.params[k] = {}
+ self.params[k]['url'] = params[k]
+ self.params[k]['filename'] = "IMGRIDTMP{}_{}".format(now(), k)
+
+ self.params[k]['path'] = os.path.join(WORKING_DIR, self.params[k]['filename'])
+ try:
+ download(self.params[k]['url'], self.params[k]['path'])
+ self.files_created.append(self.params[k]['path'])
+ self.params[k]['mimetype'] = get_mimetype(self.params[k]['path'])
+ frames = gif_frames(self.params[k]['path'])
+ if len(frames) > 1:
+ self.params[k]['path'] = random.choice(frames)
+
+
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise;
+ else:
+ self.params[k] = bool_correct(sanitize(params[k]))
+ else:
+ self.params[k] = False;
+
+ self.params = dotdict(self.params)
+
+ self.basename = self._get_filename();
+
+ if not self.params.finalformat:
+ self.params.finalformat = DEFAULT_FINALFORMAT
+ self.filename = "{}.{}".format(self.basename, self.params.finalformat)
+ #final filepath is stored in self.filepath
+ self.filepath = os.path.join(WORKING_DIR, self.filename)
+
+ def _get_filename(self):
+ return "{}_{}_{}".format(
+ self.tag,
+ now(),
+ self.params.username or ""
+ );
+
+ #makes a canvas file...step 1 (if not bgimage)
+ def _make_canvas(self):
+ dimensions = "{}x{}".format(
+ self.params.width or DEFAULT_WIDTH,
+ self.params.height or DEFAULT_HEIGHT
+ )
+ if self.params.bgimage:
+ return
+ bgcolor = "xc:{}".format(self.params.bgcolor or 'transparent')
+ cmd = [ BIN_CONVERT, "-size", dimensions, bgcolor, self.filepath ]
+ try:
+ call(cmd)
+ self.commands.append(" ".join(cmd))
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise;
+
+ #2nd step-- run grid
+ def _grid_command(self):
+ cmd = [GRID]
+ if self.params.spacing:
+ if self.params.vlines:
+ width = 2 * int(self.params.width or DEFAULT_WIDTH)
+ cmd += ["-s","{},{}".format(self.params.spacing,width)]
+ elif self.params.hlines:
+ height = 2 * int(self.params.height or DEFAULT_HEIGHT)
+ cmd += ["-s", "{},{}".format(height,self.params.spacing)]
+ else:
+ cmd += ["-s",self.params.spacing]
+ cmd += [ "-c", self.params.linecolor or DEFAULT_LINE_COLOR]
+ if self.params.linethickness: cmd += ['-t',self.params.linethickness]
+ if self.params.opacity: cmd += ['-o',self.params.opacity]
+ cmd += [ self.filepath, self.filepath ]
+ try:
+ call(cmd)
+ self.commands.append(" ".join(cmd))
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise;
+
+ def _shadow_cmd(self):
+ #convert 1.png \( +clone -background black -shadow 110x1+9+9 \) +swap -background none -layers merge +repage 2.png
+ cmd = [
+ BIN_CONVERT,
+ self.filepath,
+ "(","+clone","-background","black","-shadow","100x2+20+10",")",
+ "+swap","-background","none","-layers","merge","+repage" ,
+ self.filepath
+ ]
+ try:
+ call(cmd)
+ self.commands.append(" ".join(cmd))
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise;
+
+
+ def _threed_rotate_cmd (self):
+ #3rd step--run 3Drotate
+ cmd = [THREEDROTATE]
+ if self.params.swing: cmd += ["pan={}".format(self.params.swing)]
+ if self.params.tilt: cmd += ["tilt={}".format(self.params.tilt)]
+ if self.params.roll: cmd += ["roll={}".format(self.params.roll)]
+ if self.params.zoom: cmd += ["zoom={}".format(self.params.zoom)]
+ if cmd == [THREEDROTATE]: #if nothing has been added
+ return
+ if self.params.planebgcolor and not self.params.planebgimage:
+ cmd += ["bgcolor={}".format(self.params.planebgcolor)]
+ else:
+ cmd += ["bgcolor=none"]
+ cmd += ["skycolor={}".format(self.params.skycolor or 'none')]
+ if self.params.transition: cmd += ["vp={}",self.params.transition]
+ cmd += [ self.filepath, self.filepath ]
+ try:
+ call(cmd)
+ self.commands.append(" ".join(cmd))
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise
+
+ def _trim_cmd (self):
+ if self.params.trim:
+ cmd = [BIN_CONVERT, self.filepath, "-trim", "+repage", self.filepath]
+ try:
+ call(cmd)
+ self.commands.append(" ".join(cmd))
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise
+
+ def _prepare_gridimage(self, image):
+ if image['mimetype'] != 'png':
+ cmd = [BIN_CONVERT, image['path'], self.filepath]
+ try:
+ call(cmd)
+ self.commands.append(" ".join(cmd))
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise
+ else:
+ cmd = ['cp', image['path'], self.filepath]
+ try:
+ call(cmd)
+ self.commands.append(" ".join(cmd))
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise
+
+ def _overlay_planebgimage(self):
+ cmd = [
+ BIN_COMPOSITE,
+ "-compose", "Dst_Over", "-gravity", "center",
+ self.params.planebgimage["path"],
+ self.filepath,
+ self.filepath
+ ]
+ try:
+ call(cmd)
+ self.commands.append(" ".join(cmd))
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise
+
+ def _cleanup(self):
+ if not len(self.files_created):
+ pass
+ cmd = ["rm", "-f"] + self.files_created
+ try:
+ call(cmd)
+ self.commands.append(" ".join(cmd))
+ except Exception as e:
+ sys.stderr.write(str(e))
+ raise
+
+ def create(self):
+ if self.params.bgimage:
+ self._prepare_gridimage(self.params.bgimage)
+ self._grid_command()
+ elif self.params.imageinstead:
+ self._prepare_gridimage(self.params.imageinstead)
+ else:
+ self._make_canvas()
+ self._grid_command()
+ if self.params.shadow: self._shadow_cmd()
+ self._threed_rotate_cmd()
+ if self.params.planebgimage: self._overlay_planebgimage()
+ if self.params.trim: self._trim_cmd()
+ self._cleanup()
+
+if __name__ == "__main__":
+ g = Imgrid({
+ 'bgimage' : 'http://i.asdf.us/im/1a/imBreak_1424909483_xx_abridged___.gif',
+ 'planebgimage' : 'http://i.imgur.com/FICZtph.png',
+ 'tilt' : '30',
+ 'spacing' : '30',
+ 'hlines' : 'true',
+ 'roll' : '30',
+ 'shadow' : 'true',
+ 'trim' : 'true'
+ })
+ g.create()
+ print g.now
+ print g.filepath
+ print g.commands
generated by cgit v1.2.3-70-g09d2 (git 2.51.0) at 2025-12-22 00:27:06 +0000