Add partial backup reload options

aiohasupervisor/backups.py

@@ -15,6 +15,7 @@
     NewBackup,
     PartialBackupOptions,
     PartialRestoreOptions,
+    ReloadOptions,
 )
 
 
@@ -35,9 +36,11 @@ async def set_options(self, options: BackupsOptions) -> None:
         """Set options for backups."""
         await self._client.post("backups/options", json=options.to_dict())
 
-    async def reload(self) -> None:
+    async def reload(self, options: ReloadOptions | None = None) -> None:
         """Reload backups cache."""
-        await self._client.post("backups/reload")
+        await self._client.post(
+            "backups/reload", json=options.to_dict() if options else None
+        )
 
     async def freeze(self, options: FreezeOptions | None = None) -> None:
         """Start a freeze for external snapshot process."""

aiohasupervisor/models/__init__.py

@@ -40,6 +40,7 @@
     NewBackup,
     PartialBackupOptions,
     PartialRestoreOptions,
+    ReloadOptions,
 )
 from aiohasupervisor.models.discovery import (
     Discovery,
@@ -203,6 +204,7 @@
     "NewBackup",
     "PartialBackupOptions",
     "PartialRestoreOptions",
+    "ReloadOptions",
     "Discovery",
     "DiscoveryConfig",
     "AccessPoint",

aiohasupervisor/models/backups.py

@@ -48,6 +48,7 @@ class BackupBaseFields(ABC):
     type: BackupType
     size: float
     location: str | None
+    locations: list[str | None]
     protected: bool
     compressed: bool
 
@@ -167,3 +168,11 @@ class FullRestoreOptions(Request):
 @dataclass(frozen=True, slots=True)
 class PartialRestoreOptions(FullRestoreOptions, PartialBackupRestoreOptions):
     """PartialRestoreOptions model."""
+
+
+@dataclass(frozen=True, slots=True)
+class ReloadOptions(Request):
+    """ReloadOptions model."""
+
+    location: str | None
+    filename: str

tests/test_backups.py

@@ -15,6 +15,7 @@
     FullBackupOptions,
     PartialBackupOptions,
     PartialRestoreOptions,
+    ReloadOptions,
 )
 
 from . import load_fixture
@@ -28,7 +29,7 @@
     responses.get(
         f"{SUPERVISOR_URL}/backups", status=200, body=load_fixture("backups_list.json")
     )
     backups = await supervisor_client.backups.list()
     assert backups[0].slug == "58bc7491"
     assert backups[0].type == "full"
     assert backups[0].date == datetime(2024, 4, 6, 7, 5, 40, 0, UTC)
@@ -48,7 +49,7 @@
         status=200,
         body=load_fixture("backups_info.json"),
     )
     info = await supervisor_client.backups.info()
     assert info.backups[0].slug == "58bc7491"
     assert info.backups[1].slug == "69558789"
     assert info.days_until_stale == 30
@@ -79,6 +80,22 @@
     }
 
 
+async def test_backups_partial_reload(
+    responses: aioresponses, supervisor_client: SupervisorClient
+) -> None:
+    """Test backups partial reload API."""
+    responses.post(f"{SUPERVISOR_URL}/backups/reload", status=200)
+    assert (
+        await supervisor_client.backups.reload(
+            ReloadOptions(location=None, filename="test.tar")
+        )
+        is None
+    )
+    assert responses.requests.keys() == {
+        ("POST", URL(f"{SUPERVISOR_URL}/backups/reload"))
+    }
+
+
 @pytest.mark.parametrize("options", [None, FreezeOptions(timeout=1000)])
 async def test_backups_freeze(
     responses: aioresponses,
@@ -192,7 +209,7 @@
         status=200,
         body=load_fixture("backup_info.json"),
     )
     result = await supervisor_client.backups.backup_info("69558789")
     assert result.slug == "69558789"
     assert result.type == "partial"
     assert result.date == datetime(2024, 5, 31, 0, 0, 0, 0, UTC)
@@ -222,7 +239,7 @@
         status=200,
         body=load_fixture("backup_info_no_homeassistant.json"),
     )
     result = await supervisor_client.backups.backup_info("d13dedd0")
     assert result.slug == "d13dedd0"
     assert result.type == "partial"
     assert result.homeassistant is None