Files
wlan-lanforge-scripts/py-scripts/stations_connected.py

58 lines
1.9 KiB
Python
Executable File

#!/usr/bin/env python3
# Contains examples of using realm to query stations and get specific information from them
import sys
if sys.version_info[0] != 3:
print("This script requires Python 3")
exit(1)
if 'py-json' not in sys.path:
sys.path.append('../py-json')
from LANforge.lfcli_base import LFCliBase
from realm import Realm
import pprint
from pprint import pprint
class StationsConnected(LFCliBase):
def __init__(self, lfjson_host, lfjson_port):
super().__init__(_lfjson_host=lfjson_host, _lfjson_port=lfjson_port, _halt_on_error=True, _debug=False)
self.localrealm = Realm(lfclient_host=lfjson_host, lfclient_port=lfjson_port, debug=False)
self.check_connect()
def run(self):
self.clear_test_results()
fields = "_links,port,alias,ip,ap,port+type"
self.station_results = self.localrealm.find_ports_like("sta*", fields, debug_=False)
#pprint(self.station_results)
if (self.station_results is None) or (len(self.station_results) < 1):
self.get_failed_result_list()
return False
return True
def num_associated(self, bssid):
counter = 0
# print("there are %d results" % len(self.station_results))
for eid,record in self.station_results.items():
#print("-- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- ")
#pprint(eid)
#pprint(record)
if record["ap"] == bssid:
counter += 1
#print("-- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- -- ")
return counter
def main():
qstationsx = StationsConnected("localhost", 8080)
bssid = "00:0E:8E:7B:DF:9B"
if qstationsx.run():
associated_stations = qstationsx.num_associated(bssid)
print("Number of stations associated to %s: %s" % (bssid, associated_stations))
else:
print("problem querying for stations for %s" % bssid)
if __name__ == "__main__":
main()