python中的psutil模块怎么使用

其他教程   发布日期:2025年03月29日   浏览次数:97

这篇文章主要介绍“python中的psutil模块怎么使用”的相关知识,小编通过实际案例向大家展示操作过程,操作方法简单快捷,实用性强,希望这篇“python中的psutil模块怎么使用”文章能帮助大家解决问题。

获取CPU信息

我们先来获取CPU的信息:

  1. >>> import psutil
  2. >>> psutil.cpu_count() # CPU逻辑数量
  3. 4
  4. >>> psutil.cpu_count(logical=False) # CPU物理核心
  5. 2
  6. # 2说明是双核超线程, 4则是4核非超线程

统计CPU的用户/系统/空闲时间:

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

再实现类似top命令的CPU使用率,每秒刷新一次,累计10次

  1. >>> for x in range(10):
  2. ... print(psutil.cpu_percent(interval=1, percpu=True))
  3. ...
  4. [14.0, 4.0, 4.0, 4.0]
  5. [12.0, 3.0, 4.0, 3.0]
  6. [8.0, 4.0, 3.0, 4.0]
  7. [12.0, 3.0, 3.0, 3.0]
  8. [18.8, 5.1, 5.9, 5.0]
  9. [10.9, 5.0, 4.0, 3.0]
  10. [12.0, 5.0, 4.0, 5.0]
  11. [15.0, 5.0, 4.0, 4.0]
  12. [19.0, 5.0, 5.0, 4.0]
  13. [9.0, 3.0, 2.0, 3.0]

获取内存信息

使用psutil获取物理内存和交换内存信息,分别使用:

  1. >>> psutil.virtual_memory()
  2. svmem(total=8589934592, available=2866520064, percent=66.6, used=7201386496, free=216178688, active=3342192640, inactive=2650341376, wired=1208852480)
  3. >>> psutil.swap_memory()
  4. 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信息

  1. >>> psutil.disk_partitions() # 磁盘分区信息
  2. [sdiskpart(device='/dev/disk1', mountpoint='/', fstype='hfs', opts='rw,local,rootfs,dovolfs,journaled,multilabel')]
  3. >>> psutil.disk_usage('/') # 磁盘使用情况
  4. sdiskusage(total=998982549504, used=390880133120, free=607840272384, percent=39.1)
  5. >>> psutil.disk_io_counters() # 磁盘IO
  6. 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表示支持日志。

获取网络信息

psutil可以获取网络接口和网络连接信息

  1. >>> psutil.net_io_counters() # 获取网络读写字节/包的个数
  2. snetio(bytes_sent=3885744870, bytes_recv=10357676702, packets_sent=10613069, packets_recv=10423357, errin=0, errout=0, dropin=0, dropout=0)
  3. >>> psutil.net_if_addrs() # 获取网络接口信息
  4. {
  5. 'lo0': [snic(family=<AddressFamily.AF_INET: 2>, address='127.0.0.1', netmask='255.0.0.0'), ...],
  6. 'en1': [snic(family=<AddressFamily.AF_INET: 2>, address='10.0.1.80', netmask='255.255.255.0'), ...],
  7. 'en0': [...],
  8. 'en2': [...],
  9. 'bridge0': [...]
  10. }
  11. >>> psutil.net_if_stats() # 获取网络接口状态
  12. {
  13. 'lo0': snicstats(isup=True, duplex=<NicDuplex.NIC_DUPLEX_UNKNOWN: 0>, speed=0, mtu=16384),
  14. 'en0': snicstats(isup=True, duplex=<NicDuplex.NIC_DUPLEX_UNKNOWN: 0>, speed=0, mtu=1500),
  15. 'en1': snicstats(...),
  16. 'en2': snicstats(...),
  17. 'bridge0': snicstats(...)
  18. }

要获取当前网络连接信息,使用net_connections():

  1. >>> psutil.net_connections()
  2. Traceback (most recent call last):
  3. ...
  4. PermissionError: [Errno 1] Operation not permitted
  5. During handling of the above exception, another exception occurred:
  6. Traceback (most recent call last):
  7. ...
  8. psutil.AccessDenied: psutil.AccessDenied (pid=3847)

你可能会得到一个AccessDenied错误,原因是psutil获取信息也是要走系统接口,而获取网络连接信息需要root权限,这种情况下,可以退出Python交互环境,用sudo重新启动:

  1. $ sudo python3
  2. Password: ******
  3. Python 3.8 ... on darwin
  4. Type "help", ... for more information.
  5. >>> import psutil
  6. >>> psutil.net_connections()
  7. [
  8. 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),
  9. 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),
  10. sconn(fd=93, family=<AddressFamily.AF_INET6: 30>, type=1, laddr=addr(ip='::', port=8080), raddr=(), status='LISTEN', pid=3725),
  11. 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),
  12. sconn(fd=105, family=<AddressFamily.AF_INET6: 30>, type=1, ..., pid=3725),
  13. sconn(fd=106, family=<AddressFamily.AF_INET6: 30>, type=1, ..., pid=3725),
  14. sconn(fd=107, family=<AddressFamily.AF_INET6: 30>, type=1, ..., pid=3725),
  15. ...
  16. sconn(fd=27, family=<AddressFamily.AF_INET: 2>, type=2, ..., pid=1)
  17. ]

获取进程信息

