pylint rework for light/mqtt

This commit is contained in:
root 2015-10-25 22:58:07 +01:00
parent 469d0619ba
commit 7cfce94dfb

View File

@ -5,9 +5,8 @@ Allows to configure a MQTT light.
""" """
import logging import logging
import homeassistant.components.mqtt as mqtt import homeassistant.components.mqtt as mqtt
from homeassistant.components.light import (Light, ATTR_BRIGHTNESS, ATTR_RGB_COLOR, ATTR_XY_COLOR) from homeassistant.components.light import (Light,
ATTR_BRIGHTNESS, ATTR_RGB_COLOR)
import random
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -55,47 +54,61 @@ def setup_platform(hass, config, add_devices_callback, discovery_info=None):
class MqttLight(Light): class MqttLight(Light):
""" Provides a demo switch. """ """ Provides a demo switch. """
def __init__(self, hass, name, state_topic, command_topic, brightness_state_topic, brightness_command_topic, rgb_state_topic, rgb_command_topic, rgb, qos, payload_on, payload_off, brightness):
self._name = name # pylint: disable=too-many-instance-attributes,too-many-arguments,too-many-locals,bad-builtin
# Eight is reasonable in this case.
def __init__(self, hass, name,
state_topic, command_topic,
brightness_state_topic, brightness_command_topic,
rgb_state_topic, rgb_command_topic,
rgb, qos,
payload_on, payload_off,
brightness):
self._hass = hass self._hass = hass
self._state = False self._name = name
self._command_topic = command_topic
self._state_topic = state_topic self._state_topic = state_topic
self._command_topic = command_topic
self._brightness_state_topic = brightness_state_topic self._brightness_state_topic = brightness_state_topic
self._brightness_command_topic = brightness_command_topic self._brightness_command_topic = brightness_command_topic
self._rgb_state_topic = rgb_state_topic self._rgb_state_topic = rgb_state_topic
self._rgb_command_topic = rgb_command_topic self._rgb_command_topic = rgb_command_topic
self._rgb = rgb
self._qos = qos self._qos = qos
self._payload_on = payload_on self._payload_on = payload_on
self._payload_off = payload_off self._payload_off = payload_off
self._rgb = rgb
self._brightness = brightness self._brightness = brightness
self._xy = [[0.5, 0.5]] self._xy = [[0.5, 0.5]]
self._state = False
def message_received(topic, payload, qos): def message_received(topic, payload, qos):
""" A new MQTT message has been received. """ """ A new MQTT message has been received. """
if payload == self._payload_on: if payload == self._payload_on:
self._state = True self._state = True
self.update_ha_state() self._hass.update_ha_state()
elif payload == self._payload_off: elif payload == self._payload_off:
self._state = False self._state = False
self.update_ha_state() self._hass.update_ha_state()
def brightness_received(topic, payload, qos): def brightness_received(topic, payload, qos):
""" A new MQTT message has been received. """ """ A new MQTT message has been received. """
self._brightness = int(payload) self._brightness = int(payload)
self.update_ha_state() self._hass.update_ha_state()
def rgb_received(topic, payload, qos): def rgb_received(topic, payload, qos):
""" A new MQTT message has been received. """ """ A new MQTT message has been received. """
rgb = payload.split(",") rgb = payload.split(",")
self._rgb = list(map(int, rgb)) self._rgb = list(map(int, rgb))
self.update_ha_state() self._hass.update_ha_state()
# subscribe the state_topic # subscribe the state_topic
mqtt.subscribe(self._hass, self._state_topic, message_received, self._qos) mqtt.subscribe(self._hass, self._state_topic,
mqtt.subscribe(self._hass, self._brightness_state_topic, brightness_received, self._qos) message_received, self._qos)
mqtt.subscribe(self._hass, self._rgb_state_topic, rgb_received, self._qos) mqtt.subscribe(self._hass, self._brightness_state_topic,
brightness_received, self._qos)
mqtt.subscribe(self._hass, self._rgb_state_topic,
rgb_received, self._qos)
@property @property
def should_poll(self): def should_poll(self):
@ -128,6 +141,7 @@ class MqttLight(Light):
return self._state return self._state
def turn_on(self, **kwargs): def turn_on(self, **kwargs):
""" Turn the device on. """
if ATTR_RGB_COLOR in kwargs: if ATTR_RGB_COLOR in kwargs:
self._rgb = kwargs[ATTR_RGB_COLOR] self._rgb = kwargs[ATTR_RGB_COLOR]
@ -136,17 +150,18 @@ class MqttLight(Light):
if ATTR_BRIGHTNESS in kwargs: if ATTR_BRIGHTNESS in kwargs:
self._brightness = kwargs[ATTR_BRIGHTNESS] self._brightness = kwargs[ATTR_BRIGHTNESS]
mqtt.publish(self._hass, self._brightness_command_topic, self._brightness, self._qos) mqtt.publish(self._hass, self._brightness_command_topic,
self._brightness, self._qos)
if not self._state: if not self._state:
""" Turn the device on. """
self._state = True self._state = True
mqtt.publish(self._hass, self._command_topic, self._payload_on, self._qos) mqtt.publish(self._hass, self._command_topic,
self.update_ha_state() self._payload_on, self._qos)
self._hass.update_ha_state()
def turn_off(self, **kwargs): def turn_off(self, **kwargs):
""" Turn the device off. """ """ Turn the device off. """
self._state = False self._state = False
mqtt.publish(self._hass, self._command_topic, self._payload_off,self._qos) mqtt.publish(self._hass, self._command_topic,
self.update_ha_state() self._payload_off, self._qos)
self._hass.update_ha_state()