FORMAT_CONVERSION_MAXCOMPUTE/format_convert/kill_all.py

import os
import re
import sys
import psutil
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + "/../")
from format_convert.utils import get_ip_port, get_intranet_ip, get_using_ip, get_args_from_config

import time

ip_port_dict = get_ip_port()
# ip = "http://127.0.0.1"
ip = get_using_ip()
# python_path = ip_port_dict.get(ip).get("MASTER").get("python_path")
# gunicorn_path = ip_port_dict.get(ip).get("MASTER").get("gunicorn_path")
# project_path = ip_port_dict.get(ip).get("MASTER").get("project_path")
python_path = get_args_from_config(ip_port_dict, ip, "python_path")[0]
project_path = get_args_from_config(ip_port_dict, ip, "project_path")[0]
gunicorn_path = get_args_from_config(ip_port_dict, ip, "gunicorn_path")[0]


def kill():
    if python_path and project_path:
        pid_list = psutil.pids()
        for pid in pid_list:
            try:
                process = psutil.Process(pid)
            except:
                continue
            process_cmd = ''
            for c in process.cmdline():
                process_cmd += c + " "
            if process_cmd.strip() == "":
                continue
            if "monitor" in process_cmd or "kill" in process_cmd:
                continue

            if re.search(project_path, process_cmd):
                comm = "kill -9 " + str(pid)
                print(comm, process_cmd)
                os.system(comm)

    else:
        print("cannot kill! checkout config...")
        print(ip_port_dict)
        print(ip, python_path, project_path)


if __name__ == "__main__":
    for _ in range(3):
        kill()
        time.sleep(0.5)