fixed style errors

This commit is contained in:
theolind 2015-03-07 15:49:58 +01:00
parent b42b680ef5
commit 828c78cb1f

View File

@ -12,7 +12,7 @@ from homeassistant.const import (
import psutil
import logging
sensor_types = {
SENSOR_TYPES = {
'disk_use_percent': ['Disk Use', '%'],
'disk_use': ['Disk Use', 'GiB'],
'disk_free': ['Disk Free', 'GiB'],
@ -24,31 +24,32 @@ sensor_types = {
_LOGGER = logging.getLogger(__name__)
# pylint: disable=unused-argument
def setup_platform(hass, config, add_devices, discovery_info=None):
""" Sets up the sensors """
devices = []
dev = []
for resource in config['resources']:
if 'arg' not in resource:
resource['arg'] = ''
if resource['type'] not in sensor_types:
if resource['type'] not in SENSOR_TYPES:
_LOGGER.error('Sensor type: "%s" does not exist', resource['type'])
else:
devices.append(SystemMonitorSensor(resource['type'], resource['arg']))
dev.append(SystemMonitorSensor(resource['type'], resource['arg']))
add_devices(devices)
add_devices(dev)
class SystemMonitorSensor(Device):
""" A system monitor sensor """
def __init__(self, type, argument=''):
self._name = sensor_types[type][0] + ' ' + argument
self._name = SENSOR_TYPES[type][0] + ' ' + argument
self.argument = argument
self.type = type
self._state = None
self.unit_of_measurement = sensor_types[type][1]
self.unit_of_measurement = SENSOR_TYPES[type][1]
self.update()
@property