python中的psutil模塊詳解(cpu、內(nèi)存、磁盤情況、結(jié)束指定進程)

更新時間：2023年04月21日 10:17:38 作者：像風一樣的男人@

這篇文章主要介紹了python中的psutil(cpu、內(nèi)存、磁盤情況、結(jié)束指定進程),本文給大家介紹的非常詳細，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下

獲取CPU信息

我們先來獲取CPU的信息：

>>> import psutil
>>> psutil.cpu_count() # CPU邏輯數(shù)量
4
>>> psutil.cpu_count(logical=False) # CPU物理核心
2
# 2說明是雙核超線程, 4則是4核非超線程

統(tǒng)計CPU的用戶／系統(tǒng)／空閑時間：

>>> psutil.cpu_times()
scputimes(user=10963.31, nice=0.0, system=5138.67, idle=356102.45)

再實現(xiàn)類似top命令的CPU使用率，每秒刷新一次，累計10次

>>> for x in range(10):
...     print(psutil.cpu_percent(interval=1, percpu=True))
... 
[14.0, 4.0, 4.0, 4.0]
[12.0, 3.0, 4.0, 3.0]
[8.0, 4.0, 3.0, 4.0]
[12.0, 3.0, 3.0, 3.0]
[18.8, 5.1, 5.9, 5.0]
[10.9, 5.0, 4.0, 3.0]
[12.0, 5.0, 4.0, 5.0]
[15.0, 5.0, 4.0, 4.0]
[19.0, 5.0, 5.0, 4.0]
[9.0, 3.0, 2.0, 3.0]

獲取內(nèi)存信息

使用psutil獲取物理內(nèi)存和交換內(nèi)存信息，分別使用：

>>> psutil.virtual_memory()
svmem(total=8589934592, available=2866520064, percent=66.6, used=7201386496, free=216178688, active=3342192640, inactive=2650341376, wired=1208852480)
>>> psutil.swap_memory()
sswap(total=1073741824, used=150732800, free=923009024, percent=14.0, sin=10705981440, sout=40353792)

返回的是字節(jié)為單位的整數(shù)，可以看到，總內(nèi)存大小是8589934592 = 8 GB，已用7201386496 = 6.7 GB，使用了66.6%。

而交換區(qū)大小是1073741824 = 1 GB。

獲取磁盤信息

可以通過psutil獲取磁盤分區(qū)、磁盤使用率和磁盤IO信息

>>> psutil.disk_partitions() # 磁盤分區(qū)信息
[sdiskpart(device='/dev/disk1', mountpoint='/', fstype='hfs', opts='rw,local,rootfs,dovolfs,journaled,multilabel')]
>>> psutil.disk_usage('/') # 磁盤使用情況
sdiskusage(total=998982549504, used=390880133120, free=607840272384, percent=39.1)
>>> psutil.disk_io_counters() # 磁盤IO
sdiskio(read_count=988513, write_count=274457, read_bytes=14856830464, write_bytes=17509420032, read_time=2228966, write_time=1618405)

可以看到，磁盤’/'的總?cè)萘渴?98982549504 = 930 GB，使用了39.1%。文件格式是HFS，opts中包含rw表示可讀寫，journaled表示支持日志。

獲取網(wǎng)絡(luò)信息

psutil可以獲取網(wǎng)絡(luò)接口和網(wǎng)絡(luò)連接信息

>>> psutil.net_io_counters() # 獲取網(wǎng)絡(luò)讀寫字節(jié)／包的個數(shù)
snetio(bytes_sent=3885744870, bytes_recv=10357676702, packets_sent=10613069, packets_recv=10423357, errin=0, errout=0, dropin=0, dropout=0)
>>> psutil.net_if_addrs() # 獲取網(wǎng)絡(luò)接口信息
{
  'lo0': [snic(family=<AddressFamily.AF_INET: 2>, address='127.0.0.1', netmask='255.0.0.0'), ...],
  'en1': [snic(family=<AddressFamily.AF_INET: 2>, address='10.0.1.80', netmask='255.255.255.0'), ...],
  'en0': [...],
  'en2': [...],
  'bridge0': [...]
}
>>> psutil.net_if_stats() # 獲取網(wǎng)絡(luò)接口狀態(tài)
{
  'lo0': snicstats(isup=True, duplex=<NicDuplex.NIC_DUPLEX_UNKNOWN: 0>, speed=0, mtu=16384),
  'en0': snicstats(isup=True, duplex=<NicDuplex.NIC_DUPLEX_UNKNOWN: 0>, speed=0, mtu=1500),
  'en1': snicstats(...),
  'en2': snicstats(...),
  'bridge0': snicstats(...)
}

