MCPcopy
hub / github.com/encode/starlette / decorator

Function decorator

starlette/authentication.py:32–96  ·  starlette/authentication.py::decorator
(
        func: Callable[_P, Any],
    )

Source from the content-addressed store, hash-verified

30 scopes_list = [scopes] if isinstance(scopes, str) else list(scopes)
31
32 def decorator(
33 func: Callable[_P, Any],
34 ) -> Callable[_P, Any]:
35 sig = inspect.signature(func)
36 for idx, parameter in enumerate(sig.parameters.values()):
37 if parameter.name == class="st">"request" or parameter.name == class="st">"websocket":
38 type_ = parameter.name
39 break
40 else:
41 raise Exception(f&class="cm">#x27;No class="st">"request" or class="st">"websocket" argument on function class="st">"{func}"')
42
43 if type_ == class="st">"websocket":
44 class="cm"># Handle websocket functions. (Always async)
45 @functools.wraps(func)
46 async def websocket_wrapper(*args: _P.args, **kwargs: _P.kwargs) -> None:
47 websocket = kwargs.get(class="st">"websocket", args[idx] if idx < len(args) else None)
48 assert isinstance(websocket, WebSocket), (
49 class="st">"Parameter with name &class="cm">#x27;websocket' is required to be of type 'WebSocket'"
50 fclass="st">" not &class="cm">#x27;{type(websocket).__name__}'"
51 )
52
53 if not has_required_scope(websocket, scopes_list):
54 await websocket.close()
55 else:
56 await func(*args, **kwargs)
57
58 return websocket_wrapper
59
60 elif is_async_callable(func):
61 class="cm"># Handle async request/response functions.
62 @functools.wraps(func)
63 async def async_wrapper(*args: _P.args, **kwargs: _P.kwargs) -> Any:
64 request = kwargs.get(class="st">"request", args[idx] if idx < len(args) else None)
65 assert isinstance(request, Request), (
66 fclass="st">"Parameter with name &class="cm">#x27;request' is required to be of type 'Request' not '{type(request).__name__}'"
67 )
68
69 if not has_required_scope(request, scopes_list):
70 if redirect is not None:
71 orig_request_qparam = urlencode({class="st">"next": str(request.url)})
72 next_url = fclass="st">"{request.url_for(redirect)}?{orig_request_qparam}"
73 return RedirectResponse(url=next_url, status_code=303)
74 raise HTTPException(status_code=status_code)
75 return await func(*args, **kwargs)
76
77 return async_wrapper
78
79 else:
80 class="cm"># Handle sync request/response functions.
81 @functools.wraps(func)
82 def sync_wrapper(*args: _P.args, **kwargs: _P.kwargs) -> Any:
83 request = kwargs.get(class="st">"request", args[idx] if idx < len(args) else None)
84 assert isinstance(request, Request), (
85 fclass="st">"Parameter with name &class="cm">#x27;request' is required to be of type 'Request' not '{type(request).__name__}'"
86 )
87
88 if not has_required_scope(request, scopes_list):
89 if redirect is not None:

Callers

nothing calls this directly

Calls 2

is_async_callableFunction · 0.90
valuesMethod · 0.45

Tested by

no test coverage detected