bitlair/wifi-mqtt
8
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make validation less crashy

Browse source
This commit is contained in:
polyfloyd 2020-05-07 14:51:55 +02:00
parent 5e355569bb
commit de3275ca28
1 changed files with 6 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

13
wifi-mqtt.py
View file

@ -17,13 +17,12 @@ def read_ap(filepath):
regexp_mac = re.compile('^(:?[0-9a-f]{2}:){5}[0-9a-f]{2}$')
associations = json.loads(subprocess.check_output([ filepath ]).decode('utf8'))
assert type(associations) is list, 'The value returned by the AP driver is not a list'
for assoc in associations:
assert type(assoc['mac']) is str
assert regexp_mac.match(assoc['mac']), 'Invalid MAC: %s' % assoc['mac']
assert type(assoc['ssid']) is str and len(assoc['ssid']) > 1, 'Invalid SSID: %s' % assoc['ssid']
if 'signal' in assoc:
assert type(assoc['signal']) is int
return associations
def valid(assoc):
return type(assoc['mac']) is str \
and regexp_mac.match(assoc['mac']) \
and type(assoc['ssid']) is str and len(assoc['ssid']) > 1
return [ assoc for assoc in associations if valid(assoc)]
class Activity(object):