✨ 新增可查询数据

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,
     )