mirror of
https://github.com/Telecominfraproject/wlan-lanforge-scripts.git
synced 2025-10-31 18:58:01 +00:00
74 lines
2.4 KiB
Python
Executable File
74 lines
2.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Contains examples of using realm to query stations and get specific information from them
|
|
import sys
|
|
import os
|
|
import importlib
|
|
import argparse
|
|
|
|
if sys.version_info[0] != 3:
|
|
print("This script requires Python 3")
|
|
exit(1)
|
|
|
|
sys.path.append(os.path.join(os.path.abspath(__file__ + "../../../")))
|
|
|
|
lfcli_base = importlib.import_module("py-json.LANforge.lfcli_base")
|
|
LFCliBase = lfcli_base.LFCliBase
|
|
realm = importlib.import_module("py-json.realm")
|
|
Realm = realm.Realm
|
|
|
|
|
|
class StationsConnected(LFCliBase):
|
|
def __init__(self, lfjson_host, lfjson_port):
|
|
super().__init__(_lfjson_host=lfjson_host, _lfjson_port=lfjson_port)
|
|
self.localrealm = Realm(lfclient_host=lfjson_host, lfclient_port=lfjson_port)
|
|
self.check_connect()
|
|
fields = "_links,port,alias,ip,ap,port+type"
|
|
self.station_results = self.localrealm.find_ports_like("sta*", fields, debug_=False)
|
|
|
|
def run(self):
|
|
self.clear_test_results()
|
|
# pprint(self.station_results)
|
|
if not self.station_results 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():
|
|
parser = argparse.ArgumentParser(
|
|
prog='stations_connected.py',
|
|
formatter_class=argparse.RawTextHelpFormatter,
|
|
epilog='''\
|
|
stations_connected.py
|
|
|
|
''',
|
|
description='''\
|
|
Contains examples of using realm to query stations and get specific information from them
|
|
''')
|
|
# if args are added args=parser.parse_args() swap out next line
|
|
parser.parse_args()
|
|
|
|
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()
|