Move imports in watson_iot component (#27448)

This commit is contained in:
Quentame 2019-10-12 22:01:35 +02:00 committed by Paulus Schoutsen
parent 3b9ee9c901
commit 701bb666c4

View File

@ -4,6 +4,8 @@ import queue
import threading import threading
import time import time
from ibmiotf import MissingMessageEncoderException
from ibmiotf.gateway import Client
import voluptuous as vol import voluptuous as vol
from homeassistant.const import ( from homeassistant.const import (
@ -67,7 +69,6 @@ CONFIG_SCHEMA = vol.Schema(
def setup(hass, config): def setup(hass, config):
"""Set up the Watson IoT Platform component.""" """Set up the Watson IoT Platform component."""
from ibmiotf import gateway
conf = config[DOMAIN] conf = config[DOMAIN]
@ -85,7 +86,7 @@ def setup(hass, config):
"auth-method": "token", "auth-method": "token",
"auth-token": conf[CONF_TOKEN], "auth-token": conf[CONF_TOKEN],
} }
watson_gateway = gateway.Client(client_args) watson_gateway = Client(client_args)
def event_to_json(event): def event_to_json(event):
"""Add an event to the outgoing list.""" """Add an event to the outgoing list."""
@ -190,7 +191,6 @@ class WatsonIOTThread(threading.Thread):
def write_to_watson(self, events): def write_to_watson(self, events):
"""Write preprocessed events to watson.""" """Write preprocessed events to watson."""
import ibmiotf
for event in events: for event in events:
for retry in range(MAX_TRIES + 1): for retry in range(MAX_TRIES + 1):
@ -208,7 +208,7 @@ class WatsonIOTThread(threading.Thread):
_LOGGER.error("Failed to publish message to Watson IoT") _LOGGER.error("Failed to publish message to Watson IoT")
continue continue
break break
except (ibmiotf.MissingMessageEncoderException, IOError): except (MissingMessageEncoderException, IOError):
if retry < MAX_TRIES: if retry < MAX_TRIES:
time.sleep(RETRY_DELAY) time.sleep(RETRY_DELAY)
else: else: