From b2f1fa936e8709b73900093fb8c6fb7868290a12 Mon Sep 17 00:00:00 2001
From: Robohash <401330+e1ven@users.noreply.github.com>
Date: Thu, 26 Jul 2012 04:56:31 -0400
Subject: [PATCH] datauri

---
 webfront.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/webfront.py b/webfront.py
index d4934b9..0c3b1d3 100755
--- a/webfront.py
+++ b/webfront.py
@@ -22,6 +22,7 @@ try:
    from hashlib import md5 as md5_func
 except ImportError:
    from md5 import new as md5_func
+import cStringIO
 
 define("port", default=80, help="run on the given port", type=int)
 
@@ -231,7 +232,7 @@ class ImgHandler(tornado.web.RequestHandler):
             
             
         #Change to a usuable format
-        if string.endswith(('.png','.gif','.jpg','.bmp','.jpeg','.ppm')):
+        if string.endswith(('.png','.gif','.jpg','.bmp','.jpeg','.ppm','.datauri')):
             ext = string[string.rfind('.') +1 :len(string)] 
             if ext.lower() == 'jpg':
                 ext = 'jpeg'            
@@ -372,7 +373,15 @@ class ImgHandler(tornado.web.RequestHandler):
             robohash = bg               
                            
         robohash = robohash.resize((sizex,sizey),Image.ANTIALIAS)    
-        robohash.save(self,format=ext)
+        if ext != 'datauri':
+          robohash.save(self,format=ext)
+        else:
+          fakefile = cStringIO.StringIO()
+          robohash.save(fakefile,format='jpeg')
+          fakefile.seek(0)
+          data_uri = fakefile.read().encode("base64").replace("\n", "")
+          self.write('<img alt="sample" src="data:image/png;base64,{0}">'.format(data_uri))
+