don't look up using ssid

This commit is contained in:
Laura Klünder 2017-12-26 15:58:33 +01:00
parent 65e84b56c0
commit 6d2417c522

View file

@ -115,14 +115,14 @@ class LocatorStations:
self.stations_lookup = {} self.stations_lookup = {}
def get(self, bssid, ssid, frequency, create=False): def get(self, bssid, ssid, frequency, create=False):
station_id = self.stations_lookup.get((bssid, ssid), None) station_id = self.stations_lookup.get((bssid, None), None)
if station_id is not None: if station_id is not None:
station = self.stations[station_id] station = self.stations[station_id]
station.frequencies.add(frequency) station.frequencies.add(frequency)
elif create: elif create:
station = LocatorStation(bssid, ssid, set((frequency, ))) station = LocatorStation(bssid, ssid, set((frequency, )))
station_id = len(self.stations) station_id = len(self.stations)
self.stations_lookup[(bssid, ssid)] = station_id self.stations_lookup[(bssid, None)] = station_id
self.stations.append(station) self.stations.append(station)
return station_id return station_id