1313from idpyoidc .server .exception import ConfigurationError
1414from idpyoidc .server .session .grant_manager import GrantManager
1515from idpyoidc .util import rndstr
16-
17- from ..token import UnknownToken
18- from ..token import WrongTokenClass
19- from ..token import handler
20- from ..token .handler import TokenHandler
2116from .database import Database
2217from .grant import Grant
2318from .grant import SessionToken
2419from .info import ClientSessionInfo
2520from .info import UserSessionInfo
21+ from ..token import handler
22+ from ..token import UnknownToken
23+ from ..token import WrongTokenClass
24+ from ..token .handler import TokenHandler
2625
2726logger = logging .getLogger (__name__ )
2827
2928
3029class RawID (object ):
30+
3131 def __init__ (self , * args , ** kwargs ):
3232 pass
3333
@@ -42,14 +42,15 @@ def pairwise_id(uid, sector_identifier, salt="", **kwargs):
4242
4343
4444class PairWiseID (object ):
45+
4546 def __init__ (self , salt : Optional [str ] = "" , filename : Optional [str ] = "" ):
4647 if salt :
4748 self .salt = salt
4849 elif filename :
4950 if os .path .isfile (filename ):
5051 self .salt = open (filename ).read ()
5152 elif not os .path .isfile (filename ) and os .path .exists (
52- filename
53+ filename
5354 ): # Not a file, Something else
5455 raise ConfigurationError ("Salt filename points to something that is not a file" )
5556 else :
@@ -70,6 +71,7 @@ def public_id(uid, salt="", **kwargs):
7071
7172
7273class PublicID (PairWiseID ):
74+
7375 def __call__ (self , uid , sector_identifier , * args , ** kwargs ):
7476 return public_id (uid , self .salt )
7577
@@ -84,13 +86,13 @@ class SessionManager(GrantManager):
8486 init_args = ["token_handler_args" , "upstream_get" ]
8587
8688 def __init__ (
87- self ,
88- token_handler_args : dict ,
89- conf : Optional [dict ] = None ,
90- sub_func : Optional [dict ] = None ,
91- remember_token : Optional [Callable ] = None ,
92- remove_inactive_token : Optional [bool ] = False ,
93- upstream_get : Optional [Callable ] = None ,
89+ self ,
90+ token_handler_args : dict ,
91+ conf : Optional [dict ] = None ,
92+ sub_func : Optional [dict ] = None ,
93+ remember_token : Optional [Callable ] = None ,
94+ remove_inactive_token : Optional [bool ] = False ,
95+ upstream_get : Optional [Callable ] = None ,
9496 ):
9597 self .conf = conf or {"session_params" : {"encrypter" : default_crypt_config ()}}
9698 session_params = self .conf .get ("session_params" ) or {}
@@ -164,14 +166,14 @@ def make_path(self, **kwargs):
164166 return _path
165167
166168 def create_grant (
167- self ,
168- authn_event : AuthnEvent ,
169- auth_req : AuthorizationRequest ,
170- user_id : Optional [str ] = "" ,
171- client_id : Optional [str ] = "" ,
172- sub_type : Optional [str ] = "public" ,
173- token_usage_rules : Optional [dict ] = None ,
174- scopes : Optional [list ] = None ,
169+ self ,
170+ authn_event : AuthnEvent ,
171+ auth_req : AuthorizationRequest ,
172+ user_id : Optional [str ] = "" ,
173+ client_id : Optional [str ] = "" ,
174+ sub_type : Optional [str ] = "public" ,
175+ token_usage_rules : Optional [dict ] = None ,
176+ scopes : Optional [list ] = None ,
175177 ) -> str :
176178 """
177179
@@ -214,15 +216,15 @@ def create_grant(
214216 )
215217
216218 def create_exchange_grant (
217- self ,
218- exchange_request : TokenExchangeRequest ,
219- original_grant : Grant ,
220- original_session_id : str ,
221- user_id : str ,
222- client_id : Optional [str ] = "" ,
223- sub_type : Optional [str ] = "public" ,
224- token_usage_rules : Optional [dict ] = None ,
225- scopes : Optional [list ] = None ,
219+ self ,
220+ exchange_request : TokenExchangeRequest ,
221+ original_grant : Grant ,
222+ original_session_id : str ,
223+ user_id : str ,
224+ client_id : Optional [str ] = "" ,
225+ sub_type : Optional [str ] = "public" ,
226+ token_usage_rules : Optional [dict ] = None ,
227+ scopes : Optional [list ] = None ,
226228 ) -> str :
227229 """
228230
@@ -246,14 +248,14 @@ def create_exchange_grant(
246248 )
247249
248250 def create_session (
249- self ,
250- authn_event : AuthnEvent ,
251- auth_req : AuthorizationRequest ,
252- user_id : Optional [str ] = "" ,
253- client_id : Optional [str ] = "" ,
254- sub_type : Optional [str ] = "public" ,
255- token_usage_rules : Optional [dict ] = None ,
256- scopes : Optional [list ] = None ,
251+ self ,
252+ authn_event : AuthnEvent ,
253+ auth_req : AuthorizationRequest ,
254+ user_id : Optional [str ] = "" ,
255+ client_id : Optional [str ] = "" ,
256+ sub_type : Optional [str ] = "public" ,
257+ token_usage_rules : Optional [dict ] = None ,
258+ scopes : Optional [list ] = None ,
257259 ) -> str :
258260 """
259261 Create part of a user session. The parts added are user- and client
@@ -280,15 +282,15 @@ def create_session(
280282 )
281283
282284 def create_exchange_session (
283- self ,
284- exchange_request : TokenExchangeRequest ,
285- original_grant : Grant ,
286- original_session_id : str ,
287- user_id : str ,
288- client_id : Optional [str ] = "" ,
289- sub_type : Optional [str ] = "public" ,
290- token_usage_rules : Optional [dict ] = None ,
291- scopes : Optional [list ] = None ,
285+ self ,
286+ exchange_request : TokenExchangeRequest ,
287+ original_grant : Grant ,
288+ original_session_id : str ,
289+ user_id : str ,
290+ client_id : Optional [str ] = "" ,
291+ sub_type : Optional [str ] = "public" ,
292+ token_usage_rules : Optional [dict ] = None ,
293+ scopes : Optional [list ] = None ,
292294 ) -> str :
293295 """
294296 Create part of a user session. The parts added are user- and client
@@ -376,10 +378,10 @@ def revoke_token(self, session_id: str, token_value: str, recursive: bool = Fals
376378 grant .revoke_token (value = token .value )
377379
378380 def get_authentication_events (
379- self ,
380- session_id : Optional [str ] = "" ,
381- user_id : Optional [str ] = "" ,
382- client_id : Optional [str ] = "" ,
381+ self ,
382+ session_id : Optional [str ] = "" ,
383+ user_id : Optional [str ] = "" ,
384+ client_id : Optional [str ] = "" ,
383385 ) -> List [AuthnEvent ]:
384386 """
385387 Return the authentication events that exists for a user/client combination.
@@ -453,13 +455,13 @@ def revoke_grant(self, session_id: str):
453455 # return [self.get([user_id, client_id, gid]) for gid in _csi.subordinate]
454456
455457 def get_session_info (
456- self ,
457- session_id : str ,
458- user_session_info : bool = False ,
459- client_session_info : bool = False ,
460- grant : bool = False ,
461- authentication_event : bool = False ,
462- authorization_request : bool = False ,
458+ self ,
459+ session_id : str ,
460+ user_session_info : bool = False ,
461+ client_session_info : bool = False ,
462+ grant : bool = False ,
463+ authentication_event : bool = False ,
464+ authorization_request : bool = False ,
463465 ) -> dict :
464466 """
465467 Returns information connected to a session.
@@ -484,14 +486,14 @@ def get_session_info(
484486 return res
485487
486488 def get_session_info_by_token (
487- self ,
488- token_value : str ,
489- user_session_info : Optional [bool ] = False ,
490- client_session_info : Optional [bool ] = False ,
491- grant : Optional [bool ] = False ,
492- authentication_event : Optional [bool ] = False ,
493- authorization_request : Optional [bool ] = False ,
494- handler_key : Optional [str ] = "" ,
489+ self ,
490+ token_value : str ,
491+ user_session_info : Optional [bool ] = False ,
492+ client_session_info : Optional [bool ] = False ,
493+ grant : Optional [bool ] = False ,
494+ authentication_event : Optional [bool ] = False ,
495+ authorization_request : Optional [bool ] = False ,
496+ handler_key : Optional [str ] = "" ,
495497 ) -> dict :
496498
497499 if handler_key :
@@ -537,7 +539,5 @@ def encrypted_session_id(self, *args):
537539 def unpack_session_key (self , key ):
538540 return self .unpack_branch_key (key )
539541
540-
541-
542- def create_session_manager (upstream_get , token_handler_args , sub_func = None , conf = None ):
543- return SessionManager (_token_handler , sub_func = sub_func , conf = conf )
542+ # def create_session_manager(upstream_get, token_handler_args, sub_func=None, conf=None):
543+ # return SessionManager(token_handler_args, sub_func=sub_func, conf=conf, upstream_get=upstream_get)
0 commit comments