o
    ˷e                     @   sD   d dl Z d dlmZ d dlmZ G dd deZG dd deZdS )    N)	parse_qsl)	urlencodec                   @   s   e Zd ZdS )OAuth2ErrorN)__name__
__module____qualname__ r   r   d/var/www/ideatree/venv/lib/python3.10/site-packages/allauth/socialaccount/providers/oauth2/client.pyr      s    r   c                   @   s4   e Zd Z			dddZdd Zdd	 Zd
d ZdS )OAuth2Client NFc                 C   sJ   || _ || _|| _|| _|| _|| _|t|| _d | _	|	| _
|
| _d S )N)requestaccess_token_methodaccess_token_urlcallback_urlconsumer_keyconsumer_secretjoinsetscopestateheaders
basic_auth)selfr   r   r   r   r   r   r   scope_delimiterr   r   r   r   r	   __init__   s   
zOAuth2Client.__init__c                 C   s>   | j | j| jdd}| jr| j|d< || d|t|f S )Ncode)	client_idredirect_urir   response_typer   z%s?%s)r   r   r   r   updater   )r   authorization_urlextra_paramsparamsr   r   r	   get_redirect_url$   s   

zOAuth2Client.get_redirect_urlc                 C   s   | j d|d}| jrtj| j| j}nd }|| j| jd d }| | | j	}| j
dkr3|}d }tj| j
|||| j|d}d }|jdv rh|jd dd	 d
ks\|jd d dkra| }ntt|j}|rnd|vrutd|j |S )Nauthorization_code)r   
grant_typer   )r   client_secretGET)r"   datar   auth)      zcontent-type;r   zapplication/json   z{"access_tokenz!Error retrieving access token: %s)r   r   requestsr)   HTTPBasicAuthr   r   r   _strip_empty_keysr   r   r   r   status_codesplittextjsondictr   r   content)r   r   r(   r)   r"   urlrespr.   r   r   r	   get_access_token0   sF   

	

zOAuth2Client.get_access_tokenc                 C   s&   dd |  D }|D ]}||= qdS )zlAdded because the Dropbox OAuth2 flow doesn't
        work when scope is passed in, which is empty.
        c                 S   s   g | ]
\}}|d kr|qS ) r   ).0kvr   r   r	   
<listcomp>b   s    z2OAuth2Client._strip_empty_keys.<locals>.<listcomp>N)items)r   r"   keyskeyr   r   r	   r1   ^   s   zOAuth2Client._strip_empty_keys)r   NF)r   r   r   r   r#   r:   r1   r   r   r   r	   r
      s    

.r
   )	r/   urllib.parser   django.utils.httpr   	Exceptionr   objectr
   r   r   r   r	   <module>   s
    