Python3 自动杀掉程序

kill-program-name.py
1.根据程序名

#!/user/bin/env python3
# -*- coding: utf-8 -*-
import os, signal, sys
import argparse


def kill(pid):
    print('pid', pid)
    os.kill(pid, signal.SIGKILL)
    print('已杀死pid为%s' % pid)


def kill_target(target):
    cmd_run = "ps aux | grep {}".format(target)
    out = os.popen(cmd_run).read()
    for line in out.splitlines():
        print(line)
        # 程序名或程序路径
        if resource_path(target) in line:
            pid = int(line.split()[1])
            kill(pid)


# 获取根目录 测试在UBuntu上不需要根目录
def resource_path(file_name):
    dir_name = os.path.split(os.path.abspath(sys.argv[0]))[0]
    return dir_name + "/" + str(file_name)


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='program to kill')
    parser.add_argument(
        '--name', help='program name', type=str, required=True)
    args = parser.parse_args()
    kill_target(args.name)

使用方式

 python3 kill-program-name.py --name 程序名

2.根据端口
kill-port.py

#!/user/bin/env python3
# -*- coding: utf-8 -*-
import os, signal
import argparse
import time


def kill_port(target):
    cmd_run = 'lsof -i tcp:{}'.format(target)
    os.system(cmd_run)
    out = os.popen(cmd_run).read()
    for line in out.splitlines():
        print(line)
        if 'IPv4' in line or 'IPv6' in line:
            pid = str(line.split()[1])  # 得到pid
            print('pid=' + pid)
            time.sleep(3)
            os.system('kill -9 %s' % pid)
            out = os.popen(cmd_run).read()
            if out != "":
                print('kill 完成')


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='occupant port to kill')
    parser.add_argument(
        '--port', help='port number', type=str, required=True)
    args = parser.parse_args()
    kill_port(args.port)

使用方式

python3 kill-port.py --port 8888

参考:Python 用一个脚本来控制另一个脚本的运行、关闭