墨水熊

  博客园  :: 首页  :: 新随笔  :: 联系 :: 订阅 订阅  :: 管理

监控一个服务系exe在他关闭后打印,原想可以使用win32process.EnumProcesses()来获取所有程序pid,再根据pid获取的进程名进行对比,来判断程序是否关闭,但遇到个问题,我无法根据pid获取到进程名,除非使用psutil包,可这样又让自己的程序过于臃肿,我又想到去查找c++的例子希望可以得到一些帮助,然而c++的api函数在python中并不存在,我之前遇到这种问题基本都使用wmi去解决的。

1 import wmi
2     c = wmi.WMI()
3     for process in c.Win32_Process():
4         print process.ProcessId, process.Name

但wmi的速度可是出奇的慢,实在不适合,

最后我在外面找到个例子,可以很好的解决该问题

from win32com.client import GetObject
    _wmi = GetObject('winmgmts:')

    # collect all the running processes
    processes = _wmi.ExecQuery('Select * from win32_process')
    for p in processes:

            print((p.ProcessId, p.ExecutablePath, p.CommandLine, p.Name))

由此全部代码为

def wmi_sql_all_name(pname):#全部程序列表
    from win32com.client import GetObject
    _wmi = GetObject('winmgmts:')
    processes = _wmi.ExecQuery('Select * from win32_process')
    for p in processes:
        if p.name==pname:
            return p.ProcessId
def process_restat(pname):
    import win32security, win32event
#-----------------------------------------获取令牌 hToken
= win32security.OpenProcessToken(win32api.GetCurrentProcess(), win32con.TOKEN_ADJUST_PRIVILEGES | win32con.TOKEN_QUERY) # 打开自己的程序令牌 luid = win32security.LookupPrivilegeValue(None, 'SeDebugPrivilege') # 查看自己的权限,并写入权限关键字 PyTOKEN = ((luid, win32con.SE_PRIVILEGE_ENABLED),) # 将这些做成序列用做参数 pyt1 = win32security.AdjustTokenPrivileges(hToken, 0, PyTOKEN) # 权限写入
#----------------------------------------
processid=wmi_sql_all_name(pname) if processid!= None: hprocess = win32api.OpenProcess(win32con.PROCESS_ALL_ACCESS, False, processid) win32event.WaitForSingleObject(hprocess, -1) win32api.CloseHandle(hprocess) print U'已关闭' else: print u'未启动'

首先需要获取令牌,否则获取很多服务或者系统进程的信息都会报权限不足

使用OpenProcess得到程序句柄,

使用WaitForSingleObject得到程序的已通知状态也就是运行状态,第二参数为等待时间 -1 为永久等待

注意该函数会死锁线程,但不会占用cpu,很适合丢在子线程里,比无限循环跑程序列表要有效率得多

-------------------

突然想到有sql语句了 我为啥还要循环呢?

def wmi_sql_all_name(pname):
    from win32com.client import GetObject

    _wmi = GetObject('winmgmts:')
    processes = _wmi.ExecQuery("Select * from win32_process where name= '%s'"%(pname))
    try:
        return processes[0].ProcessId
    except:
        print u"没有发现程序"
        return None

将wmi_sql_all_name方法改为以以上

posted on 2020-01-12 16:31  墨水熊  阅读(2574)  评论(0编辑  收藏  举报