bittles
db2c4ec880
split out sucks constants to their own file, no need to define verify ssl. if bumper is true then verify ssl false
109 lines
3.5 KiB
Python
109 lines
3.5 KiB
Python
"""Support for Ecovacs Deebot vacuums."""
|
|
import random
|
|
import string
|
|
#import asyncio ## to do will need to convert to slixmpp to do this i believe
|
|
|
|
from homeassistant.const import (
|
|
CONF_PASSWORD,
|
|
CONF_USERNAME,
|
|
EVENT_HOMEASSISTANT_STOP,
|
|
Platform,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import discovery
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.typing import ConfigType
|
|
import voluptuous as vol
|
|
#just included the modified sucks in component
|
|
from .sucksbumper import EcoVacsAPI, VacBot
|
|
from .const import (
|
|
ECOVACS_DEVICES,
|
|
DOMAIN,
|
|
CONF_COUNTRY,
|
|
CONF_CONTINENT,
|
|
CONF_BUMPER,
|
|
CONF_BUMPER_SERVER,
|
|
SERVER_ADDRESS,
|
|
VERIFY_SSL,
|
|
LOGGER
|
|
)
|
|
|
|
CONFIG_SCHEMA = vol.Schema(
|
|
{
|
|
DOMAIN: vol.Schema(
|
|
{
|
|
vol.Required(CONF_USERNAME): cv.string,
|
|
vol.Required(CONF_PASSWORD): cv.string,
|
|
vol.Required(CONF_COUNTRY): vol.All(vol.Lower, cv.string),
|
|
vol.Required(CONF_CONTINENT): vol.All(vol.Lower, cv.string),
|
|
vol.Optional(CONF_BUMPER, default=False): cv.boolean,
|
|
vol.Optional(CONF_BUMPER_SERVER): cv.string
|
|
}
|
|
)
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
|
|
# Generate a random device ID on each bootup
|
|
ECOVACS_API_DEVICEID = "".join(
|
|
random.choice(string.ascii_uppercase + string.digits) for _ in range(8)
|
|
)
|
|
|
|
def setup(hass: HomeAssistant, config: ConfigType) -> bool:
|
|
"""Set up the Ecovacs component."""
|
|
LOGGER.debug("Creating new Ecovacs component")
|
|
|
|
hass.data[ECOVACS_DEVICES] = []
|
|
# if we're using bumper then define the server address
|
|
if CONF_BUMPER == True:
|
|
SERVER_ADDRESS = (config[DOMAIN].get(CONF_BUMPER_SERVER), 5223)
|
|
VERIFY_SSL = False
|
|
# if not server address is null and verify ssl true from const
|
|
|
|
ecovacs_api = EcoVacsAPI(
|
|
ECOVACS_API_DEVICEID,
|
|
config[DOMAIN].get(CONF_USERNAME),
|
|
EcoVacsAPI.md5(config[DOMAIN].get(CONF_PASSWORD)),
|
|
config[DOMAIN].get(CONF_COUNTRY),
|
|
config[DOMAIN].get(CONF_CONTINENT),
|
|
VERIFY_SSL # add to class call
|
|
)
|
|
|
|
devices = ecovacs_api.devices()
|
|
LOGGER.debug("Ecobot devices: %s", devices)
|
|
|
|
for device in devices:
|
|
LOGGER.info(
|
|
"Discovered Ecovacs device on account: %s with nickname %s",
|
|
device.get("did"),
|
|
device.get("nick"),
|
|
)
|
|
vacbot = VacBot(
|
|
ecovacs_api.uid,
|
|
ecovacs_api.REALM,
|
|
ecovacs_api.resource,
|
|
ecovacs_api.user_access_token,
|
|
device,
|
|
config[DOMAIN].get(CONF_CONTINENT).lower(),
|
|
SERVER_ADDRESS, # include server address in class, if it's null shoul be no effect
|
|
VERIFY_SSL, # verify ssl or not
|
|
monitor=True
|
|
)
|
|
hass.data[ECOVACS_DEVICES].append(vacbot)
|
|
|
|
def stop(event: object) -> None:
|
|
"""Shut down open connections to Ecovacs XMPP server."""
|
|
for device in hass.data[ECOVACS_DEVICES]:
|
|
LOGGER.info(
|
|
"Shutting down connection to Ecovacs device %s",
|
|
device.vacuum.get("did"),
|
|
)
|
|
device.disconnect()
|
|
|
|
# Listen for HA stop to disconnect.
|
|
hass.bus.listen_once(EVENT_HOMEASSISTANT_STOP, stop)
|
|
if hass.data[ECOVACS_DEVICES]:
|
|
LOGGER.debug("Starting vacuum components")
|
|
discovery.load_platform(hass, Platform.VACUUM, DOMAIN, {}, config)
|
|
return True
|