"""Sensor for Xbox Live account status.""" import logging import voluptuous as vol import homeassistant.helpers.config_validation as cv from homeassistant.components.sensor import PLATFORM_SCHEMA from homeassistant.const import CONF_API_KEY, STATE_UNKNOWN from homeassistant.helpers.entity import Entity _LOGGER = logging.getLogger(__name__) CONF_XUID = "xuid" ICON = "mdi:xbox" PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( { vol.Required(CONF_API_KEY): cv.string, vol.Required(CONF_XUID): vol.All(cv.ensure_list, [cv.string]), } ) def setup_platform(hass, config, add_entities, discovery_info=None): """Set up the Xbox platform.""" from xboxapi import xbox_api api = xbox_api.XboxApi(config.get(CONF_API_KEY)) devices = [] # request personal profile to check api connection profile = api.get_profile() if profile.get("error_code") is not None: _LOGGER.error( "Can't setup XboxAPI connection. Check your account or " " api key on xboxapi.com. Code: %s Description: %s ", profile.get("error_code", STATE_UNKNOWN), profile.get("error_message", STATE_UNKNOWN), ) return for xuid in config.get(CONF_XUID): new_device = XboxSensor(hass, api, xuid) if new_device.success_init: devices.append(new_device) if devices: add_entities(devices, True) class XboxSensor(Entity): """A class for the Xbox account.""" def __init__(self, hass, api, xuid): """Initialize the sensor.""" self._hass = hass self._state = None self._presence = {} self._xuid = xuid self._api = api # get profile info profile = self._api.get_user_gamercard(self._xuid) if profile.get("success", True) and profile.get("code") is None: self.success_init = True self._gamertag = profile.get("gamertag") self._gamerscore = profile.get("gamerscore") self._picture = profile.get("gamerpicSmallSslImagePath") self._tier = profile.get("tier") else: _LOGGER.error( "Can't get user profile %s. " "Error Code: %s Description: %s", self._xuid, profile.get("code", STATE_UNKNOWN), profile.get("description", STATE_UNKNOWN), ) self.success_init = False @property def name(self): """Return the name of the sensor.""" return self._gamertag @property def state(self): """Return the state of the sensor.""" return self._state @property def device_state_attributes(self): """Return the state attributes.""" attributes = {} attributes["gamerscore"] = self._gamerscore attributes["tier"] = self._tier for device in self._presence: for title in device.get("titles"): attributes[ "{} {}".format(device.get("type"), title.get("placement")) ] = title.get("name") return attributes @property def entity_picture(self): """Avatar of the account.""" return self._picture @property def icon(self): """Return the icon to use in the frontend.""" return ICON def update(self): """Update state data from Xbox API.""" presence = self._api.get_user_presence(self._xuid) self._state = presence.get("state") self._presence = presence.get("devices", {})