Browse Source

11-11 新增公众号 银发生活畅谈
新增文章

luojunhui 5 months ago
parent
commit
9c80edfbc4
2 changed files with 46 additions and 0 deletions
  1. 20 0
      routes/__init__.py
  2. 26 0
      routes/process_killer.py

+ 20 - 0
routes/__init__.py

@@ -9,6 +9,7 @@ from quart import Blueprint, jsonify, request
 from .insertVideoRoute import insert
 from .sourceIdRoute import saveSourceToDB
 from .recallPQVideos import recallPQVideos
+from .process_killer import kill_task_by_name
 
 from deal import tempServer
 
@@ -88,5 +89,24 @@ def Routes(db_client):
         response = ts.chooseCards()
         return jsonify(response)
 
+    @TL_blueprint.route("/KILL", methods=['GET'])
+    async def kill():
+        """
+        杀掉进程
+        :return:
+        """
+        task_name = request.args.get("taskName")
+        if task_name:
+            if task_name in ['historyTask.py', 'newContentIdTask.py']:
+                status = await kill_task_by_name(task_name)
+                if status:
+                    return jsonify({"message": "success"})
+                else:
+                    return jsonify({"message": "failed"})
+            else:
+                return jsonify({"message": "task name error"})
+        else:
+            return jsonify({"message": "no task name"})
+
     return TL_blueprint
 

+ 26 - 0
routes/process_killer.py

@@ -0,0 +1,26 @@
+"""
+@author: luojunhui
+"""
+import subprocess
+
+
+async def kill_task_by_name(task_name) -> bool:
+    """
+    通过进程名称来杀掉进程
+    :param task_name:
+    """
+    try:
+        # 构造命令
+        command = f"ps aux | grep {task_name} | grep -v grep | awk '{{print $2}}' | xargs kill -9"
+        print(command)
+
+        # 执行命令
+        result = subprocess.run(command, shell=True, text=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+        # 检查执行结果
+        if result.returncode == 0:
+            return True
+        else:
+            return False
+    except Exception as e:
+        return False