evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Black

Browse source
This commit is contained in:
Paulus Schoutsen 2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download patch file Download diff file Expand all files Collapse all files

111
homeassistant/components/tile/device_tracker.py
View file

@ -5,38 +5,40 @@ from datetime import timedelta
import voluptuous as vol
from homeassistant.components.device_tracker import PLATFORM_SCHEMA
from homeassistant.const import (
CONF_USERNAME, CONF_MONITORED_VARIABLES, CONF_PASSWORD)
from homeassistant.const import CONF_USERNAME, CONF_MONITORED_VARIABLES, CONF_PASSWORD
from homeassistant.helpers import aiohttp_client, config_validation as cv
from homeassistant.helpers.event import async_track_time_interval
from homeassistant.util import slugify
from homeassistant.util.json import load_json, save_json
_LOGGER = logging.getLogger(__name__)
CLIENT_UUID_CONFIG_FILE = '.tile.conf'
DEVICE_TYPES = ['PHONE', 'TILE']
CLIENT_UUID_CONFIG_FILE = ".tile.conf"
DEVICE_TYPES = ["PHONE", "TILE"]
ATTR_ALTITUDE = 'altitude'
ATTR_CONNECTION_STATE = 'connection_state'
ATTR_IS_DEAD = 'is_dead'
ATTR_IS_LOST = 'is_lost'
ATTR_RING_STATE = 'ring_state'
ATTR_VOIP_STATE = 'voip_state'
ATTR_TILE_ID = 'tile_identifier'
ATTR_TILE_NAME = 'tile_name'
ATTR_ALTITUDE = "altitude"
ATTR_CONNECTION_STATE = "connection_state"
ATTR_IS_DEAD = "is_dead"
ATTR_IS_LOST = "is_lost"
ATTR_RING_STATE = "ring_state"
ATTR_VOIP_STATE = "voip_state"
ATTR_TILE_ID = "tile_identifier"
ATTR_TILE_NAME = "tile_name"
CONF_SHOW_INACTIVE = 'show_inactive'
CONF_SHOW_INACTIVE = "show_inactive"
DEFAULT_ICON = 'mdi:view-grid'
DEFAULT_ICON = "mdi:view-grid"
DEFAULT_SCAN_INTERVAL = timedelta(minutes=2)
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_USERNAME): cv.string,
vol.Required(CONF_PASSWORD): cv.string,
vol.Optional(CONF_SHOW_INACTIVE, default=False): cv.boolean,
vol.Optional(CONF_MONITORED_VARIABLES, default=DEVICE_TYPES):
vol.All(cv.ensure_list, [vol.In(DEVICE_TYPES)]),
})
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
{
vol.Required(CONF_USERNAME): cv.string,
vol.Required(CONF_PASSWORD): cv.string,
vol.Optional(CONF_SHOW_INACTIVE, default=False): cv.boolean,
vol.Optional(CONF_MONITORED_VARIABLES, default=DEVICE_TYPES): vol.All(
cv.ensure_list, [vol.In(DEVICE_TYPES)]
),
}
)
async def async_setup_scanner(hass, config, async_see, discovery_info=None):
@ -45,26 +47,30 @@ async def async_setup_scanner(hass, config, async_see, discovery_info=None):
websession = aiohttp_client.async_get_clientsession(hass)
config_file = hass.config.path(".{}{}".format(
slugify(config[CONF_USERNAME]), CLIENT_UUID_CONFIG_FILE))
config_data = await hass.async_add_job(
load_json, config_file)
config_file = hass.config.path(
".{}{}".format(slugify(config[CONF_USERNAME]), CLIENT_UUID_CONFIG_FILE)
)
config_data = await hass.async_add_job(load_json, config_file)
if config_data:
client = Client(
config[CONF_USERNAME],
config[CONF_PASSWORD],
websession,
client_uuid=config_data['client_uuid'])
client_uuid=config_data["client_uuid"],
)
else:
client = Client(
config[CONF_USERNAME], config[CONF_PASSWORD], websession)
client = Client(config[CONF_USERNAME], config[CONF_PASSWORD], websession)
config_data = {'client_uuid': client.client_uuid}
config_data = {"client_uuid": client.client_uuid}
await hass.async_add_job(save_json, config_file, config_data)
scanner = TileScanner(
client, hass, async_see, config[CONF_MONITORED_VARIABLES],
config[CONF_SHOW_INACTIVE])
client,
hass,
async_see,
config[CONF_MONITORED_VARIABLES],
config[CONF_SHOW_INACTIVE],
)
return await scanner.async_init()
@ -86,13 +92,12 @@ class TileScanner:
try:
await self._client.async_init()
except TileError as err:
_LOGGER.error('Unable to set up Tile scanner: %s', err)
_LOGGER.error("Unable to set up Tile scanner: %s", err)
return False
await self._async_update()
async_track_time_interval(
self._hass, self._async_update, DEFAULT_SCAN_INTERVAL)
async_track_time_interval(self._hass, self._async_update, DEFAULT_SCAN_INTERVAL)
return True
@ -100,39 +105,37 @@ class TileScanner:
"""Update info from Tile."""
from pytile.errors import SessionExpiredError, TileError
_LOGGER.debug('Updating Tile data')
_LOGGER.debug("Updating Tile data")
try:
await self._client.async_init()
tiles = await self._client.tiles.all(
whitelist=self._types, show_inactive=self._show_inactive)
whitelist=self._types, show_inactive=self._show_inactive
)
except SessionExpiredError:
_LOGGER.info('Session expired; trying again shortly')
_LOGGER.info("Session expired; trying again shortly")
return
except TileError as err:
_LOGGER.error('There was an error while updating: %s', err)
_LOGGER.error("There was an error while updating: %s", err)
return
if not tiles:
_LOGGER.warning('No Tiles found')
_LOGGER.warning("No Tiles found")
return
for tile in tiles:
await self._async_see(
dev_id='tile_{0}'.format(slugify(tile['tile_uuid'])),
gps=(
tile['tileState']['latitude'],
tile['tileState']['longitude']
),
dev_id="tile_{0}".format(slugify(tile["tile_uuid"])),
gps=(tile["tileState"]["latitude"], tile["tileState"]["longitude"]),
attributes={
ATTR_ALTITUDE: tile['tileState']['altitude'],
ATTR_CONNECTION_STATE:
tile['tileState']['connection_state'],
ATTR_IS_DEAD: tile['is_dead'],
ATTR_IS_LOST: tile['tileState']['is_lost'],
ATTR_RING_STATE: tile['tileState']['ring_state'],
ATTR_VOIP_STATE: tile['tileState']['voip_state'],
ATTR_TILE_ID: tile['tile_uuid'],
ATTR_TILE_NAME: tile['name']
ATTR_ALTITUDE: tile["tileState"]["altitude"],
ATTR_CONNECTION_STATE: tile["tileState"]["connection_state"],
ATTR_IS_DEAD: tile["is_dead"],
ATTR_IS_LOST: tile["tileState"]["is_lost"],
ATTR_RING_STATE: tile["tileState"]["ring_state"],
ATTR_VOIP_STATE: tile["tileState"]["voip_state"],
ATTR_TILE_ID: tile["tile_uuid"],
ATTR_TILE_NAME: tile["name"],
},
icon=DEFAULT_ICON)
icon=DEFAULT_ICON,
)