File: //usr/lib/python3/dist-packages/jwt/__pycache__/api_jws.cpython-38.pyc
U
�b�[� � @ s� d dl Z d dlZd dlZz d dlmZmZmZmZmZ W n e k
rL Y nX ddl
m Z m Z m
Z
mZ ddlmZmZmZmZ ddlmZmZmZmZ ddlmZmZm Z m Z G dd � d e �Z e � Z e j!Z!e j"Z"e j#Z#e j$Z$e j%Z%dS )
� N)�Callable�Dict�List�Optional�Union� )� Algorithm�get_default_algorithms�
has_crypto�requires_cryptography)�Mapping� binary_type� string_types� text_type)� DecodeError�InvalidAlgorithmError�InvalidSignatureError�InvalidTokenError)�base64url_decode�base64url_encode� force_bytes�
merge_dictc @ s| e Zd ZdZd dd�Zedd� �Zdd� Zd d
� Zd d � Z d dd�Z
d dd�Z dd� Z dd� Z
d!dd�Zdd� Zd d � ZdS )"�PyJWSZJWTNc C sf t � | _|d k rt|�nt| j�| _t| j�� �D ]}|| jkr2| j|= q2|sRi }t| �� |�| _d S )N) r � _algorithms�set� _valid_algs�list�keysr �_get_default_options�options)�self�
algorithmsr �key� r# �-/usr/lib/python3/dist-packages/jwt/api_jws.py�__init__ s �
zPyJWS.__init__c C s ddiS )N�verify_signatureTr# r# r# r# r$ r ' s �zPyJWS._get_default_optionsc C s> || j krtd��t|t�s$td��|| j |< | j�|� dS )zW
Registers a new Algorithm for use when creating and verifying tokens.
z Algorithm already has a handler.z!Object is not of type `Algorithm`N)r �
ValueError�
isinstancer � TypeErrorr �add)r �alg_id�alg_objr# r# r$ �register_algorithm- s
zPyJWS.register_algorithmc C s* || j krtd��| j |= | j�|� dS )z�
Unregisters an Algorithm for use when creating and verifying tokens
Throws KeyError if algorithm is not registered.
zJThe specified algorithm could not be removed because it is not registered.N)r �KeyErrorr �remove)r r+ r# r# r$ �unregister_algorithm: s
zPyJWS.unregister_algorithmc C s
t | j�S )zM
Returns a list of supported values for the 'alg' parameter.
)r r )r r# r# r$ �get_algorithmsF s zPyJWS.get_algorithms�HS256c C s� g }|d krd}|| j kr| j|d�}|r>| �|� |�|� ttj|d|d��}|�t|�� |�t|�� d� |�} z$| j
| }
|
� |�}|
� | |�} W n6 t
k
r� ts�|tkr�td| ��ntd��Y nX |�t| �� d� |�S )NZnone)�typ�alg)�,�:)Z
separators�cls� .zFAlgorithm '%s' could not be found. Do you have cryptography installed?�Algorithm not supported)r �
header_typ�_validate_headers�updater �json�dumps�appendr �joinr � prepare_keyZsignr. r
r �NotImplementedError) r �payloadr" � algorithm�headersZ json_encoderZsegments�headerZ json_header�
signing_inputr, � signaturer# r# r$ �encodeL s@
��
��z PyJWS.encode� Tc
K sl t | j|�}|d }|r(|s(t�dt� | �|�\} }
} } |sPtjdtdd� n|rh| �| |
| | ||� | S )Nr& z�It is strongly recommended that you pass in a value for the "algorithms" argument when calling decode(). This argument will be mandatory in a future version.zSThe verify parameter is deprecated. Please use verify_signature in options instead.� )�
stacklevel)r r �warnings�warn�DeprecationWarning�_load�_verify_signature)
r �jwtr" �verifyr! r �kwargsZmerged_optionsr& rC rG rF rH r# r# r$ �decode� s& � � �z PyJWS.decodec C s | � |�d }| �|� |S )z�Returns back the JWT header parameters as a dict()
Note: The signature is not verified so the header parameters
should not be fully trusted until signature verification is complete
rK )rP r; )r rR rE r# r# r$ �get_unverified_header� s
zPyJWS.get_unverified_headerc
C sh t |t�r|�d�}tt|�t�s0td�t���z$|�dd�\}}|� dd�\}}W n t
k
rp td��Y nX z t |�}W n" t t
jfk
r� td��Y nX zt�|�d��}W n. t
k
r� } ztd| ��W 5 d }~X Y nX t |t�s�td��z t |�} W n$ t t
jfk
�r( td ��Y nX z t |�}
W n$ t t
jfk
�rZ td
��Y nX | |||
fS ) Nzutf-8z'Invalid token type. Token must be a {0}r8 r zNot enough segmentszInvalid header paddingzInvalid header string: %sz,Invalid header string: must be a json objectzInvalid payload paddingzInvalid crypto padding)r( r rI �
issubclass�typer
r �format�rsplit�splitr' r r) �binascii�Errorr= �loadsrU r ) r rR rG Zcrypto_segmentZheader_segmentZpayload_segmentZ header_datarF �erC rH r# r# r$ rP � s<
�
z PyJWS._loadc C sr |� d�}|d k r"||kr"td��z.| j| }|�|�}|�|||�sNtd��W n tk
rl td��Y nX d S )Nr4 z&The specified alg value is not allowedz Signature verification failedr9 )�getr r rA rS r r. ) r rC rG rF rH r" r! r4 r, r# r# r$ rQ � s
zPyJWS._verify_signaturec C s d|kr| � |d � d S )N�kid)�
_validate_kid)r rE r# r# r$ r; � s zPyJWS._validate_headersc C s t |t�std��d S )Nz(Key ID header parameter must be a string)r( r r )r ra r# r# r$ rb � s
zPyJWS._validate_kid)NN)r2 NN)rJ TNN)rJ N)�__name__�
__module__� __qualname__r: r% � staticmethodr r- r0 r1 rI rU rV rP rQ r; rb r# r# r# r$ r s.
�
7 �
( �
r )&r\ r= rM �typingr r r r r � ImportErrorr! r r r
r �compatr r
r r �
exceptionsr r r r Zutilsr r r r �objectr Z_jws_global_objrI rU r- r0 rV r# r# r# r$ �<module> s$ Y