Format: application/x-python-code
1 o
2
3 Gh_ � @ s4 d dl mZ d dlmZmZmZ d dlZdd� ZdS )� )�jsonify)�create_access_token� jwt_required�get_jwt_identityNc C sN d| j d< d| j d< | jddgd�dd � �}| jd
4 d gd�t� d d
5 � ��}d S )Nz super-secretZJWT_SECRET_KEYi ZJWT_ACCESS_TOKEN_EXPIRESz
6 /api/login�POST)�methodsc S s� t j�d�} t j�d�}dt�d�� ��� d�}| r |s$tddi�dfS t�|�� ��� }| |d krE||d krEt| d�}t|d �d
7 fS tdd i�d fS )
8 N�username�passwordZadmin)r r �errorzMissing credentialsi� )�identity)� access_token�� zInvalid credentialsi� ) �request�json�get�hashlib�sha256�encode� hexdigestr r )r r Z test_userZ hashed_pwr � r �CC:\Users\shubh\Videos\agent\agent_preprod\modules\authentication.py�login
9 s �
10 z configure_auth.<locals>.loginz/api/protected�GETc S s t � } t| d�dfS )N)Z logged_in_asr
11 )r r )� current_userr r r � protected! s z!configure_auth.<locals>.protected)�config�router )�appr r r r r �configure_auth s
12
13
14 r )�flaskr Zflask_jwt_extendedr r r r r r r r r �<module> s