1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
from flask import Flask
from flask import abort, redirect, url_for, request, jsonify
from cherrypy import wsgiserver
import sys, os
sys.path.append("./lib")
from pb import *
from config import SERVER_HOST, SERVER_PORT
class InvalidUsage(Exception):
status_code = 400
def __init__(self, message, status_code=None, payload=None):
Exception.__init__(self)
self.message = message
if status_code is not None:
self.status_code = status_code
self.payload = payload
def to_dict(self):
rv = dict(self.payload or ())
rv['message'] = self.message
return rv
class Server(object):
def __init__(self):
self.app = Flask(__name__)
self._wsgi_server = None
@self.app.route('/test', methods=['GET'])
def test():
return "HELLO WORLD!"
@self.app.route('/<pb_classname>', methods=['POST'])
def pb(pb_classname):
return self._response_post(pb_classname, request.form.to_dict())
@self.app.errorhandler(InvalidUsage)
def handle_invalid_usage(error):
response = jsonify(error.to_dict())
response.status_code = error.status_code
return response
self._classname_aliases = {
'generate' : 'PbGenerate',
'imgrid' : 'PbGrid',
'imbreak' : 'PbBreak',
'impattern' : 'PbPattern',
'imgradient' : 'PbGradient',
'landscape' : 'PbLandscape',
}
def _find_class_by_name(self, pb_classname):
pb_classname = self._classname_aliases.get(pb_classname, None) or pb_classname
try:
return filter(lambda c: c.__name__ == pb_classname, Pb.__subclasses__())[0]
except IndexError:
raise InvalidUsage('No such api', status_code=410)
def _response_post(self, pb_classname, request_form):
pb_class = self._find_class_by_name(pb_classname)
classnames = map(lambda c: c.__name__, Pb.__subclasses__())
try:
pb = pb_class(**request_form)
pb.create();
pb.file_s3move()
pb.db_send();
return jsonify(pb.file_dict());
#FIXME handle BadParamsError and PbError separately
except Exception as e:
sys.stderr.write("%s failure" % pb_class.__name__)
sys.stderr.write("params:\n")
sys.stderr.write(str(e))
for i in request_form.keys():
sys.stderr.write("{}:{}\n".format(i, request_form[i]))
raise;
return jsonify({ 'error' : 'Request could not be processed' })
def run(self):
self.app.run()
def run_wsgi(self, server_port=SERVER_PORT, host=SERVER_HOST):
d = wsgiserver.WSGIPathInfoDispatcher({'/': self.app})
sys.stderr.write("Starting a wsgi server on %s port %s\n" % (host, server_port))
self._wsgi_server = wsgiserver.CherryPyWSGIServer(('0.0.0.0', server_port), d)
self._wsgi_server.start()
def stop(self):
if self._wsgi_server:
self._wsgi_server.stop()
self._wsgi_server = None
return
else:
return
|
