Removed decorator callback
The decorator callback was not actually necessary so it was removed and replaced with a partial function instead.
This commit is contained in:
parent
bcdfc555e0
commit
8406f81811
2 changed files with 6 additions and 16 deletions
|
@ -7,18 +7,13 @@ from homeassistant.helpers import event
|
|||
HASS = None
|
||||
|
||||
|
||||
def _callback(action, *args, **kwargs):
|
||||
""" adds HASS to callback arguments """
|
||||
action(HASS, *args, **kwargs)
|
||||
|
||||
|
||||
def track_state_change(entity_ids, from_state=None, to_state=None):
|
||||
""" Decorator factory to track state changes for entity id """
|
||||
|
||||
def track_state_change_decorator(action):
|
||||
""" Decorator to track state changes """
|
||||
event.track_state_change(HASS, entity_ids,
|
||||
functools.partial(_callback, action),
|
||||
functools.partial(action, HASS),
|
||||
from_state, to_state)
|
||||
return action
|
||||
|
||||
|
@ -31,7 +26,7 @@ def track_sunrise(offset=None):
|
|||
def track_sunrise_decorator(action):
|
||||
""" Decorator to track sunrise events """
|
||||
event.track_sunrise(HASS,
|
||||
functools.partial(_callback, action),
|
||||
functools.partial(action, HASS),
|
||||
offset)
|
||||
return action
|
||||
|
||||
|
@ -44,7 +39,7 @@ def track_sunset(offset=None):
|
|||
def track_sunset_decorator(action):
|
||||
""" Decorator to track sunset events """
|
||||
event.track_sunset(HASS,
|
||||
functools.partial(_callback, action),
|
||||
functools.partial(action, HASS),
|
||||
offset)
|
||||
return action
|
||||
|
||||
|
@ -59,7 +54,7 @@ def track_time_change(year=None, month=None, day=None, hour=None, minute=None,
|
|||
def track_time_change_decorator(action):
|
||||
""" Decorator to track time changes """
|
||||
event.track_time_change(HASS,
|
||||
functools.partial(_callback, action),
|
||||
functools.partial(action, HASS),
|
||||
year, month, day, hour, minute, second)
|
||||
return action
|
||||
|
||||
|
@ -74,7 +69,7 @@ def track_utc_time_change(year=None, month=None, day=None, hour=None,
|
|||
def track_utc_time_change_decorator(action):
|
||||
""" Decorator to track time changes """
|
||||
event.track_utc_time_change(HASS,
|
||||
functools.partial(_callback, action),
|
||||
functools.partial(action, HASS),
|
||||
year, month, day, hour, minute, second)
|
||||
return action
|
||||
|
||||
|
|
|
@ -15,18 +15,13 @@ CONF_SERVICE_DATA = 'data'
|
|||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
||||
def _callback(action, *args, **kwargs):
|
||||
""" adds HASS to callback arguments """
|
||||
action(HASS, *args, **kwargs)
|
||||
|
||||
|
||||
def service(domain, service_name):
|
||||
""" Decorator factory to register a service """
|
||||
|
||||
def register_service_decorator(action):
|
||||
""" Decorator to register a service """
|
||||
HASS.services.register(domain, service_name,
|
||||
functools.partial(_callback, action))
|
||||
functools.partial(action, HASS))
|
||||
return action
|
||||
|
||||
return register_service_decorator
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue