
    Ii                        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 dlmZ d dlmZmZmZ d dlmZmZ d	ej(                  _        ej(                  j,                  d
        Z G d de      Z ej2                  e      d        Zy)    )absolute_import)app)db_model)current_user)identity_loadedRoleNeedUserNeed)	FlaskForm)StringFieldPasswordFieldSubmitField)DataRequiredLengthz/loginc                 T    t         j                  j                  j                  |       S )N)r   	AdminUserqueryget)user_ids    */var/www/dostadmin/dostadmin/loginutils.py	load_userr      s    ##''00    c            	           e Zd Z ed ed       eddd      g      Z ed ed       edd	d      g      Z e	d
      Z
y)	LoginFormUsernamezInvalid username/password)message      )minmaxr   Password      LoginN)__name__
__module____qualname__r   r   r   usernamer   passwordr   submit r   r   r   r      s`    !<=qb*EF	
H !<=qb*EF	
H !Fr   r   c                 F   t         |_        t        t         d      r2|j                  j	                  t        t         j                               t        t         d      rDt         j                  D ]0  }|j                  j	                  t        |j                               2 y y )Nidroles)
r   userhasattrprovidesaddr	   r,   r-   r   name)senderidentityroles      r   on_identity_loadedr6   $   sr     !HM |T"h|78 |W% &&D!!(499"56 ' &r   N)
__future__r   	dostadminr   r   flask_loginr   flask_principalr   r   r	   	flask_wtfr
   wtformsr   r   r   wtforms.validatorsr   r   login_manager
login_viewuser_loaderr   r   connect_viar6   r*   r   r   <module>rB      s~    &   $ ? ?  ; ; 3'    1 1"	 "$ S!7 "7r   