a |f@sddlZddlZddlmZmZddlmZddlmZddZ ddZ d d Z d d Z Gd dde ZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZeddZGdd d eZdS)!N)urlsafe_b64decodeurlsafe_b64encode) namedtuple)MutableMappingcCs,t|ts|d}t|}|ddS)Nutf-8=) isinstancebytesencoderdecoderstrip)payloadr rID:\Projects\storyit_web\backend\venv\Lib\site-packages\jwcrypto/common.pybase64url_encode s  rcCsNt|d}|dkr|d7}n"|dkr0|d7}n|dkr@tdt|dS) Nz==rrzInvalid base64 stringr)len ValueErrorrr )r sizerrrbase64url_decodes   rcCs$t|tr|d}tj|dddS)Nr),:T) separators sort_keys)rr r jsondumpsstringrrr json_encode!s  r cCst|tr|d}t|S)Nr)rr r rloadsrrrr json_decode's  r"c@s eZdZdS) JWExceptionN)__name__ __module__ __qualname__rrrrr#-sr#cseZdZdfdd ZZS)InvalidJWAAlgorithmNcs(d}|r|d|7}tt||dS)NzInvalid JWA Algorithm namez (%s))superr'__init__selfmessagemsg __class__rrr)2s zInvalidJWAAlgorithm.__init__)N)r$r%r&r) __classcell__rrr.rr'1sr'cs eZdZdZfddZZS)InvalidCEKeyLengthzInvalid CEK Key Length. This exception is raised when a Content Encryption Key does not match the required length. cs d||f}tt||dS)Nz&Expected key of length %d bits, got %d)r(r1r)r+expectedZobtainedr-r.rrr)@s zInvalidCEKeyLength.__init__r$r%r&__doc__r)r0rrr.rr19sr1cs"eZdZdZdfdd ZZS)InvalidJWEOperationzInvalid JWS Object. This exception is raised when a requested operation cannot be execute due to unsatisfied conditions. Ncs:d}|r|}nd}|r&|dt|7}tt||dSNzUnknown Operation Failurez {%s})reprr(r6r)r+r, exceptionr-r.rrr)LszInvalidJWEOperation.__init__)NNr4rrr.rr6Esr6cs eZdZdZfddZZS)InvalidJWEKeyTypezInvalid JWE Key Type. This exception is raised when the provided JWK Key does not match the type required by the specified algorithm. cs d||f}tt||dS)NzExpected key type %s, got %s)r(r;r)r2r.rrr)^s zInvalidJWEKeyType.__init__r4rrr.rr;Wsr;cs eZdZdZfddZZS)InvalidJWEKeyLengthzInvalid JWE Key Length. This exception is raised when the provided JWK Key does not match the length required by the specified algorithm. cs d||f}tt||dS)Nz!Expected key of length %d, got %d)r(r<r)r2r.rrr)js zInvalidJWEKeyLength.__init__r4rrr.rr<csr<cs"eZdZdZdfdd ZZS)InvalidJWSERegOperationzInvalid JWSE Header Registry Operation. This exception is raised when there is an error in trying to add a JW Signature or Encryption header to the Registry. Ncs:d}|r|}nd}|r&|dt|7}tt||dSr7)r8r(r=r)r9r.rrr)vsz InvalidJWSERegOperation.__init__)NNr4rrr.rr=osr=cs"eZdZdZdfdd ZZS) JWKeyNotFoundzThe key needed to complete the operation was not found. This exception is raised when a JWKSet is used to perform some operation and the key required to successfully complete the operation is not found. Ncs"|r |}nd}tt||dS)Nz Key Not Found)r(r>r)r*r.rrr)szJWKeyNotFound.__init__)Nr4rrr.rr>sr> Parameterz*description mustprotect supported check_fnc@sFeZdZdddZddZddZdd Zd d Zd d ZddZ dS)JWSEHeaderRegistryNcCs:|r&t|trt||_q,tdni|_t|dS)NzUnknown input type)rdictcopydeepcopy _registryr=rr))r+Z init_registryrrrr)s   zJWSEHeaderRegistry.__init__cCs<||jvrtd||j|}|jdur.dS||SdS)Nz No header "%s" found in registryT)rDr=Zcheck_fn)r+hvalueparamrrr check_headers   zJWSEHeaderRegistry.check_headercCs |j|SN)rD __getitem__r+keyrrrrJszJWSEHeaderRegistry.__getitem__cCs |jSrI)rD__iter__r+rrrrMszJWSEHeaderRegistry.__iter__cCs2|j|js|j|jr"tdn |j|dS)Nz-Unable to delete protected or supported field)rD mustprotect supportedr= __delitem__rKrrrrQs    zJWSEHeaderRegistry.__delitem__cCsP||jvrB|j|}|jr$tdn|jr:|js:tdn|j|=||j|<dS)Nz+Supported header already exists in registryz,Header specified should bea protected header)rDrPr=rO)r+rEZjwse_header_paramprrr __setitem__s     zJWSEHeaderRegistry.__setitem__cCs |jSrI)rD__len__rNrrrrTszJWSEHeaderRegistry.__len__)N) r$r%r&r)rHrJrMrQrSrTrrrrr@s  r@)rBrbase64rr collectionsrcollections.abcrrrr r" Exceptionr#r'r1r6r;r<r=r>ZJWSEHeaderParameterr@rrrrs(