Browse Source

update APP_PREFIX

Zhang Li 2 years ago
parent
commit
d646f8d6fb
1 changed files with 11 additions and 4 deletions
  1. 11 4
      server.py

+ 11 - 4
server.py

@@ -19,10 +19,15 @@ from app.routers.run import router_af_run
 from app.routers.job import router_af_job
 from app.routers.task import router_af_task
 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 跨源资源共享
@@ -51,13 +56,15 @@ 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!!"