Python Process_Manager

外星球的地球人是外星人吗？

于 2023-07-13 17:54:11 发布

阅读量254

点赞数

文章标签： python 开发语言

本文链接：https://blog.csdn.net/wjl__ai__/article/details/131708420

版权

该代码定义了一个`Process_Manager`类，用于管理指定的Python进程。它能检查进程是否存在，如果不存在则启动，如果存在多个则认为异常。还能重启进程，通过`pgrep`获取进程ID，并使用`subprocess`模块执行操作如启动、杀死进程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import subprocess


class Process_Manager:
    def __init__(self, process_name, log_name):
        self.process_name = process_name
        self.log_name = log_name

    def check(self):
        # 执行命令，获取进程 ID
        pids = self._get_pids()
        if len(pids) < 2:
            print(f"进程 {self.process_name} 不存在， 尝试重新拉起...")
            self.start()
            return
        elif len(pids) > 2:
            print(f"进程 {self.process_name} 异常...")
        print("进程运行中, ID:", pids[0])

    def restart(self):
        pids = self._get_pids()
        self.kill(pids)
        self.start()

    def start(self):
        command = f"nohup python {self.process_name} > {self.log_name} 2>&1 &"
        subprocess.call(command, shell=True)

    def kill(self, pids=None):
        if pids is None:
            pids = self._get_pids()
        print("退出进程！！")
        for p in pids[:-1]:
            command = f"kill {p}"
            subprocess.call(command, shell=True)
            print(command)

    def _get_pids(self):
        command = f"pgrep -f 'python {self.process_name}'"
        process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        output, _ = process.communicate()
        pids = output.decode().split("\n")[:-1]
        return pids