Python使用psutil獲取系統信息
psutil模塊
用Python來編寫腳本簡化日常的運維工作是Python的一個重要用途。在Linux下，有許多系統命令可以讓我們時刻監控系統運行的狀態，如ps，top，free等等。要獲取這些系統信息，Python可以通過subprocess模塊調用並獲取結果。但這樣做顯得很麻煩，尤其是要寫很多解析代碼。
在Python中獲取系統信息的另一個好辦法是使用psutil這個第三方模塊。顧名思義，psutil = process and system utilities，它不僅可以通過一兩行代碼實現系統監控，還可以跨平臺使用，支持Linux／UNIX／OSX／Windows等，是系統管理員和運維小夥伴不可或缺的必備模塊。
1、安裝psutil
如果安裝瞭Anaconda，psutil就已經可用瞭。否則，需要在命令行下通過pip安裝：
$ pip install psutil
如果遇到Permission denied安裝失敗，請加上sudo重試。
2、獲取CPU信息
我們先來獲取CPU的信息：
>>> import psutil >>> psutil.cpu_count() # CPU邏輯數量 4 >>> psutil.cpu_count(logical=False) # CPU物理核心 2 # 2說明是雙核超線程, 4則是4核非超線程
統計CPU的用戶／系統／空閑時間：
>>> psutil.cpu_times() scputimes(user=10963.31, nice=0.0, system=5138.67, idle=356102.45)
再實現類似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]
3、獲取內存信息
使用psutil獲取物理內存和交換內存信息，分別使用：
>>> 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)
返回的是字節為單位的整數，可以看到，總內存大小是8589934592 = 8 GB，已用7201386496 = 6.7 GB，使用瞭66.6%。
而交換區大小是1073741824 = 1 GB。
獲取磁盤信息
可以通過psutil獲取磁盤分區、磁盤使用率和磁盤IO信息：
>>> psutil.disk_partitions() # 磁盤分區信息
[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)
可以看到，磁盤'/'的總容量是998982549504 = 930 GB，使用瞭39.1%。文件格式是HFS，opts中包含rw表示可讀寫，journaled表示支持日志。
4、獲取網絡信息
psutil可以獲取網絡接口和網絡連接信息：
>>> psutil.net_io_counters() # 獲取網絡讀寫字節／包的個數
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() # 獲取網絡接口信息
{
'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() # 獲取網絡接口狀態
{
'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(...)
}
要獲取當前網絡連接信息，使用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獲取信息也是要走系統接口，而獲取網絡連接信息需要root權限，這種情況下，可以退出Python交互環境，用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)
]
5、獲取進程信息
通過psutil可以獲取到所有進程的詳細信息：
>>> psutil.pids() # 所有進程ID
[3865, 3864, 3863, 3856, 3855, 3853, 3776, ..., 45, 44, 1, 0]
>>> p = psutil.Process(3776) # 獲取指定進程ID=3776，其實就是當前Python交互環境
>>> 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() # 進程狀態
'running'
>>> p.username() # 進程用戶名
'michael'
>>> p.create_time() # 進程創建時間
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() # 進程使用的內存
pmem(rss=8310784, vms=2481725440, pfaults=3207, pageins=18)
>>> p.open_files() # 進程打開的文件
[]
>>> p.connections() # 進程相關網絡連接
[]
>>> p.num_threads() # 進程的線程數量
1
>>> p.threads() # 所有線程信息
[pthread(id=1, user_time=0.090318, system_time=0.062736)]
>>> p.environ() # 進程環境變量
{'SHELL': '/bin/bash', 'PATH': '/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin:...', 'PWD': '/Users/michael', 'LANG': 'zh_CN.UTF-8', ...}
>>> p.terminate() # 結束進程
Terminated: 15 <-- 自己把自己結束瞭
和獲取網絡連接類似，獲取一個root用戶的進程需要root權限，啟動Python交互環境或者.py文件時，需要sudo權限。
psutil還提供瞭一個test()函數，可以模擬出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 ...
小結
psutil使得Python程序獲取系統信息變得易如反掌。
psutil還可以獲取用戶信息、Windows服務等很多有用的系統信息，具體請參考psutil的官網：https://github.com/giampaolo/psutil
推薦閱讀：
- Python利用psutil實現獲取硬件,網絡和進程信息
- Python使用psutil庫對系統數據進行采集監控的方法
- Python中的socket網絡模塊介紹
- python 基於UDP協議套接字通信的實現
- python 實現syslog 服務器的詳細過程