Added load_unit_obj wrapper function such as get_unit_obj.

bernhardkaindl · May 24, 2023 · f9b30d6 · f9b30d6
1 parent 3b16a8c
commit f9b30d6
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/sdbus_async/systemd/objects.py b/sdbus_async/systemd/objects.py
@@ -22,7 +22,11 @@ def __init__(self, bus: Optional[SdBus] = None) -> None:
 
     async def get_unit_obj(self, name: str) -> SystemdUnit:
         object_path = await self.get_unit(name)
-        return SystemdUnit(object_path)
+        return SystemdUnit(object_path, self._attached_bus)
+
+    async def load_unit_obj(self, name: str) -> SystemdUnit:
+        object_path = await self.load_unit(name)
+        return SystemdUnit(object_path, self._attached_bus)
 
 
 class SystemdUnit(SystemdUnitInterfaceAsync):

diff --git a/sdbus_block/systemd/objects.py b/sdbus_block/systemd/objects.py
@@ -23,7 +23,11 @@ def __init__(self, bus: Optional[SdBus] = None) -> None:
 
     def get_unit_obj(self, name: str) -> SystemdUnit:
         object_path = self.get_unit(name)
-        return SystemdUnit(object_path)
+        return SystemdUnit(object_path, self._attached_bus)
+
+    def load_unit_obj(self, name: str) -> SystemdUnit:
+        object_path = self.load_unit(name)
+        return SystemdUnit(object_path, self._attached_bus)
 
 
 class SystemdUnit(SystemdUnitInterface):