From 10389769f980397f96b317cdd8e716cd03a98bae Mon Sep 17 00:00:00 2001
From: xuezhixin <xuezhixin@uniontech.com>
Date: Mon, 4 Nov 2024 14:18:49 +0800
Subject: [PATCH] modify task stream interface

---
 0014-modify-task-stream-interface.patch | 96 +++++++++++++++++++++++++
 migration-tools.spec                    |  6 +-
 2 files changed, 101 insertions(+), 1 deletion(-)
 create mode 100644 0014-modify-task-stream-interface.patch

diff --git a/0014-modify-task-stream-interface.patch b/0014-modify-task-stream-interface.patch
new file mode 100644
index 0000000..61d8ca5
--- /dev/null
+++ b/0014-modify-task-stream-interface.patch
@@ -0,0 +1,96 @@
+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
+
diff --git a/migration-tools.spec b/migration-tools.spec
index b1275fb..17d64fc 100644
--- a/migration-tools.spec
+++ b/migration-tools.spec
@@ -1,6 +1,6 @@
 Name:       migration-tools
 Version:    1.0.2
-Release:    13
+Release:    14
 License:    MulanPSL-2.0
 Summary:    A tool to help users migrate the Centos system to the UOS system and openEuler system
 Source0:    %{name}-%{version}.tar.gz
@@ -20,6 +20,7 @@ Patch10:    0010-create-task-stream-interface.patch
 Patch11:    0011-replace-agent-with-func-diractory.patch
 Patch12:    0012-add-agent_task-in-database.patch
 Patch13:    0013-display-agent-host-information.patch
+Patch14:    0014-modify-task-stream-interface.patch
 
 BuildArch:  noarch
 BuildRequires:  systemd	
@@ -118,6 +119,9 @@ rm -rf /usr/bin/migration-tools
 %endif
 
 %changelog
+* Mon Nov 4 2024 xuezhixin <xuezhixin@uniontech.com> - 1.0.2-14
+- 0014-modify-task-stream-interface.patch
+
 * Mon Nov 4 2024 xuezhixin <xuezhixin@uniontech.com> - 1.0.2-13
 - 0013-display-agent-host-information.patch
 
-- 
Gitee