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

Make owntracks.py pylint clean (#1808)

Browse Source 
* refactor validation logic into validate_payload function
* add debugging messages to validation
This commit is contained in:
Alexander Fortin 2016-04-12 07:02:47 +02:00 committed by Paulus Schoutsen
parent 4a16d8064d
commit 24569e6169
1 changed files with 39 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
homeassistant/components/device_tracker/owntracks.py
View File

@ -34,21 +34,33 @@ def setup_scanner(hass, config, see):
"""Setup an OwnTracks tracker.""" """Setup an OwnTracks tracker."""
max_gps_accuracy = config.get(CONF_MAX_GPS_ACCURACY) max_gps_accuracy = config.get(CONF_MAX_GPS_ACCURACY)
def owntracks_location_update(topic, payload, qos): def validate_payload(payload, data_type):
"""MQTT message received.""" """Validate OwnTracks payload."""
# Docs on available data:
# http://owntracks.org/booklet/tech/json/#_typelocation
try: try:
data = json.loads(payload) data = json.loads(payload)
except ValueError: except ValueError:
# If invalid JSON # If invalid JSON
_LOGGER.error( _LOGGER.error('Unable to parse payload as JSON: %s', payload)
'Unable to parse payload as JSON: %s', payload) return None
return if not isinstance(data, dict) or data.get('_type') != data_type:
_LOGGER.debug('Skipping %s update for following data '
'because of missing or malformatted data: %s',
data_type, data)
return None
if max_gps_accuracy is not None and \
convert(data.get('acc'), float, 0.0) > max_gps_accuracy:
_LOGGER.debug('Skipping %s update because expected GPS '
'accuracy %s is not met: %s',
data_type, max_gps_accuracy, data)
return None
return data
if (not isinstance(data, dict) or data.get('_type') != 'location') or ( def owntracks_location_update(topic, payload, qos):
max_gps_accuracy is not None and """MQTT message received."""
convert(data.get('acc'), float, 0.0) > max_gps_accuracy): # Docs on available data:
# http://owntracks.org/booklet/tech/json/#_typelocation
data = validate_payload(payload, 'location')
if not data:
return return
dev_id, kwargs = _parse_see_args(topic, data) dev_id, kwargs = _parse_see_args(topic, data)
@ -65,24 +77,16 @@ def setup_scanner(hass, config, see):
see_beacons(dev_id, kwargs) see_beacons(dev_id, kwargs)
def owntracks_event_update(topic, payload, qos): def owntracks_event_update(topic, payload, qos):
# pylint: disable=too-many-branches, too-many-statements
"""MQTT event (geofences) received.""" """MQTT event (geofences) received."""
# Docs on available data: # Docs on available data:
# http://owntracks.org/booklet/tech/json/#_typetransition # http://owntracks.org/booklet/tech/json/#_typetransition
try: data = validate_payload(payload, 'transition')
data = json.loads(payload) if not data:
except ValueError:
# If invalid JSON
_LOGGER.error(
'Unable to parse payload as JSON: %s', payload)
return
if not isinstance(data, dict) or data.get('_type') != 'transition':
return return
if data.get('desc') is None: if data.get('desc') is None:
_LOGGER.error( _LOGGER.error(
"Location missing from `enter/exit` message - " "Location missing from `Entering/Leaving` message - "
"please turn `Share` on in OwnTracks app") "please turn `Share` on in OwnTracks app")
return return
# OwnTracks uses - at the start of a beacon zone # OwnTracks uses - at the start of a beacon zone
@ -93,16 +97,16 @@ def setup_scanner(hass, config, see):
dev_id, kwargs = _parse_see_args(topic, data) dev_id, kwargs = _parse_see_args(topic, data)
if data['event'] == 'enter': def enter_event():
"""Execute enter event."""
zone = hass.states.get("zone.{}".format(location)) zone = hass.states.get("zone.{}".format(location))
with LOCK: with LOCK:
if zone is None: if zone is None and data['t'] == 'b':
if data['t'] == 'b': # Not a HA zone, and a beacon so assume mobile
# Not a HA zone, and a beacon so assume mobile beacons = MOBILE_BEACONS_ACTIVE[dev_id]
beacons = MOBILE_BEACONS_ACTIVE[dev_id] if location not in beacons:
if location not in beacons: beacons.append(location)
beacons.append(location) _LOGGER.info("Added beacon %s", location)
_LOGGER.info("Added beacon %s", location)
else: else:
# Normal region # Normal region
regions = REGIONS_ENTERED[dev_id] regions = REGIONS_ENTERED[dev_id]
@ -114,7 +118,8 @@ def setup_scanner(hass, config, see):
see(**kwargs) see(**kwargs)
see_beacons(dev_id, kwargs) see_beacons(dev_id, kwargs)
elif data['event'] == 'leave': def leave_event():
"""Execute leave event."""
with LOCK: with LOCK:
regions = REGIONS_ENTERED[dev_id] regions = REGIONS_ENTERED[dev_id]
if location in regions: if location in regions:
@ -146,6 +151,10 @@ def setup_scanner(hass, config, see):
beacons.remove(location) beacons.remove(location)
_LOGGER.info("Remove beacon %s", location) _LOGGER.info("Remove beacon %s", location)
if data['event'] == 'enter':
enter_event()
elif data['event'] == 'leave':
leave_event()
else: else:
_LOGGER.error( _LOGGER.error(
'Misformatted mqtt msgs, _type=transition, event=%s', 'Misformatted mqtt msgs, _type=transition, event=%s',