使用Python编写脚本-根据端口号杀掉进程

我的GitHub：Powerveil - GitHub
我的Gitee：Powercs12 - Gitee

皮卡丘每天学Java

从前段开始遇到一个问题，服务在启动的时候总是端口被占用，发现还是Java程序，但是当时并没有启动Java程序，电脑出问题了。

一个办法是重启，另一个办法就是手动杀掉进程。

第一种当然不推荐，毕竟服务多起来，重启电脑，重启环境需要不少时间[痛]，不能在这上面花费太多时间。

第二种方法是自己找

netstat -ano|findstr "端口号"

taskkill /F /PID 进程pid

如果每次查看删除进程都要先查看再删除，这也没啥事，最重要的是有时候忘记命令还要上网搜一下，这个浪费时间。所以我准备写一个程序完成这项操作。本来写了个bat半成品脚本，看到一片文章一句话直接放弃了。windows bat批处理脚本由于低成本、高效益，从某种角度上说更像是一门艺术，人们用其可以以更简单的方式完成复杂的任务。遗憾的是，随着c、java、python、golang、javascript等高级语言的蓬勃发展，选择使用传统脚本方式解决问题的人员越来越少，甚至很多类脚本任务也通过perl或python等高级语言变通实现。

为啥不用高级编程语言编写，非要再学一下bat脚本呢？我决定使用Python来编写，发现确实方便，有现成的接口供调用。下面是Python代码。

import psutil
import re

def get_process_by_port(port):
    # 创建一个集合用于存储已经发现的进程ID
    found_pids = set()

    # 创建一个列表用于存储找到的进程ID
    process_ids = []

    for conn in psutil.net_connections(kind='inet'):
        if conn.laddr.port == port:
            pid = conn.pid
            # 如果当前进程ID已经在集合中，则跳过
            if pid in found_pids:
                continue
            found_pids.add(pid)
            try:
                process = psutil.Process(pid)
                process_info = {"pid": process.pid, "name": process.name()}
                process_ids.append(process_info)
            except psutil.NoSuchProcess:
                process_info = {"pid": pid, "name": "未知"}
                process_ids.append(process_info)
    return process_ids


def is_positive_integer(str):
    # 使用正则表达式判断字符串是否是大于零的整数
    return bool(re.match(r'^[1-9]\d*$', str))

def kill_processes(processes):
    if not processes:
        print("未找到该端口的进程")
        return

    for index, process in enumerate(processes):
        print("索引: {}, 进程ID: {}, 进程名称: {}".format(
            index + 1, process["pid"], process["name"]))

    while True:
        flagStr = str(input("是否删除该进程(Y/N，删除下标（默认第一个）。[Y 1]或[1])"))
        # print("flagStr:" + flagStr)

        index = 0
        if is_positive_integer(flagStr[0]):
            flag = 'Y'
            index = int(flagStr[0]) - 1
        else:
            flag = flagStr[0].upper()
            if len(flagStr) == 3:
                index = int(flagStr[2]) - 1

        if flag == 'Y':
            process = processes[index]
            psutil.Process(process["pid"]).terminate()
            print("您已经删除了该进程！！！")
            print("已删除进程信息==>索引: {}, 进程ID: {}, 进程名称: {}".format(
                index + 1, process["pid"], process["name"]))
            break
        elif flag == 'N':
            print("您取消了删除进程")
            break
        else:
            print("是否删除该进程(Y/N，删除下标（默认第一个）。[Y 1]或[1])")


def main():
    while True:
        input_info = input("请输入端口号（输入N退出）：")
        if is_positive_integer(input_info):
            port_number = int(input_info)
            processes = get_process_by_port(port_number)
            kill_processes(processes)
        elif input_info.upper() == 'N':
            exit()
        else:
            print("请输入正确的端口号")
            continue



if __name__ == '__main__':
    main()