U
    XHg                     @   s   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
d ZG dd deZeedd ZdS )    )absolute_import)app)db_model)current_user)identity_loadedRoleNeedUserNeed)	FlaskForm)StringFieldPasswordFieldSubmitField)DataRequiredLengthz/loginc                 C   s   t jj| S )N)r   	AdminUserqueryget)user_id r   */var/www/dostadmin/dostadmin/loginutils.py	load_user   s    r   c                   @   sP   e Zd ZededdeddddgZededdedd	ddgZe	d
Z
dS )	LoginFormZUsernamezInvalid username/password)message      )minmaxr   ZPassword      ZLoginN)__name__
__module____qualname__r
   r   r   usernamer   passwordr   submitr   r   r   r   r      s   r   c                 C   sN   t |_tt dr"|jtt j tt drJt jD ]}|jt|j	 q2d S )Nidroles)
r   userhasattrprovidesaddr   r$   r%   r   name)senderidentityroler   r   r   on_identity_loaded$   s    


r.   N)
__future__r   	dostadminr   r   flask_loginr   flask_principalr   r   r   Z	flask_wtfr	   Zwtformsr
   r   r   Zwtforms.validatorsr   r   login_manager
login_viewuser_loaderr   r   connect_viar.   r   r   r   r   <module>   s   
