
    fH                     
   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	 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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Z(efdee)   fdZ*dedfde)fdZ+da,de)fdZ-de)fdZ.dddZ/dddZ0dddZ1d1de'fdZ2	 	 	 	 	 d2dZ3d Z4d3d Z5d! Z6d1d"Z7d# Z8d$ Z9d4d%Z:d& Z;d' Z<	 d5d(e)d)ee'   d*e'fd+Z=d, Z>de)fd-Z?d. Z@dee'   fd/ZAd6d0ZBy)7    N)OrderedDict)Optional)settings)messages)REDIRECT_FIELD_NAMEget_user_model)FieldDoesNotExist)models)Q)	force_str)base36_to_intint_to_base36)app_settingssignals)get_adapter)flows)Login)httpkit)get_request_paramimport_callablevalid_email_or_nonereturnc                     t        j                  d|       j                         }t        j                  d|      j                         }||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_s2s       P/var/www/cs2snipe.com/venv/lib/python3.12/site-packages/allauth/account/utils.py_unicode_ci_comparer#      sE     ##FB/88:G##FB/88:Gg    c                 X    t        | |      }|rt               j                  |      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_tos      r"   get_next_redirect_urlr*   &   s.     $G-@AK;=44[Ar$   Fc                     |}|rt        |      r |       }|st        | |      }|s6|rt               j                  |       }|S t               j	                  |       }|S )N)r(   )callabler*   r   get_signup_redirect_urlget_login_redirect_url)r'   urlr(   signuprets        r"   r.   r.   3   sc     C
x} e#GATU-77@C J -66w?CJr$   c                     d}t         j                  rt        | t         j                        }|xs# t        |       xs | j                  j
                  S )N )r   USER_MODEL_USERNAME_FIELDgetattrr   _metaverbose_name)userr1   s     r"   default_user_displayr9   H   s@    
C--dLBBC<)D/<TZZ%<%<<r$   c                 d    t         s t        t        dt              }t	        |      a t        |       S )NACCOUNT_USER_DISPLAY)_user_display_callabler5   r   r9   r   )r8   fs     r"   user_displayr>   O   s+    !H46JK!0!3!$''r$   )commitc                &   |syt               }	 |j                  j                  |      }|j                  }|r0|d   }|r|d| }t        | ||       |r| j                  |g       yyt        | |      S # t        $ r t        | |      sY yd}Y Yw xY w)zR
    Gets or sets (optional) user model fields. No-op if fields do not exist.
    Nr   )update_fields)	r   r6   	get_field
max_lengthr	   hasattrsetattrsaver5   )r8   fieldr?   argsUser
field_metarC   vs           r"   
user_fieldrL   W   s     DZZ))%0
**

 G!JAeQIIUGI,  tU##  tU#
s   'A4 4BBBc                    |r)t         j                  s|d   r|d   j                         g}t        | t         j                  g| S )Nr   )r   PRESERVE_USERNAME_CASINGlowerrL   r4   )r8   r?   rH   s      r"   user_usernamerP   r   s=    L99d1gQ dLBBJTJJr$   c                    |r|d   r|d   j                         g}t        | t        j                  g|d|i}|r|j                         }|S )Nr   r?   )rO   rL   r   USER_MODEL_EMAIL_FIELD)r8   r?   rH   r1   s       r"   
user_emailrS   x   sL    QQ 
T<>>
U
Uf
UC
iikJr$   c                     ddl m} d }|r-d}	 |j                  j                  | |      }|j                  }|S |j                  j                  | d      j                         }|S # |j
                  $ r Y |S w xY w)N   EmailAddressFT)r8   verified)r
   rW   objectsget_for_userrX   DoesNotExistfilterexists)r8   emailrW   emailaddressr1   s        r"   has_verified_emailr`      s    $L	'//<<T5IL''C
 J ""))td)CJJLJ	 (( 	 J		s   (A% %A87A8c                 d    t        ||||||      }t        j                  j                  | |      S )N)r8   email_verificationredirect_urlsignal_kwargsr0   r^   )r   r   loginperform_login)r'   r8   rb   rc   rd   r0   r^   re   s           r"   rf   rf      s9     -!#E ;;$$We44r$   c                 B    t         j                  j                  | |      S N)r   re   resume_loginr'   re   s     r"   ri   ri      s    ;;##GU33r$   c                 0   d }|r4| j                   j                  t        j                  j                        }n4| j                   j                  t        j                  j                  d       }t        |t              r	 t        j                  |      }t        j                         |j                  z
  t        j                  kD  r8d }| j                   j                  t        j                  j                  d        |S d| _        |S |S # t        $ r Y |S w xY wNT)sessiongetr   re   LOGIN_SESSION_KEYpop
