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

Scrape HTML attributes that are not key/val pairs (#58247)

Browse source
This commit is contained in:
MizterB 2022-01-21 05:17:48 -05:00 • committed by GitHub
parent 5954ca2b1f
commit 604a900658
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -160,14 +160,23 @@ class ScrapeSensor(SensorEntity):
raw_data = BeautifulSoup(self.rest.data, "html.parser")
_LOGGER.debug(raw_data)
if self._attr is not None:
value = raw_data.select(self._select)[self._index][self._attr]
else:
tag = raw_data.select(self._select)[self._index]
if tag.name in ("style", "script", "template"):
value = tag.string
try:
if self._attr is not None:
value = raw_data.select(self._select)[self._index][self._attr]
else:
value = tag.text
tag = raw_data.select(self._select)[self._index]
if tag.name in ("style", "script", "template"):
value = tag.string
else:
value = tag.text
except IndexError:
_LOGGER.warning("Index '%s' not found in %s", self._attr, self.entity_id)
value = None
except KeyError:
_LOGGER.warning(
"Attribute '%s' not found in %s", self._attr, self.entity_id
)
value = None
_LOGGER.debug(value)
return value