[QPROF] Updating session params (#1283)

* [QPROF] Updating session params

 Should close:
 - https://jira.verticacorp.com/jira/browse/VER-96476?filter=-1

* Adding session_params_current

session_params_current allows to get the current params.

Author: oualib

verticapy/__init__.py

@@ -37,9 +37,9 @@
 __url__: str = "https://github.com/vertica/verticapy/"
 __license__: str = "Apache License, Version 2.0"
 __version__: str = "1.0.5"
-__iteration__: int = 1
+__iteration__: int = 2
 __date__: str = "25092024"
-__last_commit__: str = "d15329cab9a2360454f4473e676068d2f793b965"
+__last_commit__: str = "845f1130772d54ecd184f71f57c92b76ed4b1628"
 __long_version__: str = f"{__version__}-{__iteration__}—{__date__}-{__last_commit__}"
 __codecov__: float = 0.84
 

verticapy/performance/vertica/qprof.py

@@ -225,6 +225,8 @@ class QueryProfiler:
         Current Transaction ID.
     statement_id: int
         Current Statement ID.
+    session_params_current: dict
+        Current Session Parameters.
     target_schema: dict
         Name of the schema used to store
         all the Vertica monitor and internal
@@ -239,6 +241,9 @@ class QueryProfiler:
     tables_dtypes: list
         Datatypes of all the loaded
         performance tables.
+    session_params: list
+        Non Default Session Parameters used
+        to run the transactions.
     overwrite: bool
         If set to ``True`` overwrites the
         existing performance tables.
@@ -1146,6 +1151,7 @@ def __init__(
                 session_control_loop = session_control_loop[1:]
 
             session_control_loop_all = []
+            session_params = []
 
             for sc in session_control_loop:
                 if sc not in ({}, ""):
@@ -1203,8 +1209,10 @@ def __init__(
                             " will be skipped."
                         )
                         warnings.warn(warning_message, Warning)
+                    session_params += [self._get_current_session_params()]
 
             self.session_control_params = session_control_loop_all
+            self.session_params = session_params
 
         if len(self.transactions) == 0 and isinstance(key_id, NoneType):
             raise ValueError("No transactions found.")
@@ -1252,8 +1260,38 @@ def __init__(
                 iterchecks=iterchecks, ignore_operators_check=ignore_operators_check
             )
 
+        # CORRECTING WRONG ATTRIBUTES
+        if not (hasattr(self, "session_params")) or not (self.session_params):
+            self.session_params = [{} for x in self.transactions]
+        try:
+            self.session_params_current = self.session_params[0]
+        except:
+            self.session_params_current = {}
+
     # Tools
 
+    @staticmethod
+    def _get_current_session_params():
+        """
+        Returns a ``dict`` of the current
+        session parameters.
+        """
+        query = """
+            SELECT 
+                parameter_name,
+                current_value 
+            FROM v_monitor.configuration_parameters 
+            WHERE current_level != 'DEFAULT';"""
+        res = _executeSQL(
+            query,
+            title="Getting transaction_id, statement_id.",
+            method="fetchall",
+        )
+        params = {}
+        for key, val in res:
+            params[key] = val
+        return params
+
     def _check_kind(self, kind: str, kind_list: list) -> str:
         """
         Checks if the parameter 'kind'
@@ -2024,6 +2062,10 @@ def set_position(self, idx: Union[int, tuple]) -> None:
                 self.request = self.requests[idx]
                 self.qduration = self.qdurations[idx]
                 self.query_success = self.query_successes[idx]
+                try:
+                    self.session_params_current = self.session_params[idx]
+                except:
+                    self.session_params_current = {}
             else:
                 raise TypeError(
                     "Wrong type for parameter 'idx'. Expecting: int or tuple."
@@ -2047,6 +2089,10 @@ def next(self) -> None:
         self.statement_id = self.transactions[idx][1]
         self.request = self.requests[idx]
         self.qduration = self.qdurations[idx]
+        try:
+            self.session_params_current = self.session_params[idx]
+        except:
+            self.session_params_current = {}
 
     def previous(self) -> None:
         """
@@ -2065,6 +2111,10 @@ def previous(self) -> None:
         self.statement_id = self.transactions[idx][1]
         self.request = self.requests[idx]
         self.qduration = self.qdurations[idx]
+        try:
+            self.session_params_current = self.session_params[idx]
+        except:
+            self.session_params_current = {}
 
     # Main Method
 

verticapy/performance/vertica/tree.py

@@ -2032,7 +2032,7 @@ def _gen_legend_annotations(self, rows: Optional[list] = None):
             if "RESEGMENT" in row_tmp and "BROADCAST" not in row_tmp:
                 all_legend[
                     "---"
-                ] = f'<tr><td BGCOLOR="{bgcolor}"><FONT COLOR="{fontcolor}">---</FONT></td><td BGCOLOR="{fillcolor}"><FONT COLOR="{fontcolor}">RESEGMENT | NO BROADCAST</FONT></td></tr>'
+                ] = f'<tr><td BGCOLOR="{bgcolor}"><FONT COLOR="{fontcolor}">---</FONT></td><td BGCOLOR="{fillcolor}"><FONT COLOR="{fontcolor}">RESEGMENT</FONT></td></tr>'
             if "HASH" in row_tmp:
                 all_legend[
                     "HASH"
@@ -2087,9 +2087,9 @@ def _gen_legend_annotations(self, rows: Optional[list] = None):
         ):
             res_trans = ""
             if idx == 0:
-                name_tmp = "Path transition"
+                name_tmp = "Path transitions"
             elif idx == 1:
-                name_tmp = "Link"
+                name_tmp = "Links"
             elif idx == 2:
                 name_tmp = "Information"
             for op in trans_list: