快捷導(dǎo)航

基于python編寫監(jiān)控系統(tǒng)各項資源的腳本

更新時間：2023年11月03日 14:11:19 作者：shigen01

這篇文章主要為大家詳細(xì)介紹了如何編寫一個python腳本,實現(xiàn)監(jiān)控網(wǎng)絡(luò)的流量、CPU使用率、內(nèi)存使用率和磁盤使用情況,感興趣的小伙伴可以了解下

今天的文章涉及到docker的操作和一個python腳本，實現(xiàn)監(jiān)控網(wǎng)絡(luò)的流量、CPU使用率、內(nèi)存使用率和磁盤使用情況。來和shigen一起先看看效果吧：

這是在控制臺中出現(xiàn)的數(shù)據(jù)，可以很簡單的看到我們想要的監(jiān)控指標(biāo)。如果實現(xiàn)定時任務(wù)和數(shù)據(jù)的存儲、數(shù)據(jù)的展示，就是很好看的數(shù)據(jù)監(jiān)控圖表了。

我的代碼主要是運行在一個centos的docker容器中。

是的，你沒有看錯，我還暴露了容器的22端口，供我在宿主機(jī)進(jìn)行連接。

但是這不是重點，我拉取的centos其實是一個裸機(jī)，跟我們最開始買的服務(wù)器一樣的。運行代碼，肯定是需要安裝python解釋器的，這里shigen找到了珍藏許久的shell腳本，直接復(fù)制粘貼即可配置好python的環(huán)境。

這里我選取的python版本是：3.9.2。

wget https://registry.npmmirror.com/-/binary/python/3.9.2/Python-3.9.2.tgz
tar -zxvf Python*
yum -y install zlib-devel bzip2-devel openssl-devel ncurses-devel sqlite-devel readline-devel tk-devel gcc make libffi-devel
./configure
make && make install

驗證python是否安裝成功，只需要執(zhí)行一下命令即可：

which python3 && python3 --version

那其實還有很重要的一步：配置pip的鏡像，不然網(wǎng)絡(luò)會慢到你懷疑人生。

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

現(xiàn)在，我們的前置的準(zhǔn)備工作全部的完成了。我們直接在vscode中編輯好代碼：

# -*- encoding: utf-8 -*-
__date__ = '2023/11/02 19:01:26'

import psutil
from rich.table import Table
from rich.console import Console

# 創(chuàng)建新表格
table = Table(show_header=True, header_style="bold magenta", expand=True)
table.add_column("指標(biāo)", style="dim", width=12)
table.add_column("數(shù)值", justify="right")

# 獲取系統(tǒng)信息
cpu_percent = psutil.cpu_percent(interval=1)
cpu_freq = psutil.cpu_freq()
cpu_cores = psutil.cpu_count(logical=False)
memory = psutil.virtual_memory()
network_stats = psutil.net_io_counters()
disk_stats = psutil.disk_partitions(all=True)

# 添加 CPU 信息到表格
table.add_row("CPU 使用率", f"{cpu_percent}%")
if cpu_freq is not None:
    table.add_row("CPU 頻率", f"{cpu_freq.current:.2f} MHz")
else:
    table.add_row("CPU 頻率", "N/A")
table.add_row("CPU 核心數(shù)", f"{cpu_cores} 核")

# 添加內(nèi)存信息到表格
table.add_row("總內(nèi)存", f"{memory.total / (1024**3):.2f} GB")
table.add_row("已用內(nèi)存", f"{memory.used / (1024**3):.2f} GB")
table.add_row("內(nèi)存使用百分比", f"{memory.percent}%")

# 添加網(wǎng)絡(luò)信息到表格
table.add_row("發(fā)送字節(jié)數(shù)", f"{network_stats.bytes_sent / (1024**2):.2f} MB")
table.add_row("接收字節(jié)數(shù)", f"{network_stats.bytes_recv / (1024**2):.2f} MB")

# 添加磁盤信息到表格
for partition in disk_stats:
    disk_usage = psutil.disk_usage(partition.mountpoint)
    table.add_row(f"{partition.device} ({partition.mountpoint})", f"{disk_usage.percent}%")

# 創(chuàng)建 console 對象并輸出表格
console = Console()
console.print(table)

這里主要用到了兩個庫：psutil和rich，借助psutil我們可以輕易的獲得系統(tǒng)的各種指標(biāo)，那rich主要是在控制臺實現(xiàn)炫酷的效果。這里是時間關(guān)系，shigen寫的比較簡單。

那文件在我的主機(jī)上，centos的服務(wù)器在我的docker里邊，文件到我的docker容器就需要docker cp派上用場了。在主機(jī)執(zhí)行：