Code refactoring

domalab · Dec 27, 2024 · b6eb270 · b6eb270
1 parent 619e2c5
commit b6eb270
Show file tree

Hide file tree

Showing 17 changed files with 2,124 additions and 1,219 deletions.
diff --git a/custom_components/unraid/api/__init__.py b/custom_components/unraid/api/__init__.py
@@ -9,6 +9,7 @@
 from .smart_operations import SmartDataManager
 from .disk_state import DiskStateManager, DiskState
 from .disk_utils import is_valid_disk_name
+from .disk_mapping import get_unraid_disk_mapping, get_disk_info
 
 __all__ = [
     "DiskOperationsMixin",
@@ -23,4 +24,6 @@
     "DiskStateManager",
     "DiskState",
     "is_valid_disk_name",
+    "get_unraid_disk_mapping",
+    "get_disk_info",
 ]
diff --git a/custom_components/unraid/disk_mapping.py → custom_components/unraid/api/disk_mapping.py b/custom_components/unraid/disk_mapping.py → custom_components/unraid/api/disk_mapping.py
@@ -3,7 +3,7 @@
 from typing import Dict, Any
 import logging
 
-from .helpers import format_bytes
+from ..helpers import format_bytes
 
 _LOGGER = logging.getLogger(__name__)
 

diff --git a/custom_components/unraid/api/disk_operations.py b/custom_components/unraid/api/disk_operations.py
@@ -10,7 +10,7 @@
 from datetime import datetime
 
 from .disk_utils import is_valid_disk_name
-from ..disk_mapping import parse_disk_config, parse_disks_ini
+from .disk_mapping import parse_disk_config, parse_disks_ini
 from .smart_operations import SmartDataManager
 from .disk_state import DiskState, DiskStateManager
 

diff --git a/custom_components/unraid/api/disk_state.py b/custom_components/unraid/api/disk_state.py
@@ -194,4 +194,4 @@ def get_spindown_delay(self, disk_name: str) -> int:
         return self._spindown_delays.get(
             disk_name, 
             self._spindown_delays.get("default", 1800)
-        )
+        )