e6data-python-connector 2.0.2__tar.gz → 2.0.4__tar.gz

{e6data-python-connector-2.0.2/e6data_python_connector.egg-info → e6data-python-connector-2.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: e6data-python-connector
-Version: 2.0.2
+Version: 2.0.4
 Summary: Client for the e6data distributed SQL Engine.
 Home-page: https://github.com/e6x-labs/e6data-python-connector
 Author: Uniphi, Inc.
@@ -27,7 +27,7 @@ Requires-Dist: sqlalchemy
 
 # e6data Python Connector
 
-![version](https://img.shields.io/badge/version-2.0.2-blue.svg)
+![version](https://img.shields.io/badge/version-2.0.4-blue.svg)
 
 ## Introduction
 

{e6data-python-connector-2.0.2 → e6data-python-connector-2.0.4}/README.md

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

{e6data-python-connector-2.0.2 → e6data-python-connector-2.0.4}/e6data_python_connector/datainputstream.py

@@ -24,6 +24,10 @@ except ImportError:
 _logger = logging.getLogger(__name__)
 
 
+def get_null(vector: Vector, index: int):
+    return vector.nullSet[0] if vector.isConstantVector else vector.nullSet[index]
+
+
 class DataInputStream:
     def __init__(self, stream):
         self.stream = stream
@@ -203,14 +207,14 @@ def get_column_from_chunk(vector: Vector) -> list:
     try:
         if d_type == VectorType.LONG:
             for row in range(vector.size):
-                if vector.nullSet[row]:
+                if get_null(vector, row):
                     value_array.append(None)
                     continue
                 value_array.append(vector.data.int64Data.data[
                                        row] if not vector.isConstantVector else vector.data.numericConstantData.data)
         elif d_type == VectorType.DATE:
             for row in range(vector.size):
-                if vector.nullSet[row]:
+                if get_null(vector, row):
                     value_array.append(None)
                     continue
                 epoch_seconds = floor_div(vector.data.dateData.data[
@@ -221,54 +225,56 @@ def get_column_from_chunk(vector: Vector) -> list:
                 value_array.append(date.strftime("%Y-%m-%d"))
         elif d_type == VectorType.DATETIME:
             for row in range(vector.size):
-                if vector.nullSet[row]:
+                if get_null(vector, row):
                     value_array.append(None)
                     continue
-                epoch_micros = vector.data.timeData.data[row] if not vector.isConstantVector else vector.data.timeConstantData.data
+                epoch_micros = vector.data.timeData.data[
+                    row] if not vector.isConstantVector else vector.data.timeConstantData.data
                 epoch_seconds = floor_div(epoch_micros, 1000_000)
                 micros_of_the_day = floor_mod(epoch_micros, 1000_000)
-                zone_offset = pytz.timezone('UTC') if vector.zoneOffset == 'Z' else pytz.timezone(get_zone(vector.zoneOffset))
+                zone_offset = pytz.timezone('UTC') if vector.zoneOffset == 'Z' else pytz.timezone(
+                    get_zone(vector.zoneOffset))
                 date_time = datetime.fromtimestamp(epoch_seconds, zone_offset)
                 date_time = date_time + timedelta(microseconds=micros_of_the_day)
                 value_array.append(date_time.strftime("%Y-%m-%d %H:%M:%S"))
         elif d_type == VectorType.STRING or d_type == VectorType.ARRAY or d_type == VectorType.MAP or d_type == VectorType.STRUCT:
             for row in range(vector.size):
-                if vector.nullSet[row]:
+                if get_null(vector, row):
                     value_array.append(None)
                     continue
                 value_array.append(vector.data.varcharData.data[
                                        row] if not vector.isConstantVector else vector.data.varcharConstantData.data)
         elif d_type == VectorType.DOUBLE:
             for row in range(vector.size):
-                if vector.nullSet[row]:
+                if get_null(vector, row):
                     value_array.append(None)
                     continue
                 value_array.append(vector.data.float64Data.data[
                                        row] if not vector.isConstantVector else vector.data.numericDecimalConstantData.data)
         elif d_type == VectorType.BINARY:
             for row in range(vector.size):
-                if vector.nullSet[row]:
+                if get_null(vector, row):
                     value_array.append(None)
                     continue
                 value_array.append(vector.data.varcharData.data[
                                        row] if not vector.isConstantVector else vector.data.varcharConstantData.data)
         elif d_type == VectorType.FLOAT:
             for row in range(vector.size):
-                if vector.nullSet[row]:
+                if get_null(vector, row):
                     value_array.append(None)
                     continue
                 value_array.append(vector.data.float32Data.data[
                                        row] if not vector.isConstantVector else vector.data.numericDecimalConstantData.data)
         elif d_type == VectorType.BOOLEAN:
             for row in range(vector.size):
-                if vector.nullSet[row]:
+                if get_null(vector, row):
                     value_array.append(None)
                     continue
                 value_array.append(vector.data.boolData.data[
                                        row] if not vector.isConstantVector else vector.data.boolConstantData.data)
         elif d_type == VectorType.INTEGER:
             for row in range(vector.size):
-                if vector.nullSet[row]:
+                if get_null(vector, row):
                     value_array.append(None)
                     continue
                 value_array.append(vector.data.int32Data.data[

{e6data-python-connector-2.0.2 → e6data-python-connector-2.0.4}/e6data_python_connector/e6data_grpc.py

@@ -454,6 +454,7 @@ class Cursor(DBAPICursor):
         get_result_metadata_response = self.connection.client.getResultMetadata(result_meta_data_request)
         buffer = BytesIO(get_result_metadata_response.resultMetaData)
         self._rowcount, self._query_columns_description = get_query_columns_info(buffer)
+        self._is_metadata_updated = True
 
     def _fetch_more(self):
         batch_size = self._arraysize
@@ -497,7 +498,6 @@ class Cursor(DBAPICursor):
         buffer = get_next_result_batch_response.resultBatch
         if not self._is_metadata_updated:
             self.update_mete_data()
-            self._is_metadata_updated = True
         if not buffer or len(buffer) == 0:
             return None
         # one batch retrieves the predefined set of rows

{e6data-python-connector-2.0.2 → e6data-python-connector-2.0.4/e6data_python_connector.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: e6data-python-connector
-Version: 2.0.2
+Version: 2.0.4
 Summary: Client for the e6data distributed SQL Engine.
 Home-page: https://github.com/e6x-labs/e6data-python-connector
 Author: Uniphi, Inc.
@@ -27,7 +27,7 @@ Requires-Dist: sqlalchemy
 
 # e6data Python Connector
 
-![version](https://img.shields.io/badge/version-2.0.2-blue.svg)
+![version](https://img.shields.io/badge/version-2.0.4-blue.svg)
 
 ## Introduction
 

{e6data-python-connector-2.0.2 → e6data-python-connector-2.0.4}/setup.py

@@ -12,7 +12,7 @@
 
 import setuptools
 
-VERSION = [2, 0, 2]
+VERSION = [2, 0, 4]
 
 
 def get_long_desc():