feat✨: add check

2026-10-03 19:00:00 +08:00 · 2024-03-19 00:34:04 +08:00
parent 606da31851
commit b74998db87
5 changed files with 160 additions and 3 deletions
@@ -0,0 +1,83 @@
+import asyncio
+import time
+from datetime import datetime
+
+import psutil
+
+from zhenxun.configs.path_config import IMAGE_PATH
+from zhenxun.services.log import logger
+from zhenxun.utils.http_utils import AsyncHttpx
+from zhenxun.utils.image_utils import BuildImage
+
+
+class Check:
+    def __init__(self):
+        self.cpu = None
+        self.memory = None
+        self.disk = None
+        self.user = None
+        self.baidu = 200
+        self.google = 200
+
+    async def check_all(self):
+        await self.check_network()
+        await asyncio.sleep(0.1)
+        self.check_system()
+        self.check_user()
+
+    def check_system(self):
+        self.cpu = psutil.cpu_percent()
+        self.memory = psutil.virtual_memory().percent
+        self.disk = psutil.disk_usage("/").percent
+
+    async def check_network(self):
+        try:
+            await AsyncHttpx.get("https://www.baidu.com/", timeout=5)
+        except Exception as e:
+            logger.warning(f"访问BaiDu失败... {type(e)}: {e}")
+            self.baidu = 404
+        try:
+            await AsyncHttpx.get("https://www.google.com/", timeout=5)
+        except Exception as e:
+            logger.warning(f"访问Google失败... {type(e)}: {e}")
+            self.google = 404
+
+    def check_user(self):
+        result = ""
+        for user in psutil.users():
+            result += f'[{user.name}] {time.strftime("%Y-%m-%d %H:%M:%S", time.localtime(user.started))}\n'
+        self.user = result[:-1]
+
+    async def show(self) -> BuildImage:
+        await self.check_all()
+        font = BuildImage.load_font(font_size=24)
+        result = (
+            f'[Time] {str(datetime.now()).split(".")[0]}\n'
+            f"-----System-----\n"
+            f"[CPU] {self.cpu}%\n"
+            f"[Memory] {self.memory}%\n"
+            f"[Disk] {self.disk}%\n"
+            f"-----Network-----\n"
+            f"[BaiDu] {self.baidu}\n"
+            f"[Google] {self.google}\n"
+        )
+        if self.user:
+            result += "-----User-----\n" + self.user
+        width = 0
+        height = 0
+        for x in result.split("\n"):
+            w, h = BuildImage.get_text_size(x, font)
+            if w > width:
+                width = w
+            height += 30
+        A = BuildImage(width + 50, height + 10, font_size=24)
+        await A.transparent(1)
+        await A.text((10, 10), result)
+        max_width = max(width, height)
+        bk = BuildImage(
+            max_width + 100,
+            max_width + 100,
+            background=IMAGE_PATH / "background" / "check" / "0.jpg",
+        )
+        await bk.paste(A, center_type="center")
+        return bk