Cleanups

hassio/dbus/utils.py

@@ -0,0 +1,14 @@
+"""Utils for dbus."""
+
+from ..exceptions import HassioNotSupportedError
+
+
+def dbus_connected(method):
+    """Wrapper for check if dbus is connected."""
+    def wrap_dbus(self, *args, **kwargs):
+        """Check if dbus is connected before call a method."""
+        if self.dbus is None:
+            raise HassioNotSupportedError(f"{self!s} not connected to dbus!")
+        return self.method(*args, **kwargs)
+
+    return wrap_dbus