使用 Python 自动执行磁盘资源使用情况监控和服务器运行状况更新

使用 Python 自动执行磁盘资源使用情况监控和服务器运行状况更新

监控服务器磁盘使用情况对于保持最佳性能和防止停机至关重要。在这篇博文中，我们将探讨如何使用 Python 脚本自动监控磁盘资源并通过 API 更新服务器运行状况。我们还将讨论如何设置 cron 作业来定期运行脚本。

先决条件

• Python编程基础知识
• 熟悉Linux命令行操作
• 访问可以运行 Python 脚本并设置 cron 作业的服务器
• 用于更新服务器运行状况的 API 端点（替换为您的实际 API URL 和令牌）

Python 脚本解释

下面是执行磁盘资源监控并通过 API 更新服务器运行状况的 Python 脚本。

本篇博文未涵盖 Health API 创建，如果您也需要，请发表评论，我也会发布该 api 创建步骤。

import subprocess
import requests
import argparse


class Resource:
    file_system = ''
    disk_size = 0.0
    used = 0.0
    avail = 0.0
    use_percent = 0.0
    mounted_on = 0.0
    disk_free_threshold = 1
    mount_partition = "/"


class ResourcesMonitor(Resource):
    def __init__(self):
        self.__file_system = Resource.file_system
        self.__disk_size = Resource.disk_size
        self.__used = Resource.used
        self.__avail = Resource.avail
        self.__use_percent = Resource.use_percent
        self.__mounted_on = Resource.mounted_on
        self.__disk_free_threshold = Resource.disk_free_threshold
        self.__mount_partition = Resource.mount_partition

    def show_resource_usage(self):
        """
        Print the resource usage of disk.
        """
        print("file_system", "disk_size", "used", "avail", "use_percent", "mounted_on")
        print(self.__file_system, self.__disk_size, self.__used, self.__avail, self.__use_percent, self.__mounted_on)

    def check_resource_usage(self):
        """
        Check the disk usage by running the Unix 'df -h' command.
        """
        response_df = subprocess.Popen(["df", "-h"], stdout=subprocess.PIPE)
        for line in response_df.stdout:
            split_line = line.decode().split()
            if split_line[5] == self.__mount_partition:
                if int(split_line[4][:-1]) > self.__disk_free_threshold:
                    self.__file_system, self.__disk_size, self.__used = split_line[0], split_line[1], split_line[2]
                    self.__avail, self.__use_percent, self.__mounted_on = split_line[3], split_line[4], split_line[5]
                    self.show_resource_usage()
                    self.update_resource_usage_api(self)

    def update_resource_usage_api(self, resource):
        """
        Call the update API using all resource details.
        """
        update_resource_url = url.format(
            resource.__file_system,
            resource.__disk_size,
            resource.__used,
            resource.__avail,
            resource.__use_percent,
            resource_id
        )

        print(update_resource_url)
        payload = {}
        files = {}
        headers = {
            'token': 'Bearer APITOKEN'
        }
        try:
            response = requests.request("GET", update_resource_url, headers=headers, data=payload, files=files)
            if response.ok:
                print(response.json())
        except Exception as ex:
            print("Error while calling update API")
            print(ex)


if __name__ == '__main__':
    url = "http://yourapi.com/update_server_health_by_server_id?path={}&size={}" \
          "&used={}&avail={}&use_percent={}&id={}"
    parser = argparse.ArgumentParser(description='Disk Resource Monitor')
    parser.add_argument('-id', metavar='id', help='ID record of server', default=7, type=int)
    args = parser.parse_args()
    resource_id = args.id
    print(resource_id)
    resource_monitor = ResourcesMonitor()
    resource_monitor.check_resource_usage()

Resource 和 ResourcesMonitor 类

Resource 类定义了与磁盘使用相关的属性，例如文件系统、磁盘大小、已用空间等。 ResourcesMonitor类继承自Resource并初始化这些属性。

检查磁盘使用情况

check_resource_usage 方法执行 Unix df -h 命令来获取磁盘使用统计信息。它解析输出以查找指定安装分区的磁盘使用情况（默认为/）。如果磁盘使用率超过阈值，则会更新资源详细信息并调用 API 更新方法。

通过 API 更新服务器运行状况

update_resource_usage_api 方法使用资源详细信息构造 API 请求 URL，并发送 GET 请求来更新服务器运行状况。确保将 http://yourapi.com/update_server_health_by_server_id 替换为您的实际 API 端点并提供正确的 API 令牌。

使用脚本

将脚本保存为resource_monitor.py并使用Python 3运行它。

命令行参数

• -id：要更新健康数据的服务器ID（默认为7）。 这将有助于在多个服务器中运行相同的脚本，只需更改 ID。

用法和输出示例

$ python3 resource_monitor.py -id=7

Output:
file_system disk_size used avail use_percent mounted_on
/dev/root 39G 31G 8.1G 80% /

API GET Request:
http://yourapi.com/update_server_health_by_server_id?path=/dev/root&size=39G&used=31G&avail=8.1G&use_percent=80%&id=7

Response
{'success': 'Servers_health data Updated.', 'data': {'id': 7, 'server_id': 1, 'server_name': 'web-server', 'server_ip': '11.11.11.11', 'size': '39G', 'path': '/dev/root', 'used': '31G', 'avail': '8.1G', 'use_percent': '80%', 'created_at': '2021-08-28T13:45:28.000000Z', 'updated_at': '2024-10-27T08:02:43.000000Z'}}

使用 Cron 实现自动化

要每 30 分钟自动执行一次脚本，请添加一个 cron 作业，如下所示：

*/30 * * * * python3 /home/ubuntu/resource_monitor.py -id=7 &

您可以通过运行 crontab -e 并添加以上行来编辑 cron 作业。这将确保脚本每 30 分钟运行一次，从而使您的服务器运行状况数据保持最新。

结论

通过自动化磁盘资源监控和服务器运行状况更新，您可以主动管理服务器的性能并避免由于磁盘空间短缺而导致的潜在问题。此 Python 脚本可作为起点，并可进行定制以满足您的特定需求。

以上是使用 Python 自动执行磁盘资源使用情况监控和服务器运行状况更新的详细内容。更多信息请关注PHP中文网其他相关文章！