Placeholder

Author: dormant-user

pyninja/monitor/resources.py

@@ -7,6 +7,7 @@
 import shutil
 import subprocess
 import time
+from collections.abc import Generator
 from datetime import datetime, timedelta
 from typing import Any, Dict, List
 
@@ -183,26 +184,27 @@ async def get_system_metrics() -> Dict[str, dict]:
 
 
 @cache.timed_cache(60)
-def disk_utils_metrics() -> List[Dict[str, int | str | float | dict]]:
-    """Placeholder for future metrics."""
-    from pyudisk.main import EnvConfig, smart_metrics
-    from pyudisk.util import kelvin_to_fahrenheit
+def pyudisk_metrics() -> Generator[Dict[str, int | str | float | dict]]:
+    """Retrieves metrics from PyUdisk library.
 
-    return [
-        {
+    Returns:
+
+    """
+    from pyudisk import util, smart_metrics, EnvConfig
+
+    for disk in smart_metrics(EnvConfig()):
+        yield {
             **{
                 "Model": disk.Info.Model,
                 "Mountpoint": str(disk.Partition.MountPoints),
-                "Temperature": f"{kelvin_to_fahrenheit(disk.Attributes.SmartTemperature)} °F",
+                "Temperature": f"{util.kelvin_to_fahrenheit(disk.Attributes.SmartTemperature)} °F / {util.kelvin_to_celsius(disk.Attributes.SmartTemperature)} °C",
                 "Bad Sectors": disk.Attributes.SmartNumBadSectors,
                 "Timestamp": datetime.fromtimestamp(
                     disk.Attributes.SmartUpdated
                 ).strftime("%Y-%m-%d %H:%M:%S"),
             },
             **disk.Usage.model_dump(),
         }
-        for disk in smart_metrics(EnvConfig())
-    ]
 
 
 async def system_resources() -> Dict[str, dict]:

requirements.txt

@@ -1,11 +1,11 @@
-click==8.1.7
-docker==7.1.0
-fastapi==0.112.0
-Jinja2==3.1.4
-psutil==6.0.0
-pydantic==2.8.2
-pydantic-settings==2.4.0
-python-multipart==0.0.16
-PyYaml==6.0.2
-requests
-uvicorn[standard]==0.30.5
+click==8.1.*
+docker==7.1.*
+fastapi==0.112.*
+Jinja2==3.1.*
+psutil==6.0.*
+pydantic==2.8.*
+pydantic-settings==2.4.*
+python-multipart==0.0.*
+PyYaml==6.0.*
+requests==2.32.*
+uvicorn[standard]==0.30.*