"""Utils for D-Bus."""

from ..exceptions import DBusNotConnectedError


def dbus_connected(method):
    """Wrapper for check if D-Bus is connected."""
    def wrap_dbus(api, *args, **kwargs):
        """Check if D-Bus is connected before call a method."""
        if api.dbus is None:
            raise DBusNotConnectedError()
        return method(api, *args, **kwargs)

    return wrap_dbus