File: //opt/alt/python37/lib/python3.7/site-packages/jwt/__pycache__/api_jws.cpython-37.pyc
B
�b�[� � @ s� d dl Z d dlZd dlZy 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 sj t � | _|d k rt|�nt| j�| _x(t| j�� �D ]}|| jkr4| j|= q4W |sVi }t| �� |�| _d S )N) r � _algorithms�set� _valid_algs�list�keysr �_get_default_options�options)�self�
algorithmsr �key� r# �</opt/alt/python37/lib/python3.7/site-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� |�} y$| 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?zAlgorithm 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 s8
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& rB rF rE rG 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
rJ )rO r: )r rQ rD r# r# r$ �get_unverified_header� s
zPyJWS.get_unverified_headerc
C sh t |t�r|�d�}tt|�t�s0td�t���y$|�dd�\}}|� dd�\}}W n t
k
rp td��Y nX y t |�}W n" t t
jfk
r� td��Y nX yt�|�d��}W n. t
k
r� } ztd| ��W d d }~X Y nX t |t�s�td��y t |�} W n$ t t
jfk
�r( td ��Y nX y 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 rH �
issubclass�typer
r �format�rsplit�splitr' r r) �binascii�Errorr< �loadsrT r ) r rQ rF Zcrypto_segmentZheader_segmentZpayload_segmentZ header_datarE �erB rG r# r# r$ rO � s:
z PyJWS._loadc C sr |� d�}|d k r"||kr"td��y.| j| }|�|�}|�|||�sNtd��W n tk
rl td��Y nX d S )Nr4 z&The specified alg value is not allowedz Signature verification failedzAlgorithm not supported)�getr r r@ rR r r. ) r rB rF rE rG r" r! r4 r, r# r# r$ rP � s
zPyJWS._verify_signaturec C s d|kr| � |d � d S )N�kid)�
_validate_kid)r rD 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 r` r# r# r$ ra � s
zPyJWS._validate_kid)NN)r2 NN)rI TNN)rI N)�__name__�
__module__� __qualname__r9 r% � staticmethodr r- r0 r1 rH rT rU rO rP r: ra r# r# r# r$ r s$
2
(
r )&r[ r< rL �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_objrH rT r- r0 rU r# r# r# r$ �<module> s$ Y