isinstancedictr   deserializetimeinitiated_atr   LOGIN_TIMEOUT_account_login_accessed
ValueError)r'   peekre   datas       r"   unstash_loginr{      s    E""5;;#@#@A""5;;#@#@$G$		7%%d+E yy{U///,2L2LL##EKK$A$A4H L 37/L5L  	 L	s   >D 	DDc                 |    |j                         | j                  t        j                  j                  <   d| _        y rl   )	serializerm   r   re   ro   rw   rj   s     r"   stash_loginr~      s)    5:__5FGOOEKK112&*G#r$   c                     |i }t        j                  j                  d|j                  | |d| t	        | ||d||      S )N)senderr'   r8   T)rb   r0   rc   rd    )r   user_signed_upsend	__class__rf   )r'   r8   rb   success_urlrd   s        r"   complete_signupr      sZ     ~~wT=J - # r$   c                    ddl m} t               }t               }g }g }g }|D ]  }t	        |j
                        }	|	s|	|_        t        j                  r0t        j                  dk7  r|j                  j                  |	g      rct        j                  r;t        j                  dk(  r(|j                  r|j                  j                  |	      r|j                  |	      }
|
r?|
j                  xs |j                  |
_        |
j                  xs |j                  |
_        n6|}
|
j                  xs |j                  | |
j
                        |
_        |
||	<   |
j                  r.|j!                  |
       |
j                  r|j!                  |
       |
j                  s~|j!                  |
        |r|d   }n1|r|d   }n)|r|d   }n!|rt#        |j%                               d   }nd}|j%                         D ]<  }
|j
                  j'                         |
j
                  j'                         k(  |
_        > t#        |j%                               |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.
    rU   rV   strictr   N)r
   rW   r   r   r   r^   r   UNIQUE_EMAILPREVENT_ENUMERATIONrY   lookuprX   is_verifiedrn   primaryis_email_verifiedappendlistvaluesrO   )r'   	addressesrW   adaptere2aprimary_addressesverified_addressesprimary_verified_addressesaddressr^   aprimary_addresss               r"   cleanup_email_addressesr      s    %mG
-C!# #)#GMM2 %%00H<$$++UG4 %%00H<  $$007 GGEN		4W__AI7w'7'7AJARw'@'@!'''RAJCJ99$$Q'zz*11!4::%%a(G#)J "4Q7	,Q/	+A.	szz|,Q/ ZZ\ E#))//1QWW]]_D	E

..r$   c                    ddl m} |j                  j                  |      j	                         rJ g }t               }|j                  |       }|r)|j                   |||j                         dd             t        |      }|r)|j                   |||j                         dd             t        | ||z         \  }}|D ]  }	||	_        |	j                           |j                  j                  ||       |rU|xs dj                         |j                  j                         k7  r&t        ||j                         |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.
    rU   rV   r8   T)r8   r^   r   rX   Fr3   )r
   rW   rY   r\   r]   r   unstash_verified_emailr   rO   rS   r   r8   rF   fill_cache_for_userr^   )
r'   r8   r   rW   priority_addressesr   stashed_emailr^   r   r   s
             r"   setup_user_emailr     s<    %##***5<<>>>mG227;M!!!4!4!6t	

 tE!!d%++-PUV	
 1#i/Iw  	 ,,T9=EKR&&(GMM,?,?,AA4'		Nr$   c           	      b   ddl m} t               }d}d}|st        |      }|sH|j                  j                  |      j                  dd      j                         }|r|j                  }|r|	 |j                  j                  ||      }|:|j                  s+|j                  | ||      }|rA|j                  | |       d	}n+d}n(d	}|j                  j                  | |||d	
      }d	}|sJ |r'|j                  | t         j"                  d|| |d       |S # |j                  $ r Y w xY w)a[  
    Email verification mails are sent:
    a) Explicitly: when a user signs up
    b) Implicitly: when a user attempts to log in using an unverified
    email while EMAIL_VERIFICATION is mandatory.

    Especially in case of b), we want to limit the number of mails
    sent (consider a user retrying a few times), which is why there is
    a cooldown period before sending a new mail. This cooldown period
    can be configured in ACCOUNT_EMAIL_CONFIRMATION_COOLDOWN setting.

    TODO: This code is doing way too much. Looking up EmailAddress, creating
    if not present, etc. To be refactored.
    rU   rV   FNr   rX   pk)r0   T)r0   confirmz,account/messages/email_confirmation_sent.txt)r^   re   r0   )r
   rW   r   rS   rY   r\   order_byfirstr^   rZ   r[   rX   should_send_confirmation_mailsend_confirmation	add_emailadd_messager   INFO)	r'   r8   r0   r^   rW   r   sentemail_address
send_emails	            r"   send_email_confirmationr   @  s_    %mGDM4   ''T'2;;JMSSU 	 !''E  , 4 4 A A$ N $ ))$BB]F
 !33GF3KD"
