如何使用Python腳本在Linux中實現系統監控與警報-Python教學-PHP中文網

首頁

後端開發

Python教學

如何使用Python腳本在Linux中實現系統監控與警報

王林

Oct 05, 2023 am 09:28 AM

python腳本監控

如何使用Python腳本在Linux中實現系統監控與警報

在現代的雲端運算環境中，確保高效穩定的系統運作是至關重要的。為了實現這一目標，對系統進行監控並及時報警成為不可或缺的工作。本文將介紹如何使用Python腳本在Linux系統中實現系統監控與警報的方法，並提供具體的程式碼範例。

安裝Python和相關函式庫

首先，確保系統中已經安裝了Python和相關的函式庫。大多數Linux發行版都會預設安裝Python，通常也會預先安裝一些常用的函式庫。如果你的系統沒有預先安裝Python，可以透過套件管理器進行安裝。例如，在Ubuntu系統上，可以使用以下命令安裝Python：

sudo apt-get install python3

在安裝了Python之後，透過以下命令來安裝相關的庫：

pip install psutil
pip install smtplib
pip install email.mime.multipart
pip install email.mime.text

psutil庫用於獲取系統信息，smtplib庫用於發送郵件，email.mime.multipart和email.mime.text庫用於建立郵件。

取得系統資訊

為了監控系統的使用情況，我們需要先取得一些系統資訊。使用psutil函式庫可以輕鬆取得CPU和記憶體的使用率、磁碟和網路的IO等資訊。以下是範例程式碼，展示如何取得CPU和記憶體的使用率：

import psutil

def get_cpu_usage():
    return psutil.cpu_percent(interval=1)

def get_memory_usage():
    mem = psutil.virtual_memory()
    return mem.percent

透過呼叫get_cpu_usage()和get_memory_usage()函數，我們可以取得目前的CPU使用率和記憶體使用率。

設定警報規則

在確定了要監控的系統資訊後，我們需要設定一些警報規則，以便在系統異常時觸發警報。例如，我們可以設定當CPU的使用率超過80%或記憶體的使用率超過90%時觸發警報。以下是一個範例程式碼，展示如何設定警報規則：

CPU_THRESHOLD = 80
MEMORY_THRESHOLD = 90

def check_cpu_usage():
    cpu_usage = get_cpu_usage()
    if cpu_usage > CPU_THRESHOLD:
        send_alert_email("CPU 使用率过高：%d%%" % cpu_usage)

def check_memory_usage():
    memory_usage = get_memory_usage()
    if memory_usage > MEMORY_THRESHOLD:
        send_alert_email("内存 使用率过高：%d%%" % memory_usage)

在上述程式碼中，我們定義了兩個常數CPU_THRESHOLD和MEMORY_THRESHOLD分別表示CPU和記憶體的閾值。然後，我們編寫了兩個函數check_cpu_usage()和check_memory_usage()來檢查CPU和記憶體的使用率，並在超過閾值時觸發警報。

發送警報郵件

一旦系統的使用情況超過了設定的閾值，我們需要發送警報郵件給相關人員。透過使用smtplib和email庫，我們可以輕鬆實現郵件的傳送功能。以下是一個範例程式碼，展示如何傳送警報郵件：

import smtplib
from email.mime.multipart import MIMEMultipart
from email.mime.text import MIMEText

def send_alert_email(message):
    sender = "sender@example.com"
    receiver = "receiver@example.com"
    password = "password"

    subject = "系统报警"
    body = message

    msg = MIMEMultipart()
    msg['From'] = sender
    msg['To'] = receiver
    msg['Subject'] = subject

    msg.attach(MIMEText(body, 'plain'))

    server = smtplib.SMTP('smtp.example.com', 587)
    server.starttls()
    server.login(sender, password)
    server.sendmail(sender, receiver, msg.as_string())
    server.quit()

在上述程式碼中，我們首先設定了寄件者、收件者和SMTP伺服器的相關資訊。然後，我們建立了一個MIMEMultipart對象，並設定了寄件者、收件者和主題。最後，我們透過呼叫server物件的sendmail()方法來傳送郵件。