Files
supervisor/hassio/dbus/utils.py
2018-04-23 21:10:48 +02:00

15 lines
427 B
Python

"""Utils for dbus."""
from ..exceptions import DBusNotConnectedError
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 DBusNotConnectedError(f"{self!s} not connected to dbus!")
return self.method(*args, **kwargs)
return wrap_dbus