要獲取當前網(wǎng)絡(luò)連接信息，使用net_connections()：

>>> psutil.net_connections()
Traceback (most recent call last):
  ...
PermissionError: [Errno 1] Operation not permitted

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  ...
psutil.AccessDenied: psutil.AccessDenied (pid=3847)

你可能會得到一個AccessDenied錯誤，原因是psutil獲取信息也是要走系統(tǒng)接口，而獲取網(wǎng)絡(luò)連接信息需要root權(quán)限，這種情況下，可以退出Python交互環(huán)境，用sudo重新啟動：

$ sudo python3
Password: ******
Python 3.8 ... on darwin
Type "help", ... for more information.
>>> import psutil
>>> psutil.net_connections()
[
    sconn(fd=83, family=<AddressFamily.AF_INET6: 30>, type=1, laddr=addr(ip='::127.0.0.1', port=62911), raddr=addr(ip='::127.0.0.1', port=3306), status='ESTABLISHED', pid=3725),
    sconn(fd=84, family=<AddressFamily.AF_INET6: 30>, type=1, laddr=addr(ip='::127.0.0.1', port=62905), raddr=addr(ip='::127.0.0.1', port=3306), status='ESTABLISHED', pid=3725),
    sconn(fd=93, family=<AddressFamily.AF_INET6: 30>, type=1, laddr=addr(ip='::', port=8080), raddr=(), status='LISTEN', pid=3725),
    sconn(fd=103, family=<AddressFamily.AF_INET6: 30>, type=1, laddr=addr(ip='::127.0.0.1', port=62918), raddr=addr(ip='::127.0.0.1', port=3306), status='ESTABLISHED', pid=3725),
    sconn(fd=105, family=<AddressFamily.AF_INET6: 30>, type=1, ..., pid=3725),
    sconn(fd=106, family=<AddressFamily.AF_INET6: 30>, type=1, ..., pid=3725),
    sconn(fd=107, family=<AddressFamily.AF_INET6: 30>, type=1, ..., pid=3725),
    ...
    sconn(fd=27, family=<AddressFamily.AF_INET: 2>, type=2, ..., pid=1)
]

獲取進程信息

通過psutil可以獲取到所有進程的詳細信息：

>>> psutil.pids() # 所有進程ID
[3865, 3864, 3863, 3856, 3855, 3853, 3776, ..., 45, 44, 1, 0]
>>> p = psutil.Process(3776) # 獲取指定進程ID=3776，其實就是當前Python交互環(huán)境
>>> p.name() # 進程名稱
'python3.6'
>>> p.exe() # 進程exe路徑
'/Users/michael/anaconda3/bin/python3.6'
>>> p.cwd() # 進程工作目錄
'/Users/michael'
>>> p.cmdline() # 進程啟動的命令行
['python3']
>>> p.ppid() # 父進程ID
3765
>>> p.parent() # 父進程
<psutil.Process(pid=3765, name='bash') at 4503144040>
>>> p.children() # 子進程列表
[]
>>> p.status() # 進程狀態(tài)
'running'
>>> p.username() # 進程用戶名
'michael'
>>> p.create_time() # 進程創(chuàng)建時間
1511052731.120333
>>> p.terminal() # 進程終端
'/dev/ttys002'
>>> p.cpu_times() # 進程使用的CPU時間
pcputimes(user=0.081150144, system=0.053269812, children_user=0.0, children_system=0.0)
>>> p.memory_info() # 進程使用的內(nèi)存
pmem(rss=8310784, vms=2481725440, pfaults=3207, pageins=18)
>>> p.open_files() # 進程打開的文件
[]
>>> p.connections() # 進程相關(guān)網(wǎng)絡(luò)連接
[]
>>> p.num_threads() # 進程的線程數(shù)量
1
>>> p.threads() # 所有線程信息
[pthread(id=1, user_time=0.090318, system_time=0.062736)]
>>> p.environ() # 進程環(huán)境變量
{'SHELL': '/bin/bash', 'PATH': '/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin:...', 'PWD': '/Users/michael', 'LANG': 'zh_CN.UTF-8', ...}
>>> p.terminate() # 結(jié)束進程
Terminated: 15 <-- 自己把自己結(jié)束了

