Move imports in greenwave component (#27998)

This commit is contained in:
Diefferson Koderer Môro 2019-10-21 05:22:52 -03:00 committed by Fabian Affolter
parent d5799d020a
commit 36ff790a39

View File

@ -1,7 +1,9 @@
"""Support for Greenwave Reality (TCP Connected) lights.""" """Support for Greenwave Reality (TCP Connected) lights."""
import logging
from datetime import timedelta from datetime import timedelta
import logging
import os
import greenwavereality as greenwave
import voluptuous as vol import voluptuous as vol
from homeassistant.components.light import ( from homeassistant.components.light import (
@ -29,9 +31,6 @@ MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=1)
def setup_platform(hass, config, add_entities, discovery_info=None): def setup_platform(hass, config, add_entities, discovery_info=None):
"""Set up the Greenwave Reality Platform.""" """Set up the Greenwave Reality Platform."""
import greenwavereality as greenwave
import os
host = config.get(CONF_HOST) host = config.get(CONF_HOST)
tokenfile = hass.config.path(".greenwave") tokenfile = hass.config.path(".greenwave")
if config.get(CONF_VERSION) == 3: if config.get(CONF_VERSION) == 3:
@ -60,8 +59,6 @@ class GreenwaveLight(Light):
def __init__(self, light, host, token, gatewaydata): def __init__(self, light, host, token, gatewaydata):
"""Initialize a Greenwave Reality Light.""" """Initialize a Greenwave Reality Light."""
import greenwavereality as greenwave
self._did = int(light["did"]) self._did = int(light["did"])
self._name = light["name"] self._name = light["name"]
self._state = int(light["state"]) self._state = int(light["state"])
@ -98,22 +95,16 @@ class GreenwaveLight(Light):
def turn_on(self, **kwargs): def turn_on(self, **kwargs):
"""Instruct the light to turn on.""" """Instruct the light to turn on."""
import greenwavereality as greenwave
temp_brightness = int((kwargs.get(ATTR_BRIGHTNESS, 255) / 255) * 100) temp_brightness = int((kwargs.get(ATTR_BRIGHTNESS, 255) / 255) * 100)
greenwave.set_brightness(self._host, self._did, temp_brightness, self._token) greenwave.set_brightness(self._host, self._did, temp_brightness, self._token)
greenwave.turn_on(self._host, self._did, self._token) greenwave.turn_on(self._host, self._did, self._token)
def turn_off(self, **kwargs): def turn_off(self, **kwargs):
"""Instruct the light to turn off.""" """Instruct the light to turn off."""
import greenwavereality as greenwave
greenwave.turn_off(self._host, self._did, self._token) greenwave.turn_off(self._host, self._did, self._token)
def update(self): def update(self):
"""Fetch new state data for this light.""" """Fetch new state data for this light."""
import greenwavereality as greenwave
self._gatewaydata.update() self._gatewaydata.update()
bulbs = self._gatewaydata.greenwave bulbs = self._gatewaydata.greenwave
@ -128,8 +119,6 @@ class GatewayData:
def __init__(self, host, token): def __init__(self, host, token):
"""Initialize the data object.""" """Initialize the data object."""
import greenwavereality as greenwave
self._host = host self._host = host
self._token = token self._token = token
self._greenwave = greenwave.grab_bulbs(host, token) self._greenwave = greenwave.grab_bulbs(host, token)
@ -142,7 +131,5 @@ class GatewayData:
@Throttle(MIN_TIME_BETWEEN_UPDATES) @Throttle(MIN_TIME_BETWEEN_UPDATES)
def update(self): def update(self):
"""Get the latest data from the gateway.""" """Get the latest data from the gateway."""
import greenwavereality as greenwave
self._greenwave = greenwave.grab_bulbs(self._host, self._token) self._greenwave = greenwave.grab_bulbs(self._host, self._token)
return self._greenwave return self._greenwave