mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
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
|