Edit File: 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_key�signr. r r �NotImplementedError) r �payloadr" � algorithm�headersZ json_encoder�segments�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& rC rH rG rI 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 rL )rQ r: )r rS rE 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 rJ � issubclass�typer r �format�rsplit�splitr' r r) �binascii�Errorr< �loadsrV r ) r rS rH Zcrypto_segmentZheader_segmentZpayload_segmentZ header_datarG �erC rI r# r# r$ rQ � 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@ rT r r. ) r rC rH rG rI r" r! r4 r, r# r# r$ rR � 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 rb r# r# r$ rc � s zPyJWS._validate_kid)NN)r2 NN)rK TNN)rK N)�__name__� __module__� __qualname__r9 r% � staticmethodr r- r0 r1 rJ rV rW rQ rR r: rc r# r# r# r$ r s$ 2 ( r )&r] r< rN �typingr r r r r � ImportErrorr! r r r r �compatr r r r � exceptionsr r r r �utilsr r r r �objectr Z_jws_global_objrJ rV r- r0 rW r# r# r# r$ �<module> s$ Y