Update GetResultToSqlflow.py

sqlparser · Aug 9, 2023 · 668c7ad · 668c7ad
1 parent c73a03c
commit 668c7ad
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/api/python/advanced/GetResultToSqlflow.py b/api/python/advanced/GetResultToSqlflow.py
@@ -10,13 +10,13 @@
 def getResult(download, userId, token, server, port, jobId, filePath):
     sep = 'data' + os.sep + 'result' + os.sep
     filePath = filePath + '_' + jobId
-    if download == '1':
+    if download == 'json':
         url = "/gspLive_backend/sqlflow/job/exportLineageAsJson"
         filePath = sep + filePath + '_json.json'
-    elif download == '3':
+    elif download == 'graphml':
         url = "/gspLive_backend/sqlflow/job/exportLineageAsGraphml"
         filePath = sep + filePath + '_graphml.graphml'
-    elif download == '2':
+    elif download == 'csv':
         url = "/gspLive_backend/sqlflow/job/exportLineageAsCsv"
         filePath = sep + filePath + '_csv.csv'
     else: