a
    _mi                     @   s   d dl mZmZmZmZmZ d dlmZ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 ejddd	gd
dd Zejddd	gd
dd Zeddd ZdS )    )render_templateredirecturl_forflashrequest)
login_userlogout_usercurrent_user)db)bp)	LoginFormRegisterForm)Userz/loginGETPOST)methodsc                  C   s   t jrttdS t } |  rvtjj| j	j
d }|rl|| jj
rlt| tjd}t|phtdS tdd td| dS )Nindex)emailnextzInvalid email or password.errorz
login.htmlform)r	   is_authenticatedr   r   r   validate_on_submitr   query	filter_byr   datafirstZcheck_passwordpasswordr   r   argsgetr   r   )r   userZ	next_page r"   S/home/japhy/trackingScienceReplication.artiswrong.com/app/blueprints/auth/routes.pylogin	   s    
r$   z	/registerc                  C   s   t jrttdS t } |  rtt| jj| j	j| j
jd}|| jj tj| tj  tdd ttdS td| dS )Nr   )r   nameinstitutionz'Registration successful. Please log in.successz
auth.loginzregister.htmlr   )r	   r   r   r   r   r   r   r   r   r%   r&   Zset_passwordr   r
   sessionaddcommitr   r   )r   r!   r"   r"   r#   register   s    

r+   z/logoutc                   C   s   t   ttdS )Nr   )r   r   r   r"   r"   r"   r#   logout(   s    r,   N)flaskr   r   r   r   r   flask_loginr   r   r	   app.extensionsr
   app.blueprints.authr   Zapp.blueprints.auth.formsr   r   Zapp.models.userr   router$   r+   r,   r"   r"   r"   r#   <module>   s   