J(00::uVT ; M D  =>f*G	 K7  ,, s   1D D.-D.c                     ddl m} t        |       }|rN|j                  j	                  | |      j                         s"|j                  j                  | |ddd       yyy)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.
    rU   rV   )r8   r^   F)r   rX   )r8   r^   defaultsN)r
   rW   rS   rY   r\   r]   get_or_create)r8   rW   r^   s      r"   sync_user_email_addressesr     sb     %tE\))00d%0HOOQ**UE-R 	+ 	
 Rur$   c            
         t         j                  rd| D cg c]   }t        di t         j                  dz   |i" }}|d   }|dd  D ]  }||z  }	 t	               j
                  j                  |      }|S  t	               j
                  j                  di t         j                  dz   | D cg c]  }|j                          c}i}|S c c}w c c}w )N__iexactr   rU   __inr   )r   rN   r   r4   r   _default_managerr\   rO   )usernameuqlistqq2r1   s         r"   filter_users_by_usernamer     s    ,, 
 I77*DaHI
 
 !H) 	BBA	//66q9 J 7n//66 
66.6717797
 J
 8s   %C
+Cr^   	is_activeprefer_verifiedc                     ddl m} t               }| j                         } t	        |j
                  j                  |       j                  d            }d}|rt	        t        d |            }|r|}d}g }|D ]4  }	t        |	j                  |       s|j                  |	j                         6 t        j                  r}|s{t        j                  | i}
 |j
                  j                  di |
}|j                         D ]:  }t        |t        j                        }t        ||       s*|j                  |       < |(t!        |      D cg c]  }|j"                  |k(  s| }}t	        t!        |            S c c}w )	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.
    rU   rV   r^   r8   Fc                     | j                   S rh   )rX   )es    r"   <lambda>z'filter_users_by_email.<locals>.<lambda>  s
    qzz r$   Tr   )r
   rW   r   rO   r   rY   r\   select_relatedr#   r^   r   r8   r   rR   iteratorr5   setr   )r^   r   r   rW   rI   mailsr   verified_mailsusersr   q_dictuser_qsr8   rS   r   s                  r"   filter_users_by_emailr     sQ    %DKKME%%,,5,9HHPQEKf%95AB"EKE !qww.LL ! **;55u=%$,,%%//$$& 	#D |'J'JKJ":u5T"	# JCq!++*BCCE
 Ds   E;!E;c                 R    t        | |      }|rt        j                  |||i      }|S rh   )r*   r   add_query_params)r'   r/   r(   next_urls       r"   passthrough_next_redirect_urlr     s/    $W.ABH&&s-@(,KLJr$   c                    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   typer6   r   
issubclassr
   	UUIDFieldrq   strhexIntegerFieldr   int)r8   rI   pk_field_classs      r"   user_pk_to_url_strr     s     D$**--(N.&"2"23dggs#77Nww{{	NF$7$7	8S\**tww<r$   c                    t               }t        |j                  j                  dd       }|rLt        |dd       r?|j                  j                  j                  j
                  j                  j                  }n|j                  j                  }t        |      }t        |t        j                        rt        |       }|j                  |      }|S |j                  |       }|S )Nremote_fieldto)r   r5   r6   r   r   r   r   r   r
   r   r   	to_python)pk_strrI   r   pk_fieldr   r   s         r"   url_str_to_user_pkr     s    D4::==.$?LdD9::==--006699::==(^N.&"5"566" # I 'Ir$   c                 
   t        |       syt        j                  syt        j                  t        j                  j
                  k(  rt        j                  sJ yt        j                  dk(  ryt        j                  du sJ y)z
    True -- email is unique
    False -- email is already in use
    None -- email is in use, but we should hide that using email verification.
    TFNr   )r   r   r   EMAIL_VERIFICATIONEmailVerificationMethod	MANDATORYr   s    r"   assess_unique_emailr     sx     !'--''//99	: ////		)	)X	5
 //4777
 r$   c                     |j                   }t        j                  j                  |j                  | |||       |r?t               j                  d||j                  |j                  d|j                         y y )N)r   r'   r8   from_email_addressto_email_addresszaccount/email/email_changed)
from_emailto_email)contextr^   )r8   r   email_changedr   r   r   send_notification_mailr^   )r'   r   r   r8   s       r"   emit_email_changedr     s      D~~-)   ,,)066,22 %** 	- 	
 r$   rh   )NNNFN)F)FN)NF)r   N)Crt   r   collectionsr   typingr   django.confr   django.contribr   django.contrib.authr   r   django.core.exceptionsr	   	django.dbr
   django.db.modelsr   django.utils.encodingr   django.utils.httpr   r   allauth.accountr   r   allauth.account.adapterr   allauth.account.internalr   allauth.account.modelsr   allauth.core.internalr   allauth.utilsr   r   r   boolr#   r   r*   r.   r<   r9   r>   rL   rP   rS   r`   rf   ri   r{   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r$   r"   <module>r     sb     #    # C 4   + : 1 / * ( ) 4  "5
c]
 +>u$  =# =(# ( +0 $6 ', K $) D & 
5(4(+
 F/R"J<~
$* KP''#D>'CG'T $!(4. !H
r$   