Sfoglia il codice sorgente

返回增加库名

liweiquan 2 anni fa
parent
commit
1a4fda6b29
1 ha cambiato i file con 9 aggiunte e 2 eliminazioni
  1. 9 2
      app/routers/data_management.py

+ 9 - 2
app/routers/data_management.py

@@ -16,6 +16,8 @@ from utils.sx_web import web_try
 from app.common.hive import hiveDs
 
 from app import get_db
+from configs.settings import DefaultOption, config
+database_name = config.get('HIVE', 'DATABASE_NAME')
 
 router = APIRouter(
     prefix="/jpt/datamanagement",
@@ -39,13 +41,18 @@ def create_data_management(item: schemas.DataManagementCreate, db: Session = Dep
 @web_try()
 @sxtimeit
 def get_data_managements(user_id: str, project_id: str, db: Session = Depends(get_db)):
-    return crud.get_data_managements(db, user_id, project_id)
+    res = crud.get_data_managements(db, user_id, project_id)
+    for item in res:
+        item.table_name = f'{database_name}.{item.table_name}'
+    return res
 
 @router.get("/local")
 @web_try()
 @sxtimeit
 def get_local_data_managements(db: Session = Depends(get_db)):
-    return hiveDs.list_tables()
+    t_list = hiveDs.list_tables()
+    res = [f'{database_name}.{t}' for t in t_list]
+    return res
 
 @router.get("/table_schema")
 @web_try()