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

Add optional attribute option to scrape sensor (#10001)

Browse source 
* Add optional attribute option to scrape sensor

* Rename attribute variable to attr
This commit is contained in:
Martin Eberhardt 2017-10-21 16:03:29 +02:00 • committed by Fabian Affolter
parent ce1a2cc2a6
commit 222cc4c393
1 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
homeassistant/components/sensor/scrape.py
View file

@ -21,6 +21,7 @@ REQUIREMENTS = ['beautifulsoup4==4.6.0']
_LOGGER = logging.getLogger(__name__)
CONF_SELECT = 'select'
CONF_ATTR = 'attribute'
DEFAULT_NAME = 'Web scrape'
DEFAULT_VERIFY_SSL = True
@ -28,6 +29,7 @@ DEFAULT_VERIFY_SSL = True
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_RESOURCE): cv.string,
vol.Required(CONF_SELECT): cv.string,
vol.Optional(CONF_ATTR): cv.string,
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
vol.Optional(CONF_UNIT_OF_MEASUREMENT): cv.string,
vol.Optional(CONF_VALUE_TEMPLATE): cv.template,
@ -43,6 +45,7 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
payload = auth = headers = None
verify_ssl = config.get(CONF_VERIFY_SSL)
select = config.get(CONF_SELECT)
attr = config.get(CONF_ATTR)
unit = config.get(CONF_UNIT_OF_MEASUREMENT)
value_template = config.get(CONF_VALUE_TEMPLATE)
if value_template is not None:
@ -56,19 +59,20 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
return False
add_devices([
ScrapeSensor(hass, rest, name, select, value_template, unit)
ScrapeSensor(hass, rest, name, select, attr, value_template, unit)
], True)
class ScrapeSensor(Entity):
"""Representation of a web scrape sensor."""
def __init__(self, hass, rest, name, select, value_template, unit):
def __init__(self, hass, rest, name, select, attr, value_template, unit):
"""Initialize a web scrape sensor."""
self.rest = rest
self._name = name
self._state = STATE_UNKNOWN
self._select = select
self._attr = attr
self._value_template = value_template
self._unit_of_measurement = unit
@ -95,6 +99,9 @@ class ScrapeSensor(Entity):
raw_data = BeautifulSoup(self.rest.data, 'html.parser')
_LOGGER.debug(raw_data)
if self._attr is not None:
value = raw_data.select(self._select)[0][self._attr]
else:
value = raw_data.select(self._select)[0].text
_LOGGER.debug(value)