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

Add type information for the first two arguments of all endpoints

Browse Source
This commit is contained in:
Kovid Goyal 2015-06-15 10:51:45 +05:30
parent 52b16e4f4b
commit 5cf181bae6
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -41,6 +41,8 @@ def endpoint(route,
postprocess=None postprocess=None
): ):
from calibre.srv.handler import Context
from calibre.srv.http_response import RequestData
def annotate(f): def annotate(f):
f.route = route.rstrip('/') or '/' f.route = route.rstrip('/') or '/'
f.route_key = route_key(f.route) f.route_key = route_key(f.route)
@ -51,6 +53,17 @@ def endpoint(route,
f.cache_control = cache_control f.cache_control = cache_control
f.postprocess = postprocess f.postprocess = postprocess
f.is_endpoint = True f.is_endpoint = True
argspec = inspect.getargspec(f)
if len(argspec.args) < 2:
raise TypeError('The endpoint %r must take at least two arguments' % f.route)
f.__annotations__ = {
argspec.args[0]: Context,
argspec.args[1]: RequestData,
}
f.__doc__ = (f.__doc__ or '') + '\n\n' + (
(':type %s: calibre.srv.handler.Context\n' % argspec.args[0]) +
(':type %s: calibre.srv.http_response.RequestData\n' % argspec.args[1])
)
return f return f
return annotate return annotate