通过psutil可以获取到所有进程的详细信息:

  1. >>> psutil.pids() # 所有进程ID
  2. [3865, 3864, 3863, 3856, 3855, 3853, 3776, ..., 45, 44, 1, 0]
  3. >>> p = psutil.Process(3776) # 获取指定进程ID=3776,其实就是当前Python交互环境
  4. >>> p.name() # 进程名称
  5. 'python3.6'
  6. >>> p.exe() # 进程exe路径
  7. '/Users/michael/anaconda3/bin/python3.6'
  8. >>> p.cwd() # 进程工作目录
  9. '/Users/michael'
  10. >>> p.cmdline() # 进程启动的命令行
  11. ['python3']
  12. >>> p.ppid() # 父进程ID
  13. 3765
  14. >>> p.parent() # 父进程
  15. <psutil.Process(pid=3765, name='bash') at 4503144040>
  16. >>> p.children() # 子进程列表
  17. []
  18. >>> p.status() # 进程状态
  19. 'running'
  20. >>> p.username() # 进程用户名
  21. 'michael'
  22. >>> p.create_time() # 进程创建时间
  23. 1511052731.120333
  24. >>> p.terminal() # 进程终端
  25. '/dev/ttys002'
  26. >>> p.cpu_times() # 进程使用的CPU时间
  27. pcputimes(user=0.081150144, system=0.053269812, children_user=0.0, children_system=0.0)
  28. >>> p.memory_info() # 进程使用的内存
  29. pmem(rss=8310784, vms=2481725440, pfaults=3207, pageins=18)
  30. >>> p.open_files() # 进程打开的文件
  31. []
  32. >>> p.connections() # 进程相关网络连接
  33. []
  34. >>> p.num_threads() # 进程的线程数量
  35. 1
  36. >>> p.threads() # 所有线程信息
  37. [pthread(id=1, user_time=0.090318, system_time=0.062736)]
  38. >>> p.environ() # 进程环境变量
  39. {'SHELL': '/bin/bash', 'PATH': '/usr/local/bin:/usr/bin:/bin:/usr/sbin:/sbin:...', 'PWD': '/Users/michael', 'LANG': 'zh_CN.UTF-8', ...}
  40. >>> p.terminate() # 结束进程
  41. Terminated: 15 <-- 自己把自己结束了

和获取网络连接类似,获取一个root用户的进程需要root权限,启动Python交互环境或者.py文件时,需要sudo权限。

psutil还提供了一个test()函数,可以模拟出ps命令的效果:

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

结束指定进程

方法一

  1. import psutil
  2. import os
  3. pid = os.getpid() # 获取当前进程 ID
  4. ppid = os.getppid() # 获取当前进程的父进程 ID
  5. def kill_process(process_name: str) -> None:
  6. '''
  7. @params process_name --> 进程名;
  8. '''
  9. for process in psutil.process_iter(['pid', 'name']):
  10. hwnd = process.as_dict(attrs=['pid', 'name'])
  11. if hwnd['name'] == process_name:
  12. father_pid = psutil.Process(hwnd['pid']).ppid()
  13. process = psutil.Process(father_pid)
  14. print(f'当前进程:【{hwnd}】')
  15. print(f'父进程pid:【{father_pid}】')
  16. process.terminate()
  17. # process.kill()
  18. break

方法二

  1. import psutil
  2. import os
  3. pid = os.getpid() # 获取当前进程 ID
  4. ppid = os.getppid() # 获取当前进程的父进程 ID
  5. def kill_process(process_name: str) -> None:
  6. '''
  7. @params process_name --> 进程名;
  8. '''
  9. for process in psutil.process_iter(['pid', 'name']):
  10. hwnd = process.as_dict(attrs=['pid', 'name'])
  11. if hwnd['name'] == process_name:
  12. father_pid = psutil.Process(hwnd['pid']).ppid()
  13. process = psutil.Process(father_pid)
  14. print(f'当前进程:【{hwnd}】')
  15. print(f'父进程pid:【{father_pid}】')
  16. process.terminate()
  17. # process.kill()
  18. break

除了使用

  1. psutil
  1. pywin32
,还有其他方法可以结束进程。

一种方法是使用 Python 内置的

  1. subprocess
模块来执行操作系统的命令,在命令行中使用
  1. taskkill
命令结束进程。示例代码如下:
  1. import subprocess
  2. def kill_process_by_name(process_name):
  3. """根据进程名结束进程"""
  4. command = "taskkill /f /im {}".format(process_name)
  5. subprocess.call(command, shell=True)
  6. if __name__ == '__main__':
  7. process_name = "myprocess.exe" # 需要结束的进程名
  8. kill_process_by_name(process_name)

在上面的示例代码中,我们使用

  1. subprocess
模块来执行
  1. taskkill
命令,其中
  1. /f
表示强制结束进程,
  1. /im
后面跟着进程名。

另一种方法是使用

  1. os.kill
函数结束进程。不过需要注意的是,
  1. os.kill
函数只适用于 Unix 和 Linux 等类 Unix 操作系统,不适用于 Windows 操作系统。示例代码如下:
  1. import os
  2. def kill_process_by_pid(pid):
  3. """根据进程ID结束进程"""
  4. os.kill(pid, 9)
  5. print("Process with ID {} has been killed.".format(pid))
  6. if __name__ == '__main__':
  7. pid = 1234 # 需要结束的进程ID
  8. kill_process_by_pid(pid)

在上面的示例代码中,我们使用

  1. os.kill
函数结束进程,其中第一个参数是进程ID,第二个参数是结束进程的信号。信号值为 9 表示强制结束进程。

以上就是python中的psutil模块怎么使用的详细内容,更多关于python中的psutil模块怎么使用的资料请关注九品源码其它相关文章!