pandas/devconfig: apply query_str when section is not specified

Signed-off-by: LucaNicosia <[email protected]>
netenglabs · May 27, 2024 · 4aa1c81 · 4aa1c81
1 parent be27341
commit 4aa1c81
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/suzieq/engines/pandas/devconfig.py b/suzieq/engines/pandas/devconfig.py
@@ -21,7 +21,12 @@ def get(self, **kwargs):
         query_str = kwargs.pop('query_str', '')
 
         df = super().get(columns=columns, **kwargs)
-        if df.empty or 'error' in df.columns or not section:
+        if df.empty or 'error' in df.columns:
+            return df
+
+        if not section:
+            if query_str:
+                df = df.query(query_str).reset_index(drop=True)
             return df
 
         devdf = self._get_table_sqobj('device') \