jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 13:17:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow multiple symbols (sensor.yahoo_finance) (#4126)

Browse Source 
* Allow multiple symbols

* Update test
This commit is contained in:
Fabian Affolter 2016-10-31 13:31:09 +01:00 committed by GitHub
parent 06de7053ce
commit b4899ec469
2 changed files with 30 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
homeassistant/components/sensor/yahoo_finance.py
View File

@ -10,7 +10,7 @@ from datetime import timedelta
import voluptuous as vol import voluptuous as vol
from homeassistant.components.sensor import PLATFORM_SCHEMA from homeassistant.components.sensor import PLATFORM_SCHEMA
from homeassistant.const import (CONF_NAME, ATTR_ATTRIBUTION) from homeassistant.const import ATTR_ATTRIBUTION
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 import homeassistant.helpers.config_validation as cv
@ -24,36 +24,44 @@ ATTR_OPEN = 'open'
ATTR_PREV_CLOSE = 'prev_close' ATTR_PREV_CLOSE = 'prev_close'
CONF_ATTRIBUTION = "Stock market information provided by Yahoo! Inc." CONF_ATTRIBUTION = "Stock market information provided by Yahoo! Inc."
CONF_SYMBOL = 'symbol' CONF_SYMBOLS = 'symbols'
DEFAULT_NAME = 'Yahoo Stock' DEFAULT_NAME = 'Yahoo Stock'
DEFAULT_SYMBOL = 'YHOO' DEFAULT_SYMBOL = 'YHOO'
ICON = 'mdi:currency-usd' ICON = 'mdi:currency-usd'
MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=1) MIN_TIME_BETWEEN_UPDATES = timedelta(minutes=5)
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Optional(CONF_SYMBOL, default=DEFAULT_SYMBOL): cv.string, vol.Optional(CONF_SYMBOLS, default=[DEFAULT_SYMBOL]):
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string, vol.All(cv.ensure_list, [cv.string]),
}) })
def setup_platform(hass, config, add_devices, discovery_info=None): def setup_platform(hass, config, add_devices, discovery_info=None):
"""Set up the Yahoo Finance sensor.""" """Set up the Yahoo Finance sensor."""
name = config.get(CONF_NAME) from yahoo_finance import Share
symbol = config.get(CONF_SYMBOL)
data = YahooFinanceData(name, symbol) symbols = config.get(CONF_SYMBOLS)
add_devices([YahooFinanceSensor(name, data, symbol)])
dev = []
for symbol in symbols:
if Share(symbol).get_price() is None:
_LOGGER.warning("Symbol %s unknown", symbol)
break
data = YahooFinanceData(symbol)
dev.append(YahooFinanceSensor(data, symbol))
add_devices(dev)
class YahooFinanceSensor(Entity): class YahooFinanceSensor(Entity):
"""Representation of a Yahoo Finance sensor.""" """Representation of a Yahoo Finance sensor."""
def __init__(self, name, data, symbol): def __init__(self, data, symbol):
"""Initialize the sensor.""" """Initialize the sensor."""
self._name = name self._name = symbol
self.data = data self.data = data
self._symbol = symbol self._symbol = symbol
self._state = None self._state = None
@ -101,17 +109,16 @@ class YahooFinanceSensor(Entity):
class YahooFinanceData(object): class YahooFinanceData(object):
"""Get data from Yahoo Finance.""" """Get data from Yahoo Finance."""
def __init__(self, name, symbol): def __init__(self, symbol):
"""Initialize the data object.""" """Initialize the data object."""
from yahoo_finance import Share from yahoo_finance import Share
self._name = name
self._symbol = symbol self._symbol = symbol
self.state = None self.state = None
self.price_change = None self.price_change = None
self.price_open = None self.price_open = None
self.prev_close = None self.prev_close = None
self.stock = Share(symbol) self.stock = Share(self._symbol)
@Throttle(MIN_TIME_BETWEEN_UPDATES) @Throttle(MIN_TIME_BETWEEN_UPDATES)
def update(self): def update(self):

15
tests/components/sensor/test_yahoo_finance.py
View File

@ -11,10 +11,13 @@ from tests.common import (
VALID_CONFIG = { VALID_CONFIG = {
'platform': 'yahoo_finance', 'platform': 'yahoo_finance',
'symbol': 'YHOO' 'symbols': [
'YHOO',
]
} }
# pylint: disable=invalid-name
class TestYahooFinanceSetup(unittest.TestCase): class TestYahooFinanceSetup(unittest.TestCase):
"""Test the Yahoo Finance platform.""" """Test the Yahoo Finance platform."""
@ -29,13 +32,13 @@ class TestYahooFinanceSetup(unittest.TestCase):
@patch('yahoo_finance.Base._request', @patch('yahoo_finance.Base._request',
return_value=json.loads(load_fixture('yahoo_finance.json'))) return_value=json.loads(load_fixture('yahoo_finance.json')))
def test_default_setup(self, m): # pylint: disable=invalid-name def test_default_setup(self, mock_request):
"""Test the default setup.""" """Test the default setup."""
with assert_setup_component(1, sensor.DOMAIN): with assert_setup_component(1, sensor.DOMAIN):
assert setup_component(self.hass, sensor.DOMAIN, { assert setup_component(self.hass, sensor.DOMAIN, {
'sensor': VALID_CONFIG}) 'sensor': VALID_CONFIG})
state = self.hass.states.get('sensor.yahoo_stock') state = self.hass.states.get('sensor.yhoo')
self.assertEqual("41.69", state.attributes.get('open')) self.assertEqual('41.69', state.attributes.get('open'))
self.assertEqual("41.79", state.attributes.get('prev_close')) self.assertEqual('41.79', state.attributes.get('prev_close'))
self.assertEqual("YHOO", state.attributes.get('unit_of_measurement')) self.assertEqual('YHOO', state.attributes.get('unit_of_measurement'))