和獲取網(wǎng)絡(luò)連接類似，獲取一個root用戶的進程需要root權(quán)限，啟動Python交互環(huán)境或者.py文件時，需要sudo權(quán)限。

psutil還提供了一個test()函數(shù)，可以模擬出ps命令的效果：

$ sudo python3
Password: ******
Python 3.6.3 ... on darwin
Type "help", ... for more information.
>>> import psutil
>>> psutil.test()
USER         PID %MEM     VSZ     RSS TTY           START    TIME  COMMAND
root           0 24.0 74270628 2016380 ?             Nov18   40:51  kernel_task
root           1  0.1 2494140    9484 ?             Nov18   01:39  launchd
root          44  0.4 2519872   36404 ?             Nov18   02:02  UserEventAgent
root          45    ? 2474032    1516 ?             Nov18   00:14  syslogd
root          47  0.1 2504768    8912 ?             Nov18   00:03  kextd
root          48  0.1 2505544    4720 ?             Nov18   00:19  fseventsd
_appleeven    52  0.1 2499748    5024 ?             Nov18   00:00  appleeventsd
root          53  0.1 2500592    6132 ?             Nov18   00:02  configd
...

結(jié)束指定進程

方法一

import psutil
import os

pid = os.getpid() # 獲取當前進程 ID
ppid = os.getppid() # 獲取當前進程的父進程 ID


def kill_process(process_name: str) -> None:
    '''
    @params  process_name --> 進程名;
    '''
    for process in psutil.process_iter(['pid', 'name']):
        hwnd = process.as_dict(attrs=['pid', 'name'])
        if hwnd['name'] == process_name:
            father_pid = psutil.Process(hwnd['pid']).ppid()
            process = psutil.Process(father_pid)
            print(f'當前進程:【{hwnd}】')
            print(f'父進程pid:【{father_pid}】')
            process.terminate()
            # process.kill()
            break

方法二

import psutil
import os

pid = os.getpid() # 獲取當前進程 ID
ppid = os.getppid() # 獲取當前進程的父進程 ID


def kill_process(process_name: str) -> None:
    '''
    @params  process_name --> 進程名;
    '''
    for process in psutil.process_iter(['pid', 'name']):
        hwnd = process.as_dict(attrs=['pid', 'name'])
        if hwnd['name'] == process_name:
            father_pid = psutil.Process(hwnd['pid']).ppid()
            process = psutil.Process(father_pid)
            print(f'當前進程:【{hwnd}】')
            print(f'父進程pid:【{father_pid}】')
            process.terminate()
            # process.kill()
            break

除了使用 psutil 和 pywin32，還有其他方法可以結(jié)束進程。

一種方法是使用 Python 內(nèi)置的 subprocess 模塊來執(zhí)行操作系統(tǒng)的命令，在命令行中使用 taskkill 命令結(jié)束進程。示例代碼如下：

import subprocess

def kill_process_by_name(process_name):
	"""根據(jù)進程名結(jié)束進程"""
	command = "taskkill /f /im {}".format(process_name)
	subprocess.call(command, shell=True)

if __name__ == '__main__':
	process_name = "myprocess.exe" # 需要結(jié)束的進程名
	kill_process_by_name(process_name)

在上面的示例代碼中，我們使用 subprocess 模塊來執(zhí)行 taskkill 命令，其中 /f 表示強制結(jié)束進程，/im 后面跟著進程名。

另一種方法是使用 os.kill 函數(shù)結(jié)束進程。不過需要注意的是，os.kill 函數(shù)只適用于 Unix 和 Linux 等類 Unix 操作系統(tǒng)，不適用于 Windows 操作系統(tǒng)。示例代碼如下：

import os

def kill_process_by_pid(pid):
	"""根據(jù)進程ID結(jié)束進程"""
	os.kill(pid, 9)
	print("Process with ID {} has been killed.".format(pid))

if __name__ == '__main__':
	pid = 1234 # 需要結(jié)束的進程ID
	kill_process_by_pid(pid)

在上面的示例代碼中，我們使用 os.kill 函數(shù)結(jié)束進程，其中第一個參數(shù)是進程ID，第二個參數(shù)是結(jié)束進程的信號。信號值為 9 表示強制結(jié)束進程。

到此這篇關(guān)于python中的psutil(cpu、內(nèi)存、磁盤情況、結(jié)束指定進程)的文章就介紹到這了,更多相關(guān)python psutil內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: