startup.py

import uvicorn
import sys
from argparse import ArgumentParser
from app.core import config  
import os
from pathlib import Path

argspack = ArgumentParser(description='Image-fastapi startup script help',add_help=True)
argspack.add_argument('-p', '--port', type=int, default=5000, help='Select which port to run the server, defaults to 5000')
argspack.add_argument('-r', '--noreload', type=bool, default=False, help='Select to disable autoreload or not, defaults to False')
argspack.add_argument('-n','--noauth', type=bool,default=False,help='Select to disable authentication on the endpoints or not, defaults to False')
args = argspack.parse_args()
abs_module_path = str(Path(config.__file__).parent.resolve()).split('/app/core')[:-1][0] #We'll use the empty configs module's file attribute to determine the absolute path of our module.
jwt_key_rdir = config.JWT_KEY_DIR
if jwt_key_rdir.startswith('*'):
    os.environ["JWT_KEY_DIR"] = jwt_key_rdir.replace('*', abs_module_path) # Using local module directory if JWT key directory starts with astericks
    config.JWT_KEY_DIR = os.environ["JWT_KEY_DIR"] 


os.path.isdir(config.IMG_CACHE_DIR) or os.makedirs(config.IMG_CACHE_DIR,exist_ok=True)
os.path.isdir(config.IMG_SAVE_DIR) or os.makedirs(config.IMG_SAVE_DIR,exist_ok=True)
os.environ["API_AUTH_CFG"] = str(int(not args.noauth))


def main():
    uvicorn.run('app.main:app', proxy_headers=True,forwarded_allow_ips='*',port=args.port ,reload= not args.noreload)

if __name__ == '__main__':
   main()