U
    ]Úd’  ã                   @   s.   d dl T d dlmZ G dd„ dejjeƒZdS )é    )Ú*)ÚResponseMixinc                   @   s@   e Zd ZdZejjdd„ ƒZejjdd„ ƒZejjdd„ ƒZ	dS )	ÚTwoFactorViewz/two-factorc                 C   s
   |   d¡S )Nztwo_factor.html)Zrender©Úself© r   ú"./modules/auth/views/two_factor.pyÚget   s    zTwoFactorView.getc                 C   s,   t j ¡  t jr|  dt j¡S |  dt j¡S )NÚfailÚok)ÚgÚsessionZtfa_send_codeZalertÚ_errorÚ_okÚauthr   r   r   r   Ú	send_code   s    
zTwoFactorView.send_codec                 C   s2   t j tjd ¡r |  dt j¡S |  dt j¡S d S )NÚcoder   r
   )r   r   Ztfa_check_codeZrequestZjsonr   r   r   r   r   r   Ú
check_code   s    zTwoFactorView.check_codeN)
Ú__name__Ú
__module__Ú__qualname__Z__url__ÚcmfÚviewsÚactionr	   r   r   r   r   r   r   r      s   

r   N)Zcmf.includeZmodules.auth.views.responser   r   r   ZBaseViewr   r   r   r   r   Ú<module>   s   