jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-09 22:37:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add transmission sensor: number of active torrents (#9914)

Browse Source 
* Add transmission sensor: number of active torrents

* Make variable name shorter
This commit is contained in:
mclem 2017-10-17 22:45:37 +02:00 committed by Fabian Affolter
parent f5b305c980
commit 1d68777981
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
homeassistant/components/sensor/transmission.py
View File

@ -9,13 +9,13 @@ from datetime import timedelta
import voluptuous as vol import voluptuous as vol
import homeassistant.helpers.config_validation as cv
from homeassistant.components.sensor import PLATFORM_SCHEMA from homeassistant.components.sensor import PLATFORM_SCHEMA
from homeassistant.const import ( from homeassistant.const import (
CONF_HOST, CONF_PASSWORD, CONF_USERNAME, CONF_NAME, CONF_PORT, CONF_HOST, CONF_PASSWORD, CONF_USERNAME, CONF_NAME, CONF_PORT,
CONF_MONITORED_VARIABLES, STATE_UNKNOWN, STATE_IDLE) CONF_MONITORED_VARIABLES, STATE_IDLE)
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
from homeassistant.util import Throttle from homeassistant.util import Throttle
import homeassistant.helpers.config_validation as cv
REQUIREMENTS = ['transmissionrpc==0.11'] REQUIREMENTS = ['transmissionrpc==0.11']
@ -26,9 +26,10 @@ DEFAULT_NAME = 'Transmission'
DEFAULT_PORT = 9091 DEFAULT_PORT = 9091
SENSOR_TYPES = { SENSOR_TYPES = {
'active_torrents': ['Active Torrents', None],
'current_status': ['Status', None], 'current_status': ['Status', None],
'download_speed': ['Down Speed', 'MB/s'], 'download_speed': ['Down Speed', 'MB/s'],
'upload_speed': ['Up Speed', 'MB/s'] 'upload_speed': ['Up Speed', 'MB/s'],
} }
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
@ -80,7 +81,7 @@ class TransmissionSensor(Entity):
def __init__(self, sensor_type, transmission_client, client_name): def __init__(self, sensor_type, transmission_client, client_name):
"""Initialize the sensor.""" """Initialize the sensor."""
self._name = SENSOR_TYPES[sensor_type][0] self._name = SENSOR_TYPES[sensor_type][0]
self.transmission_client = transmission_client self.tm_client = transmission_client
self.type = sensor_type self.type = sensor_type
self.client_name = client_name self.client_name = client_name
self._state = None self._state = None
@ -117,9 +118,9 @@ class TransmissionSensor(Entity):
self.refresh_transmission_data() self.refresh_transmission_data()
if self.type == 'current_status': if self.type == 'current_status':
if self.transmission_client.session: if self.tm_client.session:
upload = self.transmission_client.session.uploadSpeed upload = self.tm_client.session.uploadSpeed
download = self.transmission_client.session.downloadSpeed download = self.tm_client.session.downloadSpeed
if upload > 0 and download > 0: if upload > 0 and download > 0:
self._state = 'Up/Down' self._state = 'Up/Down'
elif upload > 0 and download == 0: elif upload > 0 and download == 0:
@ -129,14 +130,16 @@ class TransmissionSensor(Entity):
else: else:
self._state = STATE_IDLE self._state = STATE_IDLE
else: else:
self._state = STATE_UNKNOWN self._state = None
if self.transmission_client.session: if self.tm_client.session:
if self.type == 'download_speed': if self.type == 'download_speed':
mb_spd = float(self.transmission_client.session.downloadSpeed) mb_spd = float(self.tm_client.session.downloadSpeed)
mb_spd = mb_spd / 1024 / 1024 mb_spd = mb_spd / 1024 / 1024
self._state = round(mb_spd, 2 if mb_spd < 0.1 else 1) self._state = round(mb_spd, 2 if mb_spd < 0.1 else 1)
elif self.type == 'upload_speed': elif self.type == 'upload_speed':
mb_spd = float(self.transmission_client.session.uploadSpeed) mb_spd = float(self.tm_client.session.uploadSpeed)
mb_spd = mb_spd / 1024 / 1024 mb_spd = mb_spd / 1024 / 1024
self._state = round(mb_spd, 2 if mb_spd < 0.1 else 1) self._state = round(mb_spd, 2 if mb_spd < 0.1 else 1)
elif self.type == 'active_torrents':
self._state = self.tm_client.session.activeTorrentCount