evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Move imports in yeelight + yeelightsunflower component (#27388)

Browse source 
* Move imports in yeelight + yeelightsunflower component

* Fix pylint

* Fix pylint (again)
This commit is contained in:
Quentame 2019-10-15 13:37:40 +02:00 • committed by Martin Hjelmare
parent 0463349f02
commit 5b1f44ba19
2 changed files with 28 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

62
homeassistant/components/yeelight/light.py
View file

@ -2,8 +2,16 @@
import logging
import voluptuous as vol
from yeelight import RGBTransition, SleepTransition, Flow, BulbException
import yeelight
from yeelight import (
RGBTransition,
SleepTransition,
Flow,
BulbException,
transitions as yee_transitions,
)
from yeelight.enums import PowerMode, LightType, BulbType, SceneClass
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.service import extract_entity_ids
import homeassistant.helpers.config_validation as cv
@ -190,8 +198,6 @@ SERVICE_SCHEMA_SET_AUTO_DELAY_OFF = YEELIGHT_SERVICE_SCHEMA.extend(
def _transitions_config_parser(transitions):
"""Parse transitions config into initialized objects."""
import yeelight
transition_objects = []
for transition_config in transitions:
transition, params = list(transition_config.items())[0]
@ -652,39 +658,23 @@ class YeelightGenericLight(Light):
def set_effect(self, effect) -> None:
"""Activate effect."""
if effect:
from yeelight.transitions import (
disco,
temp,
strobe,
pulse,
strobe_color,
alarm,
police,
police2,
christmas,
rgb,
randomloop,
lsd,
slowdown,
)
if effect == EFFECT_STOP:
self._bulb.stop_flow(light_type=self.light_type)
return
effects_map = {
EFFECT_DISCO: disco,
EFFECT_TEMP: temp,
EFFECT_STROBE: strobe,
EFFECT_STROBE_COLOR: strobe_color,
EFFECT_ALARM: alarm,
EFFECT_POLICE: police,
EFFECT_POLICE2: police2,
EFFECT_CHRISTMAS: christmas,
EFFECT_RGB: rgb,
EFFECT_RANDOM_LOOP: randomloop,
EFFECT_LSD: lsd,
EFFECT_SLOWDOWN: slowdown,
EFFECT_DISCO: yee_transitions.disco,
EFFECT_TEMP: yee_transitions.temp,
EFFECT_STROBE: yee_transitions.strobe,
EFFECT_STROBE_COLOR: yee_transitions.strobe_color,
EFFECT_ALARM: yee_transitions.alarm,
EFFECT_POLICE: yee_transitions.police,
EFFECT_POLICE2: yee_transitions.police2,
EFFECT_CHRISTMAS: yee_transitions.christmas,
EFFECT_RGB: yee_transitions.rgb,
EFFECT_RANDOM_LOOP: yee_transitions.randomloop,
EFFECT_LSD: yee_transitions.lsd,
EFFECT_SLOWDOWN: yee_transitions.slowdown,
}
if effect in self.custom_effects_names:
@ -692,13 +682,15 @@ class YeelightGenericLight(Light):
elif effect in effects_map:
flow = Flow(count=0, transitions=effects_map[effect]())
elif effect == EFFECT_FAST_RANDOM_LOOP:
flow = Flow(count=0, transitions=randomloop(duration=250))
flow = Flow(
count=0, transitions=yee_transitions.randomloop(duration=250)
)
elif effect == EFFECT_WHATSAPP:
flow = Flow(count=2, transitions=pulse(37, 211, 102))
flow = Flow(count=2, transitions=yee_transitions.pulse(37, 211, 102))
elif effect == EFFECT_FACEBOOK:
flow = Flow(count=2, transitions=pulse(59, 89, 152))
flow = Flow(count=2, transitions=yee_transitions.pulse(59, 89, 152))
elif effect == EFFECT_TWITTER:
flow = Flow(count=2, transitions=pulse(0, 172, 237))
flow = Flow(count=2, transitions=yee_transitions.pulse(0, 172, 237))
try:
self._bulb.start_flow(flow, light_type=self.light_type)

2
homeassistant/components/yeelightsunflower/light.py
View file

@ -1,6 +1,7 @@
"""Support for Yeelight Sunflower color bulbs (not Yeelight Blue or WiFi)."""
import logging
import yeelightsunflower
import voluptuous as vol
import homeassistant.helpers.config_validation as cv
@ -24,7 +25,6 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({vol.Required(CONF_HOST): cv.string})
def setup_platform(hass, config, add_entities, discovery_info=None):
"""Set up the Yeelight Sunflower Light platform."""
import yeelightsunflower
host = config.get(CONF_HOST)
hub = yeelightsunflower.Hub(host)