o
    "4gH3                     @   s  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 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 d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z m!Z!m"Z" de#fddZ$efdee% fddZ&dedfde%fddZ'da(de%fddZ)de%fddZ*ddddZ+ddd d!Z,ddd"d#Z-d?de#fd$d%Z.					d@d&d'Z/d?d(d)Z0d*d+ Z1d,d- Z2dAde#fd.d/Z3d0d1 Z4d2d3 Z5	dBd4e%d5ee# d6e#fd7d8Z6d9d: Z7de%fd;d<Z8d=d> Z9dS )C    N)OrderedDict)Optional)settings)REDIRECT_FIELD_NAMEget_user_model)FieldDoesNotExist)models)Q)	force_str)base36_to_intint_to_base36)app_settings)get_adapter)flows)Login)httpkit)get_request_paramimport_callablevalid_email_or_nonereturnc                 C   s(   t d|  }t d| }||kS )z
    Perform case-insensitive comparison of two identifiers, using the
    recommended algorithm from Unicode Technical Report 36, section
    2.11.2(B)(2).
    NFKC)unicodedata	normalizecasefold)s1s2norm_s1norm_s2 r   ^/var/www/html/authentication-server/venv/lib/python3.10/site-packages/allauth/account/utils.py_unicode_ci_compare   s   r    c                 C   s"   t | |}|rt |sd}|S )z_
    Returns the next URL to redirect to, if it was explicitly passed
    via the request.
    N)r   r   is_safe_url)requestredirect_field_nameredirect_tor   r   r   get_next_redirect_url$   s   
r%   Fc                 C   sN   |}|rt |r| }|st| |d}|s%|rt | }|S t | }|S )N)r#   )callabler%   r   get_signup_redirect_urlget_login_redirect_url)r"   urlr#   signupretr   r   r   r(   1   s   r(   c                 C   s*   d}t jrt| t j}|pt| p| jjS )N )r   USER_MODEL_USERNAME_FIELDgetattrr
   _metaverbose_name)userr+   r   r   r   default_user_displayF   s   r2   c                 C   s    t sttdt}t|a t | S )NACCOUNT_USER_DISPLAY)_user_display_callabler.   r   r2   r   )r1   fr   r   r   user_displayM   s   r6   )commitc                G   s   |sdS t  }z|j|}|j}W n ty%   t| |s!Y dS d}Y nw |rG|d }|r4|d| }t| || |rE| j|gd dS dS t| |S )zR
    Gets or sets (optional) user model fields. No-op if fields do not exist.
    Nr   )update_fields)	r   r/   	get_field
max_lengthr   hasattrsetattrsaver.   )r1   fieldr7   argsUser
field_metar:   vr   r   r   
user_fieldU   s(   


rC   c                G   s4   |rt js|d r|d  g}t| t jg|R  S )Nr   )r   PRESERVE_USERNAME_CASINGlowerrC   r-   )r1   r7   r?   r   r   r   user_usernamep   s   rF   c                G   sD   |r|d r|d   g}t| tjg|R d|i}|r |  }|S )Nr   r7   )rE   rC   r   USER_MODEL_EMAIL_FIELD)r1   r7   r?   r+   r   r   r   
user_emailv   s   rH   c                 C   sb   ddl m} d }|r%d}z|j| |}|j}W |S  |jy$   Y |S w |jj| dd }|S )N   EmailAddressFT)r1   verified)r   rK   objectsget_for_userrL   DoesNotExistfilterexists)r1   emailrK   emailaddressr+   r   r   r   has_verified_email   s   rT   c                 C   s"   t ||||||d}tj| |S )N)r1   email_verificationredirect_urlsignal_kwargsr*   rR   )r   r   loginperform_login)r"   r1   rU   rV   rW   r*   rR   rX   r   r   r   rY      s   	rY   c                 C   s   t jj| ||||dS )N)r1   rU   rV   rW   )r   r*   complete_signup)r"   r1   rU   success_urlrW   r   r   r   rZ      s   rZ   c                 C   s~  ddl m} t }t }g }g }g }|D ]n}t|j}	|	sq|	|_tjr1tjdkr1|j	
|	gr1qtjrCtjdkrC|jrC|j	|	rCq||	}
|
rY|
jpO|j|
_|
jpV|j|
_n|}
|
jpd|| |
j|
_|
||	< |
jrz||
 |
jrz||
 |
jr||
 q|r|d }n|r|d }n|r|d }n|rt| d }nd}| D ]}
|j |
j k|
_qt| |fS )a
  
    Takes a list of EmailAddress instances and cleans it up, making
    sure only valid ones remain, without multiple primaries etc.

    Order is important: e.g. if multiple primary email addresses
    exist, the first one encountered will be kept as primary.
    rI   rJ   strictr   N)r   rK   r   r   r   rR   r   UNIQUE_EMAILPREVENT_ENUMERATIONrM   lookuprL   is_verifiedgetprimaryis_email_verifiedappendlistvaluesrE   )r"   	addressesrK   adaptere2aprimary_addressesverified_addressesprimary_verified_addressesaddressrR   aprimary_addressr   r   r   cleanup_email_addresses   sf   










