利用Python调用Windows API，实现任务管理器功能

任务管理器具体功能有：

1、 列出系统当前所有进程。
2、 列出隶属于该进程的所有线程。
3、 如果进程有窗口，可以显示和隐藏窗口。
4、 强行结束指定进程。

通过Python调用Windows API还是很实用的，能够结合Python的简洁和Windows API的强大，写出各种各样的脚本。

编码中的几个难点有：

1、API的入参是结构体时，怎么解决？

答：Python内手动建立结构体。

详见：https://baijiahao.baidu.com/s?id=1615380723396624885&wfr=spider&for=pc

2、调用API的具体方式是什么？

答：通过win32模块或ctypes模块。前者更简便，后者函数库更全。

3、不熟悉Windows API怎么办？

通过API伴侣这个软件查询API所在的DLL库，通过MSDN查询API的详细解释等等。

 

完整代码如下：

import os
# import win32api
import win32gui
import win32process
from ctypes import *


# 列出系统当前所有进程。
def getProcessList():
    os.system("tasklist")


# 结构体
class THREADENTRY32(Structure):
    _fields_ = [('dwSize', c_ulong),
                ('cntUsage', c_ulong),
                ('th32ThreadID', c_ulong),
                ('th32OwnerProcessID', c_ulong),
                ('tpBasePri', c_long),
                ('tpDeltaPri', c_long),
                ('dwFlags', c_ulong)]


# 获取指定进程的所有线程
def getThreadOfProcess(pid):
    dll = windll.LoadLibrary("KERNEL32.DLL")
    snapshotHandle = dll.CreateToolhelp32Snapshot(0x00000004, pid)
    struct = THREADENTRY32()
    struct.dwSize = sizeof(THREADENTRY32)
    flag = dll.Thread32First(snapshotHandle, byref(struct))

    while flag != 0:
        if(struct.th32OwnerProcessID == int(pid)):
            print("线程id："+str(struct.th32ThreadID))
        flag = dll.Thread32Next(snapshotHandle, byref(struct))
    dll.CloseHandle(snapshotHandle)


# EnumWindows的回调函数
def callback(hwnd, windows):
    pidList = win32process.GetWindowThreadProcessId(hwnd)
    for pid in pidList:
        windows.setdefault(pid, [])
        windows[pid].append(hwnd)


# 显示和隐藏指定进程的窗口
def changeWindowState(pid, status):
    windows = {}
    win32gui.EnumWindows(callback, windows)
    try:
        hwndList = windows[int(pid)]
        # 显示/隐藏窗口
        for hwnd in hwndList:
            win32gui.ShowWindow(hwnd, int(status))
    except:
        print("进程不存在")


# 强行结束指定进程
def killProcess(pid):
    cmd = 'taskkill /pid ' + pid + ' /f'
    try:
        os.system(cmd)
    except Exception as e:
        print(e)


if __name__ == "__main__":
    while(True):
        print()
        print()
        print("************************************")
        print("*                                  *")
        print("*     进程管理器                   *")
        print("*                                  *")
        print("*     1.获取所有进程               *")
        print("*     2.获取指定进程的所有线程     *")
        print("*     3.显示和隐藏指定进程的窗口   *")
        print("*     4.强行结束指定进程           *")
        print("*                                  *")
        print("************************************")
        option = input("请选择功能：")

        if option == "1":
            getProcessList()
        elif option == "2":
            pid = input("请输入进程的pid：")
            getThreadOfProcess(pid)
        elif option == "3":
            pid = input("请输入进程的pid：")
            status = input("隐藏输入0，显示输入1：")
            changeWindowState(pid, status)
        elif option == "4":
            pid = input("请输入进程的pid：")
            killProcess(pid)
        else:
            exit()