Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Allow a route to specify its OK code

Browse Source
This commit is contained in:
Kovid Goyal 2015-11-13 14:35:36 +05:30
parent 9f624b6d95
commit 363be7c6f3
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/calibre/srv/routes.py
View File

@ -39,6 +39,10 @@ def endpoint(route,
# Cache-Control header # Cache-Control header
cache_control=False, cache_control=False,
# The HTTP code to be used when no error occurs. By default it is
# 200 for GET and HEAD and 201 for POST
ok_code=None,
postprocess=None postprocess=None
): ):
from calibre.srv.handler import Context from calibre.srv.handler import Context
@ -52,6 +56,7 @@ def endpoint(route,
f.android_workaround = android_workaround f.android_workaround = android_workaround
f.cache_control = cache_control f.cache_control = cache_control
f.postprocess = postprocess f.postprocess = postprocess
f.ok_code = ok_code
f.is_endpoint = True f.is_endpoint = True
argspec = inspect.getargspec(f) argspec = inspect.getargspec(f)
if len(argspec.args) < 2: if len(argspec.args) < 2:
@ -259,6 +264,9 @@ class Router(object):
if endpoint_.auth_required and self.auth_controller is not None: if endpoint_.auth_required and self.auth_controller is not None:
self.auth_controller(data, endpoint_) self.auth_controller(data, endpoint_)
if endpoint_.ok_code is not None:
data.status_code = endpoint_.ok_code
self.init_session(endpoint_, data) self.init_session(endpoint_, data)
ans = endpoint_(self.ctx, data, *args) ans = endpoint_(self.ctx, data, *args)
self.finalize_session(endpoint_, data, ans) self.finalize_session(endpoint_, data, ans)