o
    ‘¸õaY+  ã                   @   sÜ   d Z ddlZddlZddlZddlmZ ddlmZ ddlm	Z	 ddl
mZmZ ddlmZ G d	d
„ d
eƒZ							d dd„Zdd„ Zd!dd„Zd"dd„Zd#dd„Zdd„ Zdd„ ZG dd„ dƒZG dd„ deƒZdS )$zÿ
oauthlib.oauth2.rfc6749.tokens
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

This module contains methods for adding two types of access tokens to requests.

- Bearer https://tools.ietf.org/html/rfc6750
- MAC https://tools.ietf.org/html/draft-ietf-oauth-v2-http-mac-01
é    N)Ú
b2a_base64)Úurlparse)Úcommon)Úadd_params_to_qsÚadd_params_to_urié   )Úutilsc                       sr   e Zd Zd‡ fdd„	Zedd„ ƒZedd„ ƒZedd	„ ƒZed
d„ ƒZedd„ ƒZ	edd„ ƒZ
edd„ ƒZ‡  ZS )ÚOAuth2TokenNc                    st   t ƒ  |¡ d | _d|v r|d rtt |d ¡ƒ| _|d ur4tt |¡ƒ| _| jd u r2| j| _d S d S | j| _d S )NÚscope)ÚsuperÚ__init__Ú
_new_scopeÚsetr   Úscope_to_listÚ
_old_scope)ÚselfÚparamsÚ	old_scope©Ú	__class__© ú@/usr/lib/python3/dist-packages/oauthlib/oauth2/rfc6749/tokens.pyr      s   
ýzOAuth2Token.__init__c                 C   s   | j | jkS ©N)r   r   ©r   r   r   r   Úscope_changed&   ó   zOAuth2Token.scope_changedc                 C   ó   t  | j¡S r   )r   Úlist_to_scoper   r   r   r   r   r   *   r   zOAuth2Token.old_scopec                 C   ó
   t | jƒS r   )Úlistr   r   r   r   r   Ú
old_scopes.   ó   
zOAuth2Token.old_scopesc                 C   r   r   )r   r   r   r   r   r   r   r
   2   r   zOAuth2Token.scopec                 C   r   r   )r   r   r   r   r   r   Úscopes6   r!   zOAuth2Token.scopesc                 C   ó   t | j| j ƒS r   )r   r   r   r   r   r   r   Úmissing_scopes:   ó   zOAuth2Token.missing_scopesc                 C   r#   r   )r   r   r   r   r   r   r   Úadditional_scopes>   r%   zOAuth2Token.additional_scopesr   )Ú__name__Ú
__module__Ú__qualname__r   Úpropertyr   r   r    r
   r"   r$   r&   Ú__classcell__r   r   r   r   r	      s     





r	   Ú ú
hmac-sha-1c                 C   s0  |  ¡ }t |¡\}}| ¡ dkrtj}n| ¡ dkrtj}ntdƒ‚|
dkr5|p3d t 	|	¡t
 ¡ ¡}nt
 ¡ }t
 ¡ }t|ƒ\}}}}}}|rP|d | }n|}|duro|
