add query_str to path

Signed-off-by: LucaNicosia <[email protected]>

Author: LucaNicosia

suzieq/cli/sqcmds/PathCmd.py

@@ -25,6 +25,7 @@ def __init__(
         src: str = "",
         dest: str = "",
         vrf: str = "",
+        query_str: str = ' ',
     ) -> None:
         super().__init__(
             engine=engine,
@@ -34,6 +35,7 @@ def __init__(
             view=view,
             namespace=namespace,
             columns=columns,
+            query_str=query_str,
             format=format,
             sqobj=PathObj
         )

suzieq/engines/pandas/path.py

@@ -752,6 +752,7 @@ def get(self, **kwargs) -> pd.DataFrame:
         src = kwargs.get("src", None)
         dest = kwargs.get("dest", None)
         dvrf = kwargs.get("vrf", "")
+        query_str = kwargs.pop('query_str', "")
 
         if not src or not dest:
             raise AttributeError("Must specify trace source and dest")
@@ -1120,7 +1121,9 @@ def get(self, **kwargs) -> pd.DataFrame:
             # This occurs when a path traversal terminates due to an error such
             # as loop detected
             final_paths = paths
-        return self._path_cons_result(final_paths)
+        return self._handle_user_query_str(
+            self._path_cons_result(final_paths), query_str)\
+            .reset_index(drop=True)
 
     def _path_cons_result(self, paths):
         df_plist = []