Explorar o código

update prefix env

Zhang Li %!s(int64=2) %!d(string=hai) anos
pai
achega
cc0c1d7b71
Modificáronse 1 ficheiros con 10 adicións e 4 borrados
  1. 10 4
      server.py

+ 10 - 4
server.py

@@ -22,10 +22,15 @@ from app.routers.job import router_af_job
 from app.routers.task import router_af_task
 from configs.globals import GlobalsMiddleware
 from utils.sx_log import format_print
+import os
 
 format_print()
+PREFIX = os.environ.get('APP_PREFIX', default='jpt')
+print(f'prefix: {PREFIX}')
+
+
 Base.metadata.create_all(bind=engine)
-app = FastAPI(docs_url='/jpt/docs', redoc_url='/jpt/redoc', openapi_url='/jpt/openapi.json', title="DAG In Airflow管理系统")
+app = FastAPI(docs_url=f'/{PREFIX}/docs', redoc_url=f'/{PREFIX}/redoc', openapi_url=f'/{PREFIX}/openapi.json', title="DAG In Airflow管理系统")
 
 
 # CORS 跨源资源共享
@@ -57,13 +62,14 @@ app.include_router(router_jm_job_info.router)
 app.include_router(router_jm_job_log.router)
 app.include_router(router_code_check.router)
 
-# AirflowJobSubmitter.auto_submit_data_transfer()
-# AirflowJobSubmitter.auto_submit_requirements_install()
+if PREFIX == 'af':
+    AirflowJobSubmitter.auto_submit_data_transfer()
+    AirflowJobSubmitter.auto_submit_requirements_install()
 print('server init finish:)!!!')
 
 
 # Get 健康检查
-@app.get("/jpt/ping", description="健康检查")
+@app.get(f"/{PREFIX}/ping", description="健康检查")
 def ping():
     return "pong!!"