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

Black

Browse source
This commit is contained in:
Paulus Schoutsen 2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download patch file Download diff file Expand all files Collapse all files

22
homeassistant/components/uscis/sensor.py
View file

@ -15,21 +15,22 @@ _LOGGER = logging.getLogger(__name__)
DEFAULT_NAME = "USCIS"
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Optional(CONF_FRIENDLY_NAME, default=DEFAULT_NAME): cv.string,
vol.Required('case_id'): cv.string,
})
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
{
vol.Optional(CONF_FRIENDLY_NAME, default=DEFAULT_NAME): cv.string,
vol.Required("case_id"): cv.string,
}
)
def setup_platform(hass, config, add_entities, discovery_info=None):
"""Set up the platform in HASS and Case Information."""
uscis = UscisSensor(config['case_id'], config[CONF_FRIENDLY_NAME])
uscis = UscisSensor(config["case_id"], config[CONF_FRIENDLY_NAME])
uscis.update()
if uscis.valid_case_id:
add_entities([uscis])
else:
_LOGGER.error("Setup USCIS Sensor Fail"
" check if your Case ID is Valid")
_LOGGER.error("Setup USCIS Sensor Fail" " check if your Case ID is Valid")
class UscisSensor(Entity):
@ -67,12 +68,11 @@ class UscisSensor(Entity):
def update(self):
"""Fetch data from the USCIS website and update state attributes."""
import uscisstatus
try:
status = uscisstatus.get_case_status(self._case_id)
self._attributes = {
self.CURRENT_STATUS: status['status']
}
self._state = status['date']
self._attributes = {self.CURRENT_STATUS: status["status"]}
self._state = status["date"]
self.valid_case_id = True
except ValueError: