File: //home/arjun/projects/env/lib/python3.10/site-packages/jwt/__pycache__/jwks_client.cpython-310.pyc
o
we~ � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZm Z m
Z
m Z d dl m
Z
ddlmZmZ ddlmZ dd lmZmZ dd
lmZ G d d � d �ZdS )
� N)� lru_cache)�
SSLContext)�Any�Dict�List�Optional)�URLError� )�PyJWK�PyJWKSet)�decode_complete)�PyJWKClientConnectionError�PyJWKClientError)� JWKSetCachec @ s� e Zd Z d"deded ed
ed ed eeeef d
edee fdd�Z
defdd�Z d#dede fdd�Z
d#dedee fdd�Zdedefdd�Zd edefd d �Zed ee dedee fd d!��ZdS )$� PyJWKClientF� T�, N� �uri�
cache_keys�max_cached_keys�
cache_jwk_set�lifespan�headers�timeout� ssl_contextc C sv |d u ri }|| _ d | _|| _|| _|| _|r)|dkr#td|� d���t|�| _nd | _|r9t|d�| j�| _d S d S )Nr z/Lifespan must be greater than 0, the input is "�")�maxsize) r �
jwk_set_cacher r r r r r �get_signing_key) �selfr r r r r r r r � r! �H/home/arjun/projects/env/lib/python3.10/site-packages/jwt/jwks_client.py�__init__ s"
� �zPyJWKClient.__init__�returnc
C s� d }zRz,t jj| j| jd�}t jj|| j| jd��
}t� |�}W d � n1 s)w Y W n t
t fyE } zt d|� d���d }~ww |W | j
d urT| j
�|� S S | j
d ura| j
�|� w w )N)�urlr )r �contextz'Fail to fetch data from the url, err: "r )�urllib�request�Requestr r �urlopenr r �json�loadr � TimeoutErrorr
r �put)r �jwk_set�r�response�er! r! r" �
fetch_data2 s,
� ��
���
��zPyJWKClient.fetch_data�refreshc C sH d }| j d ur|s| j �� }|d u r| �� }t|t�s td��t�|�S )Nz.The JWKS endpoint did not return a JSON object)r �getr3 �
isinstance�dictr r � from_dict)r r4 �datar! r! r" � get_jwk_setD s
zPyJWKClient.get_jwk_setc C s* | � |�}dd� |jD �}|std��|S )Nc S s g | ] }|j d v r|jr|�qS ))�sigN)�public_key_use�key_id)�.0� jwk_set_keyr! r! r" �
<listcomp>S s
�z0PyJWKClient.get_signing_keys.<locals>.<listcomp>z2The JWKS endpoint did not contain any signing keys)r: �keysr )r r4 r/ � signing_keysr! r! r" �get_signing_keysQ s
�z PyJWKClient.get_signing_keys�kidc C sH | � � }| �||�}|s"| j dd�}| �||�}|s"td|� d���|S )NT)r4 z,Unable to find a signing key that matches: "r )rC � match_kidr )r rD rB � signing_keyr! r! r" r ^ s
�zPyJWKClient.get_signing_key�tokenc C s( t |ddid�}|d }| �|�d��S )N�verify_signatureF)�options�headerrD )� decode_tokenr r5 )r rG �
unverifiedrJ r! r! r" �get_signing_key_from_jwtn s z$PyJWKClient.get_signing_key_from_jwtrB c C s& d }| D ] }|j |kr|} |S q|S )N)r= )rB rD rF �keyr! r! r" rE s s
�zPyJWKClient.match_kid)Fr Tr Nr N)F)�__name__�
__module__� __qualname__�str�bool�intr r r r r# r3 r r: r r
rC r rM � staticmethodrE r! r! r! r" r s@ ��������
�#
$r )r+ �urllib.requestr' � functoolsr �sslr �typingr r r r � urllib.errorr �api_jwkr
r �api_jwtr rK �
exceptionsr
r r r r r! r! r! r" �<module> s