fix: add typing to pyramid_jwt

cubicweb_api/jwt_policy.py

@@ -36,6 +36,7 @@
 from pyramid.authentication import CallbackAuthenticationPolicy
 from pyramid.interfaces import IAuthenticationPolicy, IRendererFactory
 from pyramid.request import Request
+from pyramid.response import Response
 
 log = logging.getLogger(__name__)
 marker = []
@@ -268,7 +269,7 @@
             **kwargs
         )
 
-    def _get_cookies(self, request, value, max_age=None, domains=None):
+    def _get_cookies(self, request: Request, value, max_age=None, domains=None):
         profile = self.cookie_profile(request)
         if domains is None:
             domains = [request.domain]
@@ -280,7 +281,7 @@
         headers = profile.get_headers(value, **kw)
         return headers
 
-    def remember(self, request, principal, **kw):
+    def remember(self, request: Request, principal, **kw):
         token = self.create_token(principal, self.expiration, self.audience, **kw)
 
         if hasattr(request, "_jwt_cookie_reissued"):
@@ -290,7 +291,7 @@
 
         return self._get_cookies(request, token, self.max_age, domains=domains)
 
-    def forget(self, request):
+    def forget(self, request: Request):
         request._jwt_cookie_reissue_revoked = True
         return self._get_cookies(request, None)
 
@@ -294,7 +295,7 @@
         request._jwt_cookie_reissue_revoked = True
         return self._get_cookies(request, None)
 
-    def get_claims(self, request):
+    def get_claims(self, request: Request):
         profile = self.cookie_profile.bind(request)
         cookie = profile.get_value()
 
@@ -309,7 +310,7 @@
             self._handle_reissue(request, claims)
         return claims
 
-    def _handle_reissue(self, request, claims):
+    def _handle_reissue(self, request: Request, claims: dict):
         if not request or not claims:
             raise ValueError("Cannot handle JWT reissue: insufficient arguments")
 
@@ -331,6 +332,6 @@
         )
         headers = self.remember(request, principal, **extra_claims)
 
-        def reissue_jwt_cookie(request, response):
-            if not hasattr(request, "_jwt_cookie_reissue_revoked"):
+        def reissue_jwt_cookie(inner_request: Request, inner_response: Response):
+            if not hasattr(inner_request, "_jwt_cookie_reissue_revoked"):
                 for k, v in headers:
@@ -336,5 +337,5 @@
                 for k, v in headers:
-                    response.headerlist.append((k, v))
+                    inner_response.headerlist.append((k, v))
 
         request.add_response_callback(reissue_jwt_cookie)
         request._jwt_cookie_reissued = True