dkro| d¡}t||ƒ ¡ ƒdd	…  d¡}nd
}g }|
dkr}| |¡ n
| |¡ | |¡ | |  ¡ ¡ | |¡ | |¡ | |¡ |
dkr¦| |¡ | |p«d
¡ d |¡d }t|tƒr¾| d¡}t || d¡|¡}t| ¡ ƒdd	…  d¡}g }| d|  ¡ |
dkré| d| ¡ | d| ¡ |rù| d| ¡ |r| d| ¡ | d| ¡ |pi }d |¡|d< |S )a_  Add an `MAC Access Authentication`_ signature to headers.

    Unlike OAuth 1, this HMAC signature does not require inclusion of the
    request payload/body, neither does it use a combination of client_secret
    and token_secret but rather a mac_key provided together with the access
    token.

    Currently two algorithms are supported, "hmac-sha-1" and "hmac-sha-256",
    `extension algorithms`_ are not supported.

    Example MAC Authorization header, linebreaks added for clarity

    Authorization: MAC id="h480djs93hd8",
                       nonce="1336363200:dj83hs9s",
                       mac="bhCQXTVyfj5cmA9uKkPFx1zeOXM="

    .. _`MAC Access Authentication`: https://tools.ietf.org/html/draft-ietf-oauth-v2-http-mac-01
    .. _`extension algorithms`: https://tools.ietf.org/html/draft-ietf-oauth-v2-http-mac-01#section-7.1

    :param token:
    :param uri: Request URI.
    :param key: MAC given provided by token endpoint.
    :param http_method: HTTP Request method.
    :param nonce:
    :param headers: Request headers as a dictionary.
    :param body:
    :param ext:
    :param hash_algorithm: HMAC algorithm provided by token endpoint.
    :param issue_time: Time when the MAC credentials were issued (datetime).
    :param draft: MAC authentication specification version.
    :return: headers dictionary with the authorization field added.
    r-   zhmac-sha-256zunknown hash algorithmr   z{}:{}ú?Nzutf-8éÿÿÿÿr,   Ú
zMAC id="%s"zts="%s"z
nonce="%s"zbodyhash="%s"zext="%s"zmac="%s"z, ÚAuthorization)Úupperr   Úhost_from_uriÚlowerÚhashlibÚsha1Úsha256Ú
ValueErrorÚformatÚgenerate_ager   Úgenerate_nonceÚgenerate_timestampr   Úencoder   ÚdigestÚdecodeÚappendÚjoinÚ
isinstanceÚstrÚhmacÚnew)ÚtokenÚuriÚkeyÚhttp_methodÚnonceÚheadersÚbodyÚextÚhash_algorithmÚ
issue_timeÚdraftÚhostÚportÚhÚtsÚschÚnetÚpathÚparÚqueryÚfraÚrequest_uriÚbodyhashÚbaseÚbase_stringÚsignÚheaderr   r   r   Úprepare_mac_headerC   sf   (ÿ
 








ra   c                 C   ó   t |d| fgƒS )a  Add a `Bearer Token`_ to the request URI.
    Not recommended, use only if client can't use authorization header or body.

    http://www.example.com/path?access_token=h480djs93hd8

    .. _`Bearer Token`: https://tools.ietf.org/html/rfc6750

    :param token:
    :param uri:
    Úaccess_token)r   )rF   rG   r   r   r   Úprepare_bearer_uri°   s   rd   c                 C   s   |pi }d|  |d< |S )zëAdd a `Bearer Token`_ to the request URI.
    Recommended method of passing bearer tokens.

    Authorization: Bearer h480djs93hd8

    .. _`Bearer Token`: https://tools.ietf.org/html/rfc6750

    :param token:
    :param headers:
    z	Bearer %sr1   r   )rF   rK   r   r   r   Úprepare_bearer_headers¾   s   re   c                 C   rb   )z¯Add a `Bearer Token`_ to the request body.

    access_token=h480djs93hd8

    .. _`Bearer Token`: https://tools.ietf.org/html/rfc6750

    :param token:
    :param body:
    rc   )r   )rF   rL   r   r   r   Úprepare_bearer_bodyÎ   s   
