Migrate to FastAPI JWT auth
This commit is contained in:
61
api/app.py
61
api/app.py
@@ -1,64 +1,5 @@
|
||||
import json
|
||||
import os
|
||||
from datetime import timedelta, datetime, timezone
|
||||
|
||||
import uvicorn
|
||||
|
||||
from fastapi import FastAPI
|
||||
|
||||
from mongoengine import connect
|
||||
from flask_jwt_extended import create_access_token, get_jwt, get_jwt_identity, JWTManager
|
||||
|
||||
from database.utils import create_admin_user
|
||||
|
||||
# Initialize Flask app
|
||||
app = FastAPI()
|
||||
|
||||
# Set JWT key from environment variable
|
||||
# try:
|
||||
# app.config["JWT_SECRET_KEY"] = os.environ["TAILFIN_JWT_KEY"]
|
||||
# except KeyError:
|
||||
# app.logger.error("Please set 'TAILFIN_JWT_KEY' environment variable")
|
||||
# exit(1)
|
||||
|
||||
# Set JWT keys to expire after 1 hour
|
||||
# app.config["JWT_ACCESS_TOKEN_EXPIRES"] = timedelta(hours=1)
|
||||
|
||||
# Initialize JWT manager
|
||||
# jwt = JWTManager(app)
|
||||
|
||||
# Connect to MongoDB
|
||||
connect('tailfin')
|
||||
|
||||
# @app.after_request
|
||||
# def refresh_expiring_jwts(response):
|
||||
# """
|
||||
# Refresh/reissue JWTs that are near expiry following each request containing a JWT
|
||||
#
|
||||
# :param response: Response given by previous request
|
||||
# :return: Original response with refreshed JWT
|
||||
# """
|
||||
# try:
|
||||
# exp_timestamp = get_jwt()["exp"]
|
||||
# now = datetime.now(timezone.utc)
|
||||
# target_timestamp = datetime.timestamp(now + timedelta(minutes=30))
|
||||
# if target_timestamp > exp_timestamp:
|
||||
# app.logger.info("Refreshing expiring JWT")
|
||||
# access_token = create_access_token(identity=get_jwt_identity())
|
||||
# data = response.get_json()
|
||||
# if type(data) is dict:
|
||||
# data["access_token"] = access_token
|
||||
# response.data = json.dumps(data)
|
||||
# return response
|
||||
# except (RuntimeError, KeyError):
|
||||
# # No valid JWT, return original response
|
||||
# app.logger.info("No valid JWT, cannot refresh expiry")
|
||||
# return response
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
# Create default admin user if it doesn't exist
|
||||
create_admin_user()
|
||||
|
||||
# Start the app
|
||||
uvicorn.run("fastapi_code:app", reload=True)
|
||||
uvicorn.run("app.api:app", host="0.0.0.0", port=8081, reload=True)
|
||||
|
Reference in New Issue
Block a user