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

{e6data_python_connector-2.3.12rc1/e6data_python_connector.egg-info → e6data_python_connector-2.3.12rc11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: e6data-python-connector
-Version: 2.3.12rc1
+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.12rc1-blue.svg)
+![version](https://img.shields.io/badge/version-2.3.12rc11-blue.svg)
 
 ## Introduction
 

{e6data_python_connector-2.3.12rc1 → e6data_python_connector-2.3.12rc11}/README.md

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

{e6data_python_connector-2.3.12rc1 → e6data_python_connector-2.3.12rc11}/e6data_python_connector/e6data_grpc.py

@@ -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.
@@ -1452,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

{e6data_python_connector-2.3.12rc1 → e6data_python_connector-2.3.12rc11/e6data_python_connector.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: e6data-python-connector
-Version: 2.3.12rc1
+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.12rc1-blue.svg)
+![version](https://img.shields.io/badge/version-2.3.12rc11-blue.svg)
 
 ## Introduction
 

{e6data_python_connector-2.3.12rc1 → e6data_python_connector-2.3.12rc11}/setup.py

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