rf   Fc                 C   s   t  ¡ S )zp
    :param request: OAuthlib request.
    :type request: oauthlib.common.Request
    :param refresh_token:
    )r   Úgenerate_token)ÚrequestÚrefresh_tokenr   r   r   Úrandom_token_generatorÛ   s   rj   c                    s   ‡ ‡fdd„}|S )z
    :param private_pem:
    c                    s   ˆ | _ t ˆ| ¡S r   )Úclaimsr   Úgenerate_signed_token)rh   ©ÚkwargsÚprivate_pemr   r   Úsigned_token_generatorè   s   z6signed_token_generator.<locals>.signed_token_generatorr   )ro   rn   rp   r   rm   r   rp   ä   s   rp   c                 C   sP   d}d| j v r#| j  d¡ ¡ }t|ƒdkr!|d  ¡ dkr!|d }|S | j}|S )zç
    Helper function to extract a token from the request header.

    :param request: OAuthlib request.
    :type request: oauthlib.common.Request
    :return: Return the token or None if the Authorization header is malformed.
    Nr1   é   r   Úbearerr   )rK   ÚgetÚsplitÚlenr4   rc   )rh   rF   Úsplit_headerr   r   r   Úget_token_from_headerï   s   
þrw   c                   @   s&   e Zd Zd	dd„Zdd„ Zdd„ ZdS )
Ú	TokenBaseFc                 C   ó   t dƒ‚)Nú&Subclasses must implement this method.©ÚNotImplementedError)r   rh   ri   r   r   r   Ú__call__  s   zTokenBase.__call__c                 C   ry   ©úb
        :param request: OAuthlib request.
        :type request: oauthlib.common.Request
        rz   r{   ©r   rh   r   r   r   Úvalidate_request  ó   zTokenBase.validate_requestc                 C   ry   r~   r{   r€   r   r   r   Úestimate_type  r‚   zTokenBase.estimate_typeN©F)r'   r(   r)   r}   r   rƒ   r   r   r   r   rx     s    
rx   c                   @   s8   e Zd ZdZ		ddd„Zddd„Zdd	„ Zd
d„ ZdS )ÚBearerToken)Úrequest_validatorÚtoken_generatorÚrefresh_token_generatorÚ
expires_inNc                 C   s*   || _ |pt| _|p| j| _|pd| _d S )Ni  )r†   rj   r‡   rˆ   r‰   )r   r†   r‡   r‰   rˆ   r   r   r   r     s
   
ÿzBearerToken.__init__Fc                 K   sª   d|v r
t  dt¡ t| jƒr|  |¡}n| j}||_|  |¡|ddœ}|jdur1d |j¡|d< |rI|jrB| j	 
|¡sB|j|d< n|  |¡|d< | |jpOi ¡ t|ƒS )	zÁ
        Create a BearerToken, by default without refresh token.

        :param request: OAuthlib request.
        :type request: oauthlib.common.Request
        :param refresh_token:
        Ú
save_tokenzx`save_token` has been deprecated, it was not called internally.If you do, call `request_validator.save_token()` instead.ÚBearer)rc   r‰   Ú
token_typeNú r
   ri   )ÚwarningsÚwarnÚDeprecationWarningÚcallabler‰   r‡   r"   rA   ri   r†   Úrotate_refresh_tokenrˆ   ÚupdateÚextra_credentialsr	   )r   rh   ri   rn   r‰   rF   r   r   r   Úcreate_token&  s,   þ
ý
	
ÿzBearerToken.create_tokenc                 C   s   t |ƒ}| j ||j|¡S )r   )rw   r†   Úvalidate_bearer_tokenr"   )r   rh   rF   r   r   r   r   P  s   ÿzBearerToken.validate_requestc                 C   s6   |j  dd¡ d¡d  ¡ dkrdS |jdurdS dS )	r   r1   r,   r   r   rr   é	   Né   )rK   rs   rt   r4   rc   r€   r   r   r   rƒ   Y  s
    
zBearerToken.estimate_type)NNNNr„   )r'   r(   r)   Ú	__slots__r   r•   r   rƒ   r   r   r   r   r…     s    
ÿ
	*	r…   )NNNr,   r-   Nr   r   )r,   r„   )Ú__doc__r5   rD   rŽ   Úbinasciir   Úurllib.parser   Úoauthlibr   Úoauthlib.commonr   r   r,   r   Údictr	   ra   rd   re   rf   rj   rp   rw   rx   r…   r   r   r   r   Ú<module>   s4    	.
ùm


	