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

Fixes UPS MyChoice exception (#9587)

Browse Source 
* Fixes UPS MyChoice exception

* Added unit of measurement

* Collaborator-requested changes
This commit is contained in:
Aaron Bach 2017-09-27 11:44:32 -06:00 committed by Paulus Schoutsen
parent 312de6b3a3
commit d499c18e63
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/sensor/ups.py
View File

@ -76,17 +76,26 @@ class UPSSensor(Entity):
"""Return the state of the sensor.""" """Return the state of the sensor."""
return self._state return self._state
@property
def unit_of_measurement(self):
"""Return the unit of measurement of this entity, if any."""
return 'packages'
def _update(self): def _update(self):
"""Update device state.""" """Update device state."""
import upsmychoice import upsmychoice
status_counts = defaultdict(int) status_counts = defaultdict(int)
for package in upsmychoice.get_packages(self._session): try:
status = slugify(package['status']) for package in upsmychoice.get_packages(self._session):
skip = status == STATUS_DELIVERED and \ status = slugify(package['status'])
parse_date(package['delivery_date']) < now().date() skip = status == STATUS_DELIVERED and \
if skip: parse_date(package['delivery_date']) < now().date()
continue if skip:
status_counts[status] += 1 continue
status_counts[status] += 1
except upsmychoice.UPSError:
_LOGGER.error('Could not connect to UPS My Choice account')
self._attributes = { self._attributes = {
ATTR_ATTRIBUTION: upsmychoice.ATTRIBUTION ATTR_ATTRIBUTION: upsmychoice.ATTRIBUTION
} }