o T۷i @sddlmZddlZddlZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZmZddlmZGd d d ZGd d d ZGd ddZdS)) annotationsN)Iterator)Any)get_default_algorithms has_cryptorequires_cryptography)InvalidKeyErrorMissingCryptographyError PyJWKError PyJWKSetError PyJWTError)JWKDictc@sbeZdZdddd Zeddd d ZeddddZedddZedddZ edddZ dS)PyJWKNjwk_datar algorithm str | NonereturnNonecCsr||_|jdd}|std|j|s#t|jtr#|jdd}|s|jdd}|dkrU|dks6|s9d}nM|d kr@d }nF|d krGd }n?|d krNd}n8td||dkr\d}n*|dkrcd}n#|dkr|sqtd|j|dkrxd}ntd|td|ts|tvrt|d||_zt ||_ Wnt yt d|jdw|j |j|_dS)aA class that represents a `JSON Web Key `_. :param jwk_data: The decoded JWK data. :type jwk_data: dict[str, typing.Any] :param algorithm: The key algorithm. If not specified, the key's ``alg`` will be used. :type algorithm: str or None :raises InvalidKeyError: If the key type (``kty``) is not found or unsupported, or if the curve (``crv``) is not found or unsupported. :raises MissingCryptographyError: If the algorithm requires ``cryptography`` to be installed and it is not available. :raises PyJWKError: If unable to find an algorithm for the key. ktyNzkty is not found: algcrvECzP-256ES256zP-384ES384zP-521ES512 secp256k1ES256KzUnsupported crv: RSARS256octHS256OKPzcrv is not found: Ed25519EdDSAzUnsupported kty: z) requires 'cryptography' to be installed.z%Unable to find an algorithm for key: ) _jwk_datagetr isinstancedictrrr algorithm_namer AlgorithmKeyErrorr from_jwkkey)selfrrrrr/A/home/ubuntu/vllm_env/lib/python3.10/site-packages/jwt/api_jwk.py__init__sX     zPyJWK.__init__objcCs t||S)a;Creates a :class:`PyJWK` object from a JSON-like dictionary. :param obj: The JWK data, as a dictionary :type obj: dict[str, typing.Any] :param algorithm: The key algorithm. If not specified, the key's ``alg`` will be used. :type algorithm: str or None :rtype: PyJWK )r)r2rr/r/r0 from_dictTs zPyJWK.from_dictdatastrcCst|}t||S)aECreate a :class:`PyJWK` object from a JSON string. Implicitly calls :meth:`PyJWK.from_dict()`. :param str data: The JWK data, as a JSON string. :param algorithm: The key algorithm. If not specific, the key's ``alg`` will be used. :type algorithm: str or None :rtype: PyJWK )jsonloadsrr3)r4rr2r/r/r0 from_json`s zPyJWK.from_jsoncC|jddS)zFThe `kty` property from the JWK. :rtype: str or None rNr%r&r.r/r/r0key_typenzPyJWK.key_typecCr9)zFThe `kid` property from the JWK. :rtype: str or None kidNr:r;r/r/r0key_idvr=z PyJWK.key_idcCr9)zFThe `use` property from the JWK. :rtype: str or None useNr:r;r/r/r0public_key_use~r=zPyJWK.public_key_useN)rrrrrr)r2rrrrr)r4r5rrrr)rr) __name__ __module__ __qualname__r1 staticmethodr3r8propertyr<r?rAr/r/r/r0rs @   rc@sFeZdZdddZedd d Zedd dZdddZdddZdS)PyJWKSetkeys list[JWKDict]rrc Csg|_|s tdt|tstd|D]%}z |jt|Wqty9}z t|tr/|WYd}~qd}~wwt|jdkrEtddS)Nz$The JWK Set did not contain any keyszInvalid JWK Set valuerzUThe JWK Set did not contain any usable keys. Perhaps 'cryptography' is not installed?) rIr r'listappendrr r len)r.rIr-errorr/r/r0r1s&   zPyJWKSet.__init__r2dict[str, Any]cCs|dg}t|S)NrI)r&rH)r2rIr/r/r0r3s zPyJWKSet.from_dictr4r5cCst|}t|SrB)r6r7rHr3)r4r2r/r/r0r8s  zPyJWKSet.from_jsonr>rcCs,|jD] }|j|kr|Sqtd|)Nzkeyset has no key for kid: )rIr?r+)r.r>r-r/r/r0 __getitem__s  zPyJWKSet.__getitem__Iterator[PyJWK]cCs t|jSrB)iterrIr;r/r/r0__iter__s zPyJWKSet.__iter__N)rIrJrr)r2rOrrH)r4r5rrH)r>r5rr)rrQ) rCrDrEr1rFr3r8rPrSr/r/r/r0rHs    rHc@s*eZdZd ddZd ddZdd d Zd S)PyJWTSetWithTimestampjwk_setrHcCs||_t|_dSrB)rUtime monotonic timestamp)r.rUr/r/r0r1szPyJWTSetWithTimestamp.__init__rcC|jSrB)rUr;r/r/r0 get_jwk_setz!PyJWTSetWithTimestamp.get_jwk_setfloatcCrYrB)rXr;r/r/r0 get_timestampr[z#PyJWTSetWithTimestamp.get_timestampN)rUrH)rrH)rr\)rCrDrEr1rZr]r/r/r/r0rTs  rT) __future__rr6rVcollections.abcrtypingr algorithmsrrr exceptionsr r r r r typesrrrHrTr/r/r/r0s    t,