o
    ˷e                     @   sn   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
G dd deZG d	d
 d
eZegZdS )    )EmailAddress)QUERY_EMAIL)
AuthActionProviderAccount)OAuth2Providerc                   @   s   e Zd ZdZdZdZdZdS )Scopeopenidemailprofilezorg.cilogon.userinfoN)__name__
__module____qualname__OPENIDEMAILPROFILEUSERINFO r   r   g/var/www/ideatree/venv/lib/python3.10/site-packages/allauth/socialaccount/providers/cilogon/provider.pyr      s
    r   c                       s   e Zd Z fddZ  ZS )CILogonAccountc                    s   t t|  }| jjd|S )Nname)superr   to_straccount
extra_dataget)selfdflt	__class__r   r   r      s   zCILogonAccount.to_str)r   r   r   r   __classcell__r   r   r   r   r      s    r   c                       sH   e Zd ZdZdZeZdd Z fddZdd Z	d	d
 Z
dd Z  ZS )CILogonProvidercilogonCILogonc                 C   s$   t jt jt jg}tr|t j |S )N)r   r   r   r   r   appendr   )r   scoper   r   r   get_default_scope   s   z!CILogonProvider.get_default_scopec                    s(   t t| ||}|tjkrd|d< |S )Nzselect_account consentprompt)r   r    get_auth_paramsr   REAUTHENTICATE)r   requestactionretr   r   r   r'      s   
zCILogonProvider.get_auth_paramsc                 C   s   t |dS )Nsub)strr   r   datar   r   r   extract_uid%   s   zCILogonProvider.extract_uidc                 C   s(   t |d|d|d|ddS )Nr	   family_name
given_nameeppn)r	   	last_name
first_namer3   )dictr   r.   r   r   r   extract_common_fields(   s   z%CILogonProvider.extract_common_fieldsc                 C   s4   g }| d}|r| dr|t|ddd |S )Nr	   verified_emailT)r	   verifiedprimary)r   r#   r   )r   r/   r+   r	   r   r   r   extract_email_addresses0   s
   
z'CILogonProvider.extract_email_addresses)r   r   r   idr   r   account_classr%   r'   r0   r7   r;   r   r   r   r   r   r       s    r    N)allauth.account.modelsr   "allauth.socialaccount.app_settingsr   $allauth.socialaccount.providers.baser   r   /allauth.socialaccount.providers.oauth2.providerr   objectr   r   r    provider_classesr   r   r   r   <module>   s    
$