blob: fe10dea686586cf2ad132c213922cc06887494db (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
'''
File: status.py
Created Date: 2021-02-19 21:52:56
Author: Kyomotoi
Email: [email protected]
License: GPLv3
Project: https://github.com/Kyomotoi/ATRI
--------
Last Modified: Sunday, 7th March 2021 3:11:30 pm
Modified By: Kyomotoi ([email protected])
--------
Copyright (c) 2021 Kyomotoi
'''
import psutil
from nonebot.plugin import on_command
from nonebot.adapters.cqhttp import Bot, MessageEvent
from ATRI.rule import is_in_banlist
from ATRI.exceptions import GetStatusError
ping = on_command("/ping", rule=is_in_banlist())
@ping.handle()
async def _ping(bot: Bot, event: MessageEvent) -> None:
await ping.finish("I'm fine.")
status = on_command("/status", rule=is_in_banlist())
@status.handle()
async def _status(bot: Bot, event: MessageEvent) -> None:
try:
cpu = psutil.cpu_percent(interval=1)
mem = psutil.virtual_memory().percent
disk = psutil.disk_usage("/").percent
inteSENT = psutil.net_io_counters().bytes_sent / 1000000 # type: ignore
inteRECV = psutil.net_io_counters().bytes_sent / 1000000 # type: ignore
except GetStatusError:
raise GetStatusError("Failed to get status.")
msg = "アトリは、高性能ですから！"
if cpu > 80: # type: ignore
msg = "咱感觉有些头晕..."
if mem > 80:
msg = "咱感觉有点头晕并且有点累..."
elif mem > 80:
msg = "咱感觉有点累..."
elif disk > 80:
msg = "咱感觉身体要被塞满了..."
msg0 = (
"Self status:\n"
f"* CPU: {cpu}%\n"
f"* MEM: {mem}%\n"
f"* DISK: {disk}%\n"
f"* netSENT: {inteSENT}MB\n"
f"* netRECV: {inteRECV}MB\n"
) + msg
await status.finish(msg0)
|