rp   c           
      C   s   ddl m} |jj|d rJ g }t }|| }|r*|||| ddd t	|}|r=|||| ddd t
| || \}}|D ]	}	||	_|	  qH|j|| |rp|p^d |j krpt	||j |  |S )z
    Creates proper EmailAddress for the user that was just signed
    up. Only sets up, doesn't do any other handling such as sending
    out email confirmation mails etc.
    rI   rJ   )r1   T)r1   rR   rb   rL   Fr,   )r   rK   rM   rP   rQ   r   unstash_verified_emailrd   rE   rH   rp   r1   r=   fill_cache_for_userrR   )
r"   r1   rg   rK   priority_addressesrh   stashed_emailrR   rb   rn   r   r   r   setup_user_email   s6   

ru   c                 C   s   t jj| |||dS )N)r*   rR   )r   rU   send_verification_email)r"   r1   r*   rR   r   r   r   send_email_confirmation  s   rw   c                 C   sP   ddl m} t| }|r$|jj| |d s&|jj| |dddd dS dS dS )z
    Keep user.email in sync with user.emailaddress_set.

    Under some circumstances the user.email may not have ended up as
    an EmailAddress record, e.g. in the case of manually created admin
    users.
    rI   rJ   )r1   rR   F)rb   rL   )r1   rR   defaultsN)r   rK   rH   rM   rP   rQ   get_or_create)r1   rK   rR   r   r   r   sync_user_email_addresses!  s   
rz   c                  G   st   t jr$dd | D }|d }|dd  D ]}||B }qt j|}|S t jjdi t jd dd | D i}|S )Nc                 S   s$   g | ]}t di tjd  |iqS )__iexactr   )r	   r   r-   .0ur   r   r   
<listcomp>5  s    z,filter_users_by_username.<locals>.<listcomp>r   rI   __inc                 S   s   g | ]}|  qS r   )rE   r|   r   r   r   r   A  s    r   )r   rD   r   _default_managerrP   r-   )usernameqlistqq2r+   r   r   r   filter_users_by_username3  s"   
r   rR   	is_activeprefer_verifiedc                    s   ddl m} t }|  } t|jj| dd}d}|r,ttdd |}|r,|}d}g }|D ]}	t|	j	| r>|
|	j q0tjri|sitj| i}
|jjdi |
}| D ]}t|tj}t|| rh|
| qV d	urx fd
dt|D }tt|S )aE  Return list of users by email address

    Typically one, at most just a few in length.  First we look through
    EmailAddress table, than customisable User model table. Add results
    together avoiding SQL joins and deduplicate.

    `prefer_verified`: When looking up users by email, there can be cases where
    users with verified email addresses are preferable above users who did not
    verify their email address. The password reset is such a use case -- if
    there is a user with a verified email than that user should be returned, not
    one of the other users.
    rI   rJ   )rR   r1   Fc                 S   s   | j S N)rL   )er   r   r   <lambda>]  s    z'filter_users_by_email.<locals>.<lambda>TNc                    s   g | ]	}|j  kr|qS r   r   r|   r   r   r   r   m  s    z)filter_users_by_email.<locals>.<listcomp>r   )r   rK   r   rE   re   rM   rP   select_relatedr    rR   rd   r1   r   rG   iteratorr.   set)rR   r   r   rK   r@   mailsr`   verified_mailsusersr   q_dictuser_qsr1   rH   r   r   r   filter_users_by_emailG  s4   



r   c                 C   s"   t | |}|rt|||i}|S r   )r%   r   add_query_params)r"   r)   r#   next_urlr   r   r   passthrough_next_redirect_urlq  s   
r   c                 C   s\   t  }t|jj}t|tjrt| jtr| jS | jj	S t|tj
r)tt| jS t| jS )z&
    This should return a string.
    )r   typer/   pk
issubclassr   	UUIDField
isinstancestrhexIntegerFieldr   int)r1   r@   pk_field_classr   r   r   user_pk_to_url_strx  s   
r   c                 C   sx   t  }t|jjdd }|rt|dd r|jjjjjj}n|jj}t|}t|tj	r5t
| }||}|S || }|S )Nremote_fieldto)r   r.   r/   r   r   r   r   r   r   r   r   	to_python)pk_strr@   r   pk_fieldr   r   r   r   r   url_str_to_user_pk  s   

r   r   )NNNFN)FN)NF):r   collectionsr   typingr   django.confr   django.contrib.authr   r   django.core.exceptionsr   	django.dbr   django.db.modelsr	   django.utils.encodingr
   django.utils.httpr   r   allauth.accountr   allauth.account.adapterr   allauth.account.internalr   allauth.account.modelsr   allauth.core.internalr   allauth.utilsr   r   r   boolr    r   r%   r(   r4   r2   r6   rC   rF   rH   rT   rY   rZ   rp   ru   rw   rz   r   r   r   r   r   r   r   r   r   <module>   sn    

	


I%
*