Startonix / Modular-AI

Advanced AI Training and Building Repository
0 stars 0 forks source link

JWT Authentication #150

Open Startonix opened 4 months ago

Startonix commented 4 months ago

jwt_authentication.py

from flask import Flask, request, jsonify import jwt

app = Flask(name) app.config['SECRET_KEY'] = 'supersecretkey'

@app.route('/login', methods=['POST']) def login(): auth_data = request.json token = jwt.encode({'user': auth_data['username']}, app.config['SECRET_KEY']) return jsonify({'token': token})

@app.route('/protected', methods=['GET']) def protected(): token = request.headers.get('Authorization') if not token: return jsonify({'message': 'Token is missing!'}), 403 try: data = jwt.decode(token, app.config['SECRET_KEY']) except: return jsonify({'message': 'Token is invalid!'}), 403 return jsonify({'message': 'Protected content!'})

if name == 'main': app.run()