diff options
Diffstat (limited to 'pbserver.py')
| -rwxr-xr-x | pbserver.py | 215 |
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) + + + |