2c07bfb9e0
* Remove dependencies and requirements * Revert "Remove dependencies and requirements" This reverts commitfe7171b4cd. * Remove dependencies and requirements * Revert "Remove dependencies and requirements" This reverts commit391355ee2c. * Remove dependencies and requirements * Fix flake8 complaints * Fix more flake8 complaints * Revert non-component removals
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
"""Support for interacting with UpCloud servers."""
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.switch import PLATFORM_SCHEMA, SwitchDevice
|
|
from homeassistant.const import STATE_OFF
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
from . import CONF_SERVERS, DATA_UPCLOUD, UpCloudServerEntity
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
|
vol.Required(CONF_SERVERS): vol.All(cv.ensure_list, [cv.string]),
|
|
})
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the UpCloud server switch."""
|
|
upcloud = hass.data[DATA_UPCLOUD]
|
|
|
|
servers = config.get(CONF_SERVERS)
|
|
|
|
devices = [UpCloudSwitch(upcloud, uuid) for uuid in servers]
|
|
|
|
add_entities(devices, True)
|
|
|
|
|
|
class UpCloudSwitch(UpCloudServerEntity, SwitchDevice):
|
|
"""Representation of an UpCloud server switch."""
|
|
|
|
def turn_on(self, **kwargs):
|
|
"""Start the server."""
|
|
if self.state == STATE_OFF:
|
|
self.data.start()
|
|
|
|
def turn_off(self, **kwargs):
|
|
"""Stop the server."""
|
|
if self.is_on:
|
|
self.data.stop()
|