summaryrefslogtreecommitdiff
path: root/ATRI
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2023-05-09 17:03:24 +0800
committerKyomotoi <[email protected]>2023-05-09 17:03:24 +0800
commit6b3766c65af44154eae26ecc0a0b3a7bca9b2ea3 (patch)
tree0e52a1bbecffa92e8c99424b364f1bc7e29af20c /ATRI
parentee7654a790f00184145e4470adc5be025f406d3b (diff)
downloadATRI-6b3766c65af44154eae26ecc0a0b3a7bca9b2ea3.tar.gz
ATRI-6b3766c65af44154eae26ecc0a0b3a7bca9b2ea3.tar.bz2
ATRI-6b3766c65af44154eae26ecc0a0b3a7bca9b2ea3.zip
✨ 新增可查询数据
Diffstat (limited to 'ATRI')
-rw-r--r--ATRI/utils/machine.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/ATRI/utils/machine.py b/ATRI/utils/machine.py
index ee2b533..1c13d3f 100644
--- a/ATRI/utils/machine.py
+++ b/ATRI/utils/machine.py
@@ -46,6 +46,7 @@ class DiskInfo(BaseModel):
used: int
free: int
drive: int
+ speed: list[int]
class NetInfo(BaseModel):
@@ -104,6 +105,22 @@ def get_mem_info() -> MemInfo:
)
+last_disk_io = [0, 0]
+now_disk_io = [0, 0]
+
+
[email protected]_job("interval", seconds=1, misfire_grace_time=15)
+async def _():
+ global last_disk_io, now_disk_io
+
+ disk_counters = psutil.disk_io_counters()
+ now_disk_io = [
+ disk_counters.write_bytes - last_disk_io[0], # type: ignore
+ disk_counters.read_bytes - last_disk_io[1], # type: ignore
+ ]
+ last_disk_io = [disk_counters.write_bytes, disk_counters.read_bytes] # type: ignore
+
+
def get_disk_info():
disk_name = "ATRI Disk Pro Plus"
disk_total = int()
@@ -133,6 +150,7 @@ def get_disk_info():
used=disk_used,
free=disk_free,
drive=disk_drive,
+ speed=now_disk_io,
)