65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
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)
|