1414from cryptojwt .jwk .hmac import SYMKey
1515from cryptojwt .jwk .jwk import key_from_jwk_dict
1616from cryptojwt .jws .hmac import HMACSigner
17+ from cryptojwt .key_bundle import init_key , import_jwk
1718from cryptojwt .utils import as_bytes
1819from cryptojwt .utils import as_unicode
1920from cryptojwt .utils import b64e
@@ -105,7 +106,7 @@ def ver_dec_content(parts, sign_key=None, enc_key=None, sign_alg="SHA256"):
105106 mac = base64 .b64decode (b64_mac )
106107 verifier = HMACSigner (algorithm = sign_alg )
107108 if verifier .verify (
108- load .encode ("utf-8" ) + timestamp .encode ("utf-8" ), mac , sign_key .key
109+ load .encode ("utf-8" ) + timestamp .encode ("utf-8" ), mac , sign_key .key
109110 ):
110111 return load , timestamp
111112 else :
@@ -124,9 +125,9 @@ def ver_dec_content(parts, sign_key=None, enc_key=None, sign_alg="SHA256"):
124125 if len (p ) == 3 :
125126 verifier = HMACSigner (algorithm = sign_alg )
126127 if verifier .verify (
127- load .encode ("utf-8" ) + timestamp .encode ("utf-8" ),
128- base64 .b64decode (p [2 ]),
129- sign_key .key ,
128+ load .encode ("utf-8" ) + timestamp .encode ("utf-8" ),
129+ base64 .b64decode (p [2 ]),
130+ sign_key .key ,
130131 ):
131132 return load , timestamp
132133 else :
@@ -135,15 +136,15 @@ def ver_dec_content(parts, sign_key=None, enc_key=None, sign_alg="SHA256"):
135136
136137
137138def make_cookie_content (
138- name ,
139- load ,
140- sign_key ,
141- domain = None ,
142- path = None ,
143- timestamp = "" ,
144- enc_key = None ,
145- max_age = 0 ,
146- sign_alg = "SHA256" ,
139+ name ,
140+ load ,
141+ sign_key ,
142+ domain = None ,
143+ path = None ,
144+ timestamp = "" ,
145+ enc_key = None ,
146+ max_age = 0 ,
147+ sign_alg = "SHA256" ,
147148):
148149 """
149150 Create and return a cookies content
@@ -196,15 +197,15 @@ def make_cookie_content(
196197
197198
198199def make_cookie (
199- name ,
200- payload ,
201- sign_key ,
202- domain = None ,
203- path = None ,
204- timestamp = "" ,
205- enc_key = None ,
206- max_age = 0 ,
207- sign_alg = "SHA256" ,
200+ name ,
201+ payload ,
202+ sign_key ,
203+ domain = None ,
204+ path = None ,
205+ timestamp = "" ,
206+ enc_key = None ,
207+ max_age = 0 ,
208+ sign_alg = "SHA256" ,
208209):
209210 content = make_cookie_content (
210211 name ,
@@ -285,13 +286,13 @@ class CookieDealer(object):
285286 """
286287
287288 def __init__ (
288- self ,
289- sign_key = "" ,
290- enc_key = "" ,
291- sign_alg = "SHA256" ,
292- default_values = None ,
293- sign_jwk = "" ,
294- enc_jwk = "" ,
289+ self ,
290+ sign_key = "" ,
291+ enc_key = "" ,
292+ sign_alg = "SHA256" ,
293+ default_values = None ,
294+ sign_jwk = None ,
295+ enc_jwk = None
295296 ):
296297
297298 if sign_key :
@@ -301,7 +302,7 @@ def __init__(
301302 self .sign_key = SYMKey (k = sign_key )
302303 elif sign_jwk :
303304 if isinstance (sign_jwk , dict ):
304- self .sign_key = import_jwk ( sign_jwk [ 'filename' ] )
305+ self .sign_key = init_key ( ** sign_jwk )
305306 else :
306307 self .sign_key = import_jwk (sign_jwk )
307308 else :
@@ -316,7 +317,7 @@ def __init__(
316317 self .enc_key = SYMKey (k = enc_key )
317318 elif enc_jwk :
318319 if isinstance (enc_jwk , dict ):
319- self .enc_key = import_jwk ( enc_jwk [ 'filename' ] )
320+ self .enc_key = init_key ( ** enc_jwk )
320321 else :
321322 self .enc_key = import_jwk (enc_jwk )
322323 else :
@@ -417,15 +418,15 @@ def get_cookie_value(self, cookie=None, cookie_name=None):
417418 return None
418419
419420 def append_cookie (
420- self ,
421- cookie ,
422- name ,
423- payload ,
424- typ ,
425- domain = None ,
426- path = None ,
427- timestamp = "" ,
428- max_age = 0 ,
421+ self ,
422+ cookie ,
423+ name ,
424+ payload ,
425+ typ ,
426+ domain = None ,
427+ path = None ,
428+ timestamp = "" ,
429+ max_age = 0 ,
429430 ):
430431 """
431432 Adds a cookie to a SimpleCookie instance
0 commit comments