cleanup successful, still testing

1 files changed, 215 insertions, 0 deletions

diff --git a/pbserver.py b/pbserver.py
new file mode 100755
index 0000000..a0068e9
--- /dev/null
+++ b/pbserver.py
@@ -0,0 +1,215 @@
+#!/usr/bin/python2.7
+from bottle import route, run, post, request, static_file
+
+from pb.Gradient import Gradient
+from pb.Imgrid import Imgrid
+from pb.Breaker import Breaker
+from pb.Pattern import Pattern
+from pb.Generate import Generate
+from pb.Imlandscape import Imlandscape
+
+from pb.Config import AWS_SECRET_ACCESS_KEY, AWS_ACCESS_KEY_ID, BUCKET_NAME, BIN_IDENTIFY
+import pb.lib.Utils as utils
+
+import os
+import sys
+from pb.lib.Db import Db
+
+import sha
+from subprocess import call, Popen, PIPE
+import simplejson as json
+
+from boto.s3.connection import S3Connection
+from boto.s3.key import Key
+#
+try:
+    db = Db(); 
+except Exception as e:
+    sys.stderr.write("Could not connect to db:\n{}".format(e))
+    sys.exit(1);
+BASE_URL = "http://i.asdf.us"
+
+def hashdir(filename):
+    return  sha.new(filename).hexdigest()[:2]
+
+def bin_identify (filepath):
+  ident = Popen([BIN_IDENTIFY, filepath], stdout=PIPE).communicate()[0]
+  partz = ident.split(" ")
+  width,height = partz[2].split("x")
+  return [ width, height ]
+
+def cleanup(filepath):
+    try:
+        call(['rm', filepath])
+    except Exception as e:
+        sys.stderr.write(str(e))
+        raise 
+
+def s3move(filename,objectname):
+    try:
+        conn = S3Connection(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, is_secure=False)
+        b = conn.get_bucket(BUCKET_NAME)
+        k = Key(b)
+        k.key = objectname
+        k.set_contents_from_filename(filename)
+        k.set_acl('public-read')
+        k.storage_class = 'REDUCED_REDUNDANCY'
+    except Exception as e:
+        sys.stderr.write(str(e));
+        raise(e)
+
+def format_im_data(im, insert_url="NULL"):
+    directory = hashdir(im.filename)
+    dimensions = bin_identify(im.filepath)
+    size = utils.file_size(im.filepath)
+    objectname = "im/{}/{}".format(directory, im.filename)
+    try:
+        s3move(im.filepath, objectname)
+        cleanup(im.filepath)
+        db.insert_cmd(
+            date=im.now, 
+            remote_addr=request.environ.get('REMOTE_ADDR', "NULL"),
+            username=im.params.get('username', "NULL"),
+            url=insert_url,
+            directory=directory,
+            oldfile="NULL",
+            newfile=im.filename, 
+            dataobj=";".join(im.commands),
+            cmd=json.dumps(im.params),
+            tag=im.tag,
+        )
+        return json.dumps({
+            'url' : "{}/{}".format(BASE_URL, objectname),
+            'size' : size,
+            'width' : "{}px".format(dimensions[0]),
+            'height' : "{}px".format(dimensions[1]),
+        })
+    except Exception as e:
+        sys.stderr.write(str(e))
+        raise;
+
+
+def return_image(im, insert_url="NULL"):
+  return format_im_data(im, insert_url)
+
+
+def return_jsonp(im, insert_url="NULL"):    
+  return "{}({})".format(im.get("callback"), format_im_data(im, insert_url))
+
+
+@post('/im/api/imgradient')
+def gradient():
+    try:
+        im = Gradient(**(dict(request.forms)))
+        im.create();
+        return return_image(im)
+    except Exception as e:
+        sys.stderr.write("imgradient failure\n")
+        sys.stderr.write("params:\n")
+        for i in request.forms:
+            sys.stderr.write("{}:{}\n".format(i, request.forms[i]))
+        raise;
+        return json.dumps({ 'error' : 'Request could not be processed' })
+
+@post('/im/api/imgrid')
+def imgrid():
+    try:
+        im = Imgrid(**(dict(request.forms)))
+        im.create();
+        url= "NULL"
+        for elem in [ im.params.imageinstead , im.params.bgimage, im.params.planebgimage ]:
+            if elem:
+                url = elem['url']
+                break
+        return return_image(im, url)
+    except Exception as e:
+        sys.stderr.write(str(e))
+        return json.dumps({ 'error' : 'Request could not be processed' })
+
+@post('/im/api/generate')
+def generate():
+    try:
+        im = Generate(**(dict(request.forms)))
+        im.create();
+        return return_image(im)
+    except Exception as e:
+        sys.stderr.write(str(e))
+        return json.dumps({ 'error' : 'Request could not be processed' })
+
+@post('/im/api/imbreak')
+def breaker():
+    try:
+        im = Breaker(**(dict(request.forms)))
+        im.create();
+        return return_image(im, im.params['url'])
+    except Exception as e:
+        sys.stderr.write(str(e))
+        return json.dumps({ 'error' : 'Request could not be processed' })
+
+@post('/im/api/impattern')
+def pattern():
+    try:
+        im = Pattern(**(dict(request.forms)))
+        im.create();
+        return return_image(im, im.params['image_url'])
+    except Exception as e:
+        sys.stderr.write(str(e)+"\n")
+        for i in request.forms:
+            sys.stderr.write("{}:{}\n".format(i, request.forms[i]))
+        raise;
+        return json.dumps({ 'error' : 'Request could not be processed' })
+
+@post('/im/api/imlandscape')
+def imlandscape():
+    try:
+        im = Imlandscape(**(dict(request.forms)))
+        im.create();
+        sys.stderr.write(str(im.params))
+        return return_image(im, im.params['texture'])
+    except Exception as e:
+        sys.stderr.write(str(e))
+        return json.dumps({ 'error' : 'Request could not be processed' })
+
+
+
+#static routes
+@route('/im/<filename>')
+def server_static(filename):
+    return static_file(filename, root='frontend/im/')
+@route('/im')
+def server_static():
+    return static_file("index.html", root='frontend/im/')
+@route('/imgrid')
+def server_static():
+    return static_file("index.html", root='frontend/imgrid/')
+@route('/imgradient')
+def server_static():
+    return static_file("index.html", root='frontend/imgradient/')
+@route('/imlandscape')
+def server_static():
+    return static_file("index.html", root='frontend/imlandscape/')
+@route('/impattern')
+def server_static():
+    return static_file("index.html", root='frontend/impattern/')
+@route('/imbreak')
+def server_static():
+    return static_file("index.html", root='frontend/imbreak/')
+@route('/')
+def server_static():
+    return static_file("index.html", root='frontend/im/')
+@route('/css/<filename>')
+def server_static(filename):
+    return static_file(filename, root='frontend/css/')
+@route('/js/<filename>')
+def server_static(filename):
+    return static_file(filename, root='frontend/js/')
+@route('/img/<filename>')
+def server_static(filename):
+    return static_file(filename, root='frontend/img/')
+
+
+#run(host='0.0.0.0', server='flup', port=8999, debug=True)
+run(host='0.0.0.0',  port=8999, debug=True)
+
+
+