97 lines
2.9 KiB
Diff
97 lines
2.9 KiB
Diff
From 5866f586616e48dfea53107cf6f8c3364c658951 Mon Sep 17 00:00:00 2001
|
|
From: lixin <lixinb@uniontech.com>
|
|
Date: Thu, 9 Nov 2023 17:08:00 +0800
|
|
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=BB=BB=E5=8A=A1=E6=B5=81?=
|
|
=?UTF-8?q?=E7=8A=B6=E6=80=81=E6=8E=A5=E5=8F=A3?=
|
|
MIME-Version: 1.0
|
|
Content-Type: text/plain; charset=UTF-8
|
|
Content-Transfer-Encoding: 8bit
|
|
|
|
---
|
|
index.py | 13 ++++++++++++-
|
|
views/server.py | 31 ++++++++++++++++++++++++++++++-
|
|
2 files changed, 42 insertions(+), 2 deletions(-)
|
|
|
|
diff --git a/index.py b/index.py
|
|
index 0c64990..059431b 100644
|
|
--- a/index.py
|
|
+++ b/index.py
|
|
@@ -4,7 +4,6 @@
|
|
# SPDX-License-Identifier: MulanPubL-2.0-or-later
|
|
|
|
import os
|
|
-import sys
|
|
import json
|
|
from sysmig_agent import share
|
|
from views import migration, server
|
|
@@ -28,6 +27,7 @@ mods = {
|
|
'migration_details': migration.migration_details,
|
|
'import_host_info': server.import_host_info,
|
|
'host_info_display': server.host_info_display,
|
|
+ 'sql_task': server.modify_task_stream,
|
|
}
|
|
|
|
@app.route('/import_host_info', methods=['GET', 'POST'])
|
|
@@ -52,6 +52,17 @@ def host_info_display():
|
|
return Response(mod, content_type='application/json')
|
|
|
|
|
|
+@app.route('/sql_task', methods=['GET', 'POST'])
|
|
+def modify_task_stream():
|
|
+ """
|
|
+ 修改任务流
|
|
+ :return:
|
|
+ """
|
|
+ mod = check_methods()
|
|
+ if mod:
|
|
+ return Response(mod, content_type='application/json')
|
|
+
|
|
+
|
|
def check_methods():
|
|
if request.method == 'POST':
|
|
data = request.get_data()
|
|
diff --git a/views/server.py b/views/server.py
|
|
index d2ca719..a00e696 100644
|
|
--- a/views/server.py
|
|
+++ b/views/server.py
|
|
@@ -132,4 +132,33 @@ def host_info_display(data):
|
|
res['size'] = size
|
|
|
|
json_res = json.dumps(res)
|
|
- return json_res
|
|
\ No newline at end of file
|
|
+ return json_res
|
|
+
|
|
+
|
|
+def modify_task_stream(data):
|
|
+ """
|
|
+ 修改任务流状态
|
|
+ :return:
|
|
+ """
|
|
+ task_id = json.loads(data).get('task_id')
|
|
+ get_task_status_sql = "select task_status,task_stream_id from agent_task where task_id='%s';" % task_id
|
|
+ info = DBHelper().execute(get_task_status_sql).fetchone()
|
|
+ task_status = info[0]
|
|
+ task_stream_id = info[1]
|
|
+ if task_status == 0:
|
|
+ task_status = 'None'
|
|
+ elif task_status == 1:
|
|
+ task_status = 'Doing'
|
|
+ elif task_status == 2:
|
|
+ task_status = 'Done'
|
|
+ else:
|
|
+ task_status = 'Cancel'
|
|
+ time = datetime.now()
|
|
+ modify_task_status_sql = "update cur_task set task_status='%s',stream_Updatetime='%s' " \
|
|
+ "where task_id ='%s';" % (task_status, time, task_id)
|
|
+ DBHelper().execute(modify_task_status_sql)
|
|
+
|
|
+ modify_task_stream_sql = "update task_stream set stream_status='%s',stream_Updatetime='%s' " \
|
|
+ "where task_stream_id='%s';" % (task_status, time, task_stream_id)
|
|
+ DBHelper().execute(modify_task_stream_sql)
|
|
+ return 'success'
|
|
\ No newline at end of file
|
|
--
|
|
2.20.1
|
|
|