From 8ff674f1f2867b4e052ee2de49cc0466dffcc733 Mon Sep 17 00:00:00 2001
From: Jed Reynolds <jreynolds@candelatech.com>
Date: Wed, 20 Oct 2021 17:38:58 -0700
Subject: [PATCH] jbr_monitor_bssids.py: WIP attempting to recreate load
 scenario logic with lanforge_api

---
 py-scripts/sandbox/jbr_monitor_bssids.py | 33 +++++++++++++++++++-----
 1 file changed, 27 insertions(+), 6 deletions(-)

diff --git a/py-scripts/sandbox/jbr_monitor_bssids.py b/py-scripts/sandbox/jbr_monitor_bssids.py
index 5a1d8b19..e65d0bdf 100755
--- a/py-scripts/sandbox/jbr_monitor_bssids.py
+++ b/py-scripts/sandbox/jbr_monitor_bssids.py
@@ -44,7 +44,8 @@ if sys.version_info[0] != 3:
 import os
 import importlib
 import argparse
-
+from http.client import HTTPResponse
+from typing import Optional
 
 path_hunks = os.path.abspath(__file__).split('/')
 while( path_hunks[-1] != 'lanforge-scripts'):
@@ -56,6 +57,7 @@ lanforge_api = importlib.import_module("lanforge_client.lanforge_api")
 LFSession = lanforge_api.LFSession
 LFJsonCommand = lanforge_api.LFJsonCommand
 LFJsonQuery = lanforge_api.LFJsonQuery
+# from scenario import LoadScenario
 
 import pprint
 # print(pprint.pformat(("ospath", sys.path)))
@@ -123,7 +125,7 @@ class BssidMonitor(Realm):
         self.report_file: str = report_file
         self.output_format: str = output_format
         self.port_mgr_cols: list[str] = port_mgr_cols
-        self.layer3_cols: list[str] = layer3_cols,
+        self.layer3_cols: tuple[Optional[list[str]]] = layer3_cols,
         self.monitor_interval_sec: int = monitor_interval_sec
 
         if not bssid_list:
@@ -139,13 +141,32 @@ class BssidMonitor(Realm):
                             },
                         debug=True,
                         die_on_error=False);'''
-        self.lf_session = lanforge_api.LFSession(lfclient_url="http://{host}:{port}/".format(host=host, port=port))
-        self.lf_command = self.lf_session.get_command()
-        self.lf_query = self.lf_session.get_query()
+        self.lf_session: LFSession =\
+            lanforge_api.LFSession(lfclient_url="http://{host}:{port}/".format(host=host, port=port),
+                                   debug=debug,
+                                   connection_timeout_sec=2,
+                                   stream_errors=True,
+                                   stream_warnings=True,
+                                   require_session=True,
+                                   exit_on_error=True)
+        self.lf_command : LFJsonCommand = self.lf_session.get_command()
+        self.lf_query : LFJsonQuery = self.lf_session.get_query()
 
     def build(self):
+        err_warn_list = []
+        # query for the last response
+        event_response = self.lf_query.events_last_events(event_count=1,
+                                                          debug=self.debug,
+                                                          errors_warnings=err_warn_list)
+        pprint(("event_response", dir(event_response)))
         # load a database
-
+        response: HTTPResponse = self.lf_command.post_load(name="BLANK",
+                                                           action="overwrite",
+                                                           clean_dut="NA",
+                                                           clean_chambers="NA",
+                                                           debug=self.debug_)
+        if not response:
+            raise ConnectionError("lf_command::post_load returned no response")
 
         # create a series of stations
         for bssid in self.bssid_list: