e6data-python-connector 2.0.1__tar.gz → 2.0.3__tar.gz

{e6data-python-connector-2.0.1/e6data_python_connector.egg-info → e6data-python-connector-2.0.3}/PKG-INFO

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

{e6data-python-connector-2.0.1 → e6data-python-connector-2.0.3}/README.md

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

{e6data-python-connector-2.0.1 → e6data-python-connector-2.0.3}/e6data_python_connector/datainputstream.py

@@ -8,7 +8,7 @@ from thrift.transport import TTransport
 
 from e6data_python_connector.e6x_vector.ttypes import Chunk, Vector, VectorType
 from e6data_python_connector.constants import ZONE
-from e6data_python_connector.date_time_utils import floor_div, floor_mod
+from e6data_python_connector.date_time_utils import floor_div, floor_mod, get_zone
 
 try:
     from thrift.protocol import fastbinary
@@ -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_seconds = floor_div(epoch_micros, 1000_000)
                 micros_of_the_day = floor_mod(epoch_micros, 1000_000)
-                date_time = datetime.fromtimestamp(epoch_seconds, ZONE)
+                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.1 → e6data-python-connector-2.0.3}/e6data_python_connector/e6data_grpc.py

@@ -395,6 +395,7 @@ class Cursor(DBAPICursor):
         """
         Semicolon is now not supported. So removing it from query end.
         """
+        operation = operation.strip()  # Remove leading and trailing whitespaces.
         if operation.endswith(';'):
             operation = operation[:-1]
 

{e6data-python-connector-2.0.1 → e6data-python-connector-2.0.3/e6data_python_connector.egg-info}/PKG-INFO

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

{e6data-python-connector-2.0.1 → e6data-python-connector-2.0.3}/setup.py

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