o
    &zh*                  	   @   s  d dl Z d dlmZ d dlmZmZ d dlmZ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 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# 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(d3de$fddZ)					d4ddZ*d3d d!Z+d"d# Z,d$d% Z-d&d' Z.	d5d(e&d)ee$ d*e$defd+d,Z/d-d. Z0de&fd/d0Z1d1d2 Z2dS )6    N)OrderedDict)ListOptional)REDIRECT_FIELD_NAMEget_user_model)models)Q)base36_to_intint_to_base36)app_settings)get_adapter)flows)valid_email_or_none)default_user_display)user_display)
user_email)
user_field)user_username)Login)httpkit)get_request_param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    P/var/www/html/kangema/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*   0   s   r*   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)userverified)r   r0   objectsget_for_userr2   DoesNotExistfilterexists)r1   emailr0   emailaddressr-   r    r    r!   has_verified_emailB   s   r:   c                 C   s"   t ||||||d}tj| |S )N)r1   email_verificationredirect_urlsignal_kwargsr,   r8   )r   r   loginperform_login)r$   r1   r;   r<   r=   r,   r8   r>   r    r    r!   r?   R   s   	r?   c                 C   s   t jj| ||||dS )N)r1   r;   r<   r=   )r   r,   complete_signup)r$   r1   r;   success_urlr=   r    r    r!   r@   f   s   r@   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.
    r.   r/   strictr   N)r   r0   r   r   r   r8   r   UNIQUE_EMAILPREVENT_ENUMERATIONr3   lookupr2   is_verifiedgetprimaryis_email_verifiedappendlistvalueslower)r$   	addressesr0   adaptere2aprimary_addressesverified_addressesprimary_verified_addressesaddressr8   aprimary_addressr    r    r!   cleanup_email_addressesp   sf   










rW   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.
    r.   r/   )r1   T)r1   r8   rH   r2   F )r   r0   r3   r6   r7   r   unstash_verified_emailrJ   rM   r   rW   r1   savefill_cache_for_userr8   )
r$   r1   rN   r0   priority_addressesrO   stashed_emailr8   rH   rU   r    r    r!   setup_user_email   s6   

r^   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   USER_MODEL_USERNAME_FIELD.0ur    r    r!   
<listcomp>   s    z,filter_users_by_username.<locals>.<listcomp>r   r.   __inc                 S   s   g | ]}|  qS r    )rM   ra   r    r    r!   rd      s    r    )r   PRESERVE_USERNAME_CASINGr   _default_managerr6   r`   )usernameqlistqq2r-   r    r    r!   filter_users_by_username   s"   
rl   r8   	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rj|sjtj| i}
|jjdi |
}|d	D ]}t|tj}t|| ri|
| qW d
ury 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.
    r.   r/   )r8   r1   Fc                 S   s   | j S N)r2   )er    r    r!   <lambda>  s    z'filter_users_by_email.<locals>.<lambda>Ti  Nc                    s   g | ]	}|j  kr|qS r    rm   ra   rr   r    r!   rd     s    z)filter_users_by_email.<locals>.<listcomp>r    )r   r0   r   rM   rK   r3   r6   select_relatedr"   r8   rJ   r1   r   USER_MODEL_EMAIL_FIELDiteratorgetattrset)r8   rm   rn   r0   UsermailsrF   verified_mailsusersrp   q_dictuser_qsr1   r   r    rr   r!   filter_users_by_email   s4   



r~   c                 C   s"   t | |}|rt|||i}|S ro   )r'   r   add_query_params)r$   r+   r%   next_urlr    r    r!   passthrough_next_redirect_url  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   type_metapk
issubclassr   	UUIDField
isinstancestrhexIntegerFieldr
   int)r1   rx   pk_field_classr    r    r!   user_pk_to_url_str#  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   rv   r   r   r   r   r   r   r   r   r	   	to_python)pk_strrx   r   pk_fieldr   r   r    r    r!   url_str_to_user_pk2  s   

r   ro   )NNNFN)NF)3r   collectionsr   typingr   r   django.contrib.authr   r   	django.dbr   django.db.modelsr   django.utils.httpr	   r
   allauth.accountr   allauth.account.adapterr   allauth.account.internalr   !allauth.account.internal.emailkitr    allauth.account.internal.userkitr   r   r   r   r   allauth.account.modelsr   allauth.core.internalr   allauth.utilsr   boolr"   r   r'   r*   r:   r?   r@   rW   r^   rl   r~   r   r   r   r    r    r    r!   <module>   sh    



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