|
@@ -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!!"
|
|
|
|