mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-19 02:09:38 +01:00
90 lines
2.3 KiB
Python
90 lines
2.3 KiB
Python
from rezag.inputrequest import DirectWSGIInputRequest, POSTInputRequest
|
|
from bottle import route, request, response, default_app, abort
|
|
import bottle
|
|
|
|
import traceback
|
|
import json
|
|
|
|
JSON_CT = 'application/json; charset=utf-8'
|
|
|
|
def err_handler(exc):
|
|
response.status = exc.status_code
|
|
response.content_type = JSON_CT
|
|
err_msg = json.dumps({'message': exc.body})
|
|
response.headers['ResErrors'] = err_msg
|
|
return err_msg
|
|
|
|
|
|
def wrap_error(func):
|
|
def wrap_func(*args, **kwargs):
|
|
try:
|
|
res, errs = func(*args, **kwargs)
|
|
|
|
if res:
|
|
if errs:
|
|
response.headers['ResErrors'] = json.dumps(errs)
|
|
return res
|
|
|
|
last_exc = errs.pop('last_exc', None)
|
|
if last_exc:
|
|
if bottle.debug:
|
|
traceback.print_exc()
|
|
|
|
response.status = last_exc.status()
|
|
message = last_exc.msg
|
|
else:
|
|
response.status = 404
|
|
message = 'No Resource Found'
|
|
|
|
response.content_type = JSON_CT
|
|
res = {'message': message}
|
|
if errs:
|
|
res['errors'] = errs
|
|
|
|
err_msg = json.dumps(res)
|
|
response.headers['ResErrors'] = err_msg
|
|
return err_msg
|
|
|
|
except Exception as e:
|
|
if bottle.debug:
|
|
traceback.print_exc()
|
|
abort(500, 'Internal Error: ' + str(e))
|
|
|
|
return wrap_func
|
|
|
|
|
|
route_dict = {}
|
|
|
|
def add_route(path, handler):
|
|
@route([path, path + '/<mode:path>'], 'ANY')
|
|
@wrap_error
|
|
def direct_input_request(mode=''):
|
|
params = dict(request.query)
|
|
params['mode'] = mode
|
|
params['_input_req'] = DirectWSGIInputRequest(request.environ)
|
|
return handler(params)
|
|
|
|
@route([path + '/postreq', path + '/<mode:path>/postreq'], 'POST')
|
|
@wrap_error
|
|
def post_fullrequest(mode=''):
|
|
params = dict(request.query)
|
|
params['mode'] = mode
|
|
params['_input_req'] = POSTInputRequest(request.environ)
|
|
return handler(params)
|
|
|
|
global route_dict
|
|
handler_dict = handler.get_supported_modes()
|
|
route_dict[path] = handler_dict
|
|
route_dict[path + '/postreq'] = handler_dict
|
|
|
|
|
|
@route('/')
|
|
def list_routes():
|
|
return route_dict
|
|
|
|
|
|
application = default_app()
|
|
application.default_error_handler = err_handler
|
|
|
|
|