Consolidate all platforms that have tests (#22109)
* Moved climate components with tests into platform dirs. * Updated tests from climate component. * Moved binary_sensor components with tests into platform dirs. * Updated tests from binary_sensor component. * Moved calendar components with tests into platform dirs. * Updated tests from calendar component. * Moved camera components with tests into platform dirs. * Updated tests from camera component. * Moved cover components with tests into platform dirs. * Updated tests from cover component. * Moved device_tracker components with tests into platform dirs. * Updated tests from device_tracker component. * Moved fan components with tests into platform dirs. * Updated tests from fan component. * Moved geo_location components with tests into platform dirs. * Updated tests from geo_location component. * Moved image_processing components with tests into platform dirs. * Updated tests from image_processing component. * Moved light components with tests into platform dirs. * Updated tests from light component. * Moved lock components with tests into platform dirs. * Moved media_player components with tests into platform dirs. * Updated tests from media_player component. * Moved scene components with tests into platform dirs. * Moved sensor components with tests into platform dirs. * Updated tests from sensor component. * Moved switch components with tests into platform dirs. * Updated tests from sensor component. * Moved vacuum components with tests into platform dirs. * Updated tests from vacuum component. * Moved weather components with tests into platform dirs. * Fixed __init__.py files * Fixes for stuff moved as part of this branch. * Fix stuff needed to merge with balloob's branch. * Formatting issues. * Missing __init__.py files. * Fix-ups * Fixup * Regenerated requirements. * Linting errors fixed. * Fixed more broken tests. * Missing init files. * Fix broken tests. * More broken tests * There seems to be a thread race condition. I suspect the logger stuff is running in another thread, which means waiting until the aio loop is done is missing the log messages. Used sleep instead because that allows the logger thread to run. I think the api_streams sensor might not be thread safe. * Disabled tests, will remove sensor in #22147 * Updated coverage and codeowners.
This commit is contained in:
Penny Wood
Paulus Schoutsen
parent
46ece3603f
commit
f195ecca4b
486 changed files with 662 additions and 455 deletions
|
|
@ -1,221 +0,0 @@
|
|||
"""
|
||||
Support for Radarr.
|
||||
|
||||
For more details about this platform, please refer to the documentation at
|
||||
https://home-assistant.io/components/sensor.radarr/
|
||||
"""
|
||||
import logging
|
||||
import time
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
import requests
|
||||
import voluptuous as vol
|
||||
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
from homeassistant.const import (
|
||||
CONF_API_KEY, CONF_HOST, CONF_PORT, CONF_MONITORED_CONDITIONS, CONF_SSL)
|
||||
from homeassistant.helpers.entity import Entity
|
||||
from homeassistant.components.sensor import PLATFORM_SCHEMA
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
CONF_DAYS = 'days'
|
||||
CONF_INCLUDED = 'include_paths'
|
||||
CONF_UNIT = 'unit'
|
||||
CONF_URLBASE = 'urlbase'
|
||||
|
||||
DEFAULT_HOST = 'localhost'
|
||||
DEFAULT_PORT = 7878
|
||||
DEFAULT_URLBASE = ''
|
||||
DEFAULT_DAYS = '1'
|
||||
DEFAULT_UNIT = 'GB'
|
||||
|
||||
SCAN_INTERVAL = timedelta(minutes=10)
|
||||
|
||||
SENSOR_TYPES = {
|
||||
'diskspace': ['Disk Space', 'GB', 'mdi:harddisk'],
|
||||
'upcoming': ['Upcoming', 'Movies', 'mdi:television'],
|
||||
'wanted': ['Wanted', 'Movies', 'mdi:television'],
|
||||
'movies': ['Movies', 'Movies', 'mdi:television'],
|
||||
'commands': ['Commands', 'Commands', 'mdi:code-braces'],
|
||||
'status': ['Status', 'Status', 'mdi:information']
|
||||
}
|
||||
|
||||
ENDPOINTS = {
|
||||
'diskspace': 'http{0}://{1}:{2}/{3}api/diskspace',
|
||||
'upcoming':
|
||||
'http{0}://{1}:{2}/{3}api/calendar?start={4}&end={5}',
|
||||
'movies': 'http{0}://{1}:{2}/{3}api/movie',
|
||||
'commands': 'http{0}://{1}:{2}/{3}api/command',
|
||||
'status': 'http{0}://{1}:{2}/{3}api/system/status'
|
||||
}
|
||||
|
||||
# Support to Yottabytes for the future, why not
|
||||
BYTE_SIZES = ['B', 'KB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB']
|
||||
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
|
||||
vol.Required(CONF_API_KEY): cv.string,
|
||||
vol.Optional(CONF_DAYS, default=DEFAULT_DAYS): cv.string,
|
||||
vol.Optional(CONF_HOST, default=DEFAULT_HOST): cv.string,
|
||||
vol.Optional(CONF_INCLUDED, default=[]): cv.ensure_list,
|
||||
vol.Optional(CONF_MONITORED_CONDITIONS, default=['movies']):
|
||||
vol.All(cv.ensure_list, [vol.In(list(SENSOR_TYPES))]),
|
||||
vol.Optional(CONF_PORT, default=DEFAULT_PORT): cv.port,
|
||||
vol.Optional(CONF_SSL, default=False): cv.boolean,
|
||||
vol.Optional(CONF_UNIT, default=DEFAULT_UNIT): vol.In(BYTE_SIZES),
|
||||
vol.Optional(CONF_URLBASE, default=DEFAULT_URLBASE): cv.string,
|
||||
})
|
||||
|
||||
|
||||
def setup_platform(hass, config, add_entities, discovery_info=None):
|
||||
"""Set up the Radarr platform."""
|
||||
conditions = config.get(CONF_MONITORED_CONDITIONS)
|
||||
add_entities(
|
||||
[RadarrSensor(hass, config, sensor) for sensor in conditions], True)
|
||||
|
||||
|
||||
class RadarrSensor(Entity):
|
||||
"""Implementation of the Radarr sensor."""
|
||||
|
||||
def __init__(self, hass, conf, sensor_type):
|
||||
"""Create Radarr entity."""
|
||||
from pytz import timezone
|
||||
self.conf = conf
|
||||
self.host = conf.get(CONF_HOST)
|
||||
self.port = conf.get(CONF_PORT)
|
||||
self.urlbase = conf.get(CONF_URLBASE)
|
||||
if self.urlbase:
|
||||
self.urlbase = '{}/'.format(self.urlbase.strip('/'))
|
||||
self.apikey = conf.get(CONF_API_KEY)
|
||||
self.included = conf.get(CONF_INCLUDED)
|
||||
self.days = int(conf.get(CONF_DAYS))
|
||||
self.ssl = 's' if conf.get(CONF_SSL) else ''
|
||||
self._state = None
|
||||
self.data = []
|
||||
self._tz = timezone(str(hass.config.time_zone))
|
||||
self.type = sensor_type
|
||||
self._name = SENSOR_TYPES[self.type][0]
|
||||
if self.type == 'diskspace':
|
||||
self._unit = conf.get(CONF_UNIT)
|
||||
else:
|
||||
self._unit = SENSOR_TYPES[self.type][1]
|
||||
self._icon = SENSOR_TYPES[self.type][2]
|
||||
self._available = False
|
||||
|
||||
@property
|
||||
def name(self):
|
||||
"""Return the name of the sensor."""
|
||||
return '{} {}'.format('Radarr', self._name)
|
||||
|
||||
@property
|
||||
def state(self):
|
||||
"""Return sensor state."""
|
||||
return self._state
|
||||
|
||||
@property
|
||||
def available(self):
|
||||
"""Return sensor availability."""
|
||||
return self._available
|
||||
|
||||
@property
|
||||
def unit_of_measurement(self):
|
||||
"""Return the unit of the sensor."""
|
||||
return self._unit
|
||||
|
||||
@property
|
||||
def device_state_attributes(self):
|
||||
"""Return the state attributes of the sensor."""
|
||||
attributes = {}
|
||||
if self.type == 'upcoming':
|
||||
for movie in self.data:
|
||||
attributes[to_key(movie)] = get_release_date(movie)
|
||||
elif self.type == 'commands':
|
||||
for command in self.data:
|
||||
attributes[command['name']] = command['state']
|
||||
elif self.type == 'diskspace':
|
||||
for data in self.data:
|
||||
free_space = to_unit(data['freeSpace'], self._unit)
|
||||
total_space = to_unit(data['totalSpace'], self._unit)
|
||||
percentage_used = (0 if total_space == 0
|
||||
else free_space / total_space * 100)
|
||||
attributes[data['path']] = '{:.2f}/{:.2f}{} ({:.2f}%)'.format(
|
||||
free_space, total_space, self._unit, percentage_used)
|
||||
elif self.type == 'movies':
|
||||
for movie in self.data:
|
||||
attributes[to_key(movie)] = movie['downloaded']
|
||||
elif self.type == 'status':
|
||||
attributes = self.data
|
||||
|
||||
return attributes
|
||||
|
||||
@property
|
||||
def icon(self):
|
||||
"""Return the icon of the sensor."""
|
||||
return self._icon
|
||||
|
||||
def update(self):
|
||||
"""Update the data for the sensor."""
|
||||
start = get_date(self._tz)
|
||||
end = get_date(self._tz, self.days)
|
||||
try:
|
||||
res = requests.get(
|
||||
ENDPOINTS[self.type].format(
|
||||
self.ssl, self.host, self.port, self.urlbase, start, end),
|
||||
headers={'X-Api-Key': self.apikey}, timeout=10)
|
||||
except OSError:
|
||||
_LOGGER.warning("Host %s is not available", self.host)
|
||||
self._available = False
|
||||
self._state = None
|
||||
return
|
||||
|
||||
if res.status_code == 200:
|
||||
if self.type in ['upcoming', 'movies', 'commands']:
|
||||
self.data = res.json()
|
||||
self._state = len(self.data)
|
||||
elif self.type == 'diskspace':
|
||||
# If included paths are not provided, use all data
|
||||
if self.included == []:
|
||||
self.data = res.json()
|
||||
else:
|
||||
# Filter to only show lists that are included
|
||||
self.data = list(
|
||||
filter(
|
||||
lambda x: x['path'] in self.included,
|
||||
res.json()
|
||||
)
|
||||
)
|
||||
self._state = '{:.2f}'.format(
|
||||
to_unit(
|
||||
sum([data['freeSpace'] for data in self.data]),
|
||||
self._unit
|
||||
)
|
||||
)
|
||||
elif self.type == 'status':
|
||||
self.data = res.json()
|
||||
self._state = self.data['version']
|
||||
self._available = True
|
||||
|
||||
|
||||
def get_date(zone, offset=0):
|
||||
"""Get date based on timezone and offset of days."""
|
||||
day = 60 * 60 * 24
|
||||
return datetime.date(
|
||||
datetime.fromtimestamp(time.time() + day*offset, tz=zone)
|
||||
)
|
||||
|
||||
|
||||
def get_release_date(data):
|
||||
"""Get release date."""
|
||||
date = data.get('physicalRelease')
|
||||
if not date:
|
||||
date = data.get('inCinemas')
|
||||
return date
|
||||
|
||||
|
||||
def to_key(data):
|
||||
"""Get key."""
|
||||
return '{} ({})'.format(data['title'], data['year'])
|
||||
|
||||
|
||||
def to_unit(value, unit):
|
||||
"""Convert bytes to give unit."""
|
||||
return value / 1024**BYTE_SIZES.index(unit)
|
||||
Loading…
Add table
Add a link
Reference in a new issue