PyPI - e6data-python-connector - Versions diffs - 2.3.12rc2__tar.gz → 2.3.12rc11__tar.gz - Mend

@@ -388,7 +388,6 @@ class Connection(object):
         self._auto_resume = auto_resume
         self._grpc_options = grpc_options
         if self._grpc_options is None:
             self._grpc_options = dict()
         self.grpc_prepare_timeout = self._grpc_options.get('grpc_prepare_timeout') or 10 * 60  # 10 minutes
@@ -398,15 +397,7 @@ class Connection(object):
             The default maximum time on client side to wait for the cluster to resume is 5 minutes.
             """
             self.grpc_auto_resume_timeout_seconds = self._grpc_options.pop('grpc_auto_resume_timeout_seconds')
-        self.timeout = 500
-        if 'timeout' in self._grpc_options:
-            """
-            The default maximum time on client side to wait for the cluster to resume is 5 minutes.
-            """
-            self.timeout = self._grpc_options.pop('timeout')
         # Store debug flag and register with debug connections
         self._debug = debug
         if self._debug:
@@ -862,6 +853,56 @@ class Connection(object):
         )
         return dry_run_response.dryrunValue
+    def get_logical_plan(self, query, catalog=None, schema=None):
+        """
+        Gets the logical execution plan for a query without executing it.
+        This method performs syntax, parsing, and plan compilation, then returns the logical plan.
+        Args:
+            query (str): The SQL query to get the plan for.
+            catalog (str, optional): Catalog name. Uses connection's catalog if not provided.
+            schema (str, optional): Schema/database name. Uses connection's database if not provided.
+        Returns:
+            dict: A dictionary with:
+                - 'success' (bool): True if plan generation succeeded, False otherwise
+                - 'plan' (str, optional): The logical plan string if successful
+                - 'error' (str, optional): Error message if plan generation failed
+        Example:
+            >>> result = conn.get_logical_plan("SELECT * FROM table")
+            >>> if result['success']:
+            ...     print(f"Plan: {result['plan']}")
+            >>> else:
+            ...     print(f"Error: {result['error']}")
+        """
+        catalog_to_use = catalog if catalog is not None else self.catalog_name
+        schema_to_use = schema if schema is not None else self.database
+        try:
+            dry_run_request = e6x_engine_pb2.DryRunRequestV2(
+                sessionId=self.get_session_id,
+                schema=schema_to_use,
+                catalog=catalog_to_use,
+                queryString=query
+            )
+            dry_run_response = self._client.dryRunV2(
+                dry_run_request,
+                metadata=_get_grpc_header(cluster=self.cluster_name, strategy=_get_active_strategy())
+            )
+            return {
+                'success': True,
+                'plan': dry_run_response.dryrunValue
+            }
+        except Exception as e:
+            return {
+                'success': False,
+                'error': str(e)
+            }
     def get_tables(self, catalog, database):
         """
         Retrieves the list of tables from the specified catalog and database.
@@ -1343,8 +1384,7 @@ class Cursor(DBAPICursor):
             client = self.connection.client
             execute_response = client.executeStatementV2(
                 execute_statement_request,
-                metadata=self.metadata,
-                timeout=self.connection.timeout
+                metadata=self.metadata
             )
             # Check for new strategy in execute response
@@ -1453,14 +1493,13 @@ class Cursor(DBAPICursor):
         get_next_result_batch_request = e6x_engine_pb2.GetNextResultBatchRequest(
             engineIP=self._engine_ip,
             sessionId=self.connection.get_session_id,
-            queryId=self._query_id,
+            queryId=self._query_id
         )
         # Get fresh client after session access (may have been invalidated)
         client = self.connection.client
         get_next_result_batch_response = client.getNextResultBatch(
             get_next_result_batch_request,
-            metadata=self.metadata,
-            timeout=self.connection.timeout
+            metadata=self.metadata
         )
         # Check for new strategy in batch response

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: e6data-python-connector
-Version: 2.3.12rc2
+Version: 2.3.12rc11
 Summary: Client for the e6data distributed SQL Engine.
 Home-page: https://github.com/e6x-labs/e6data-python-connector
 Author: e6data, Inc.
@@ -39,7 +39,7 @@ Dynamic: summary
 # e6data Python Connector
-![version](https://img.shields.io/badge/version-2.3.12rc2-blue.svg)
+![version](https://img.shields.io/badge/version-2.3.12rc11-blue.svg)
 ## Introduction

@@ -1,6 +1,6 @@
 # e6data Python Connector
-![version](https://img.shields.io/badge/version-2.3.12rc2-blue.svg)
+![version](https://img.shields.io/badge/version-2.3.12rc11-blue.svg)
 ## Introduction

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: e6data-python-connector
-Version: 2.3.12rc2
+Version: 2.3.12rc11
 Summary: Client for the e6data distributed SQL Engine.
 Home-page: https://github.com/e6x-labs/e6data-python-connector
 Author: e6data, Inc.
@@ -39,7 +39,7 @@ Dynamic: summary
 # e6data Python Connector
-![version](https://img.shields.io/badge/version-2.3.12rc2-blue.svg)
+![version](https://img.shields.io/badge/version-2.3.12rc11-blue.svg)
 ## Introduction

@@ -12,7 +12,7 @@
 import setuptools
-VERSION = (2, 3, 12, 'rc2',)
+VERSION = (2, 3, 12,'rc11')
 def get_long_desc():

e6data-python-connector 2.3.12rc2__tar.gz → 2.3.12rc11__tar.gz

e6data-python-connector 2.3.12rc2tar.gz → 2.3.12rc11tar.gz