apache-iotdb 2.0.2__py3-none-any.whl → 2.0.4.dev0__py3-none-any.whl

iotdb/SessionPool.py

@@ -21,12 +21,14 @@ import time
 from queue import Queue
 from threading import Lock
 
+from tzlocal import get_localzone_name
+
 from iotdb.Session import Session
 
 DEFAULT_MULTIPIE = 5
 DEFAULT_FETCH_SIZE = 5000
 DEFAULT_MAX_RETRY = 3
-DEFAULT_TIME_ZONE = "UTC+8"
+DEFAULT_TIME_ZONE = get_localzone_name()
 SQL_DIALECT = "tree"
 logger = logging.getLogger("IoTDB")
 

iotdb/template/MeasurementNode.py

@@ -19,7 +19,7 @@ import warnings
 
 from iotdb.utils.IoTDBConstants import TSDataType, TSEncoding, Compressor
 from .TemplateNode import TemplateNode
-from ..tsfile.utils.ReadWriteIOUtils import ReadWriteUtils
+from ..tsfile.utils.read_write_io_utils import ReadWriteUtils
 
 warnings.simplefilter("always", DeprecationWarning)
 

iotdb/template/Template.py

@@ -20,9 +20,9 @@ import struct
 import warnings
 
 from .TemplateNode import TemplateNode
-from ..tsfile.common.constant.TsFileConstant import TsFileConstant
-from ..tsfile.utils.Pair import Pair
-from ..tsfile.utils.ReadWriteIOUtils import ReadWriteUtils
+from ..tsfile.common.constant.tsfile_constant import TsFileConstant
+from ..tsfile.utils.pair import Pair
+from ..tsfile.utils.read_write_io_utils import ReadWriteUtils
 
 warnings.simplefilter("always", DeprecationWarning)
 

iotdb/thrift/common/ttypes.py

@@ -161,6 +161,7 @@ class TAggregationType(object):
     MIN = 27
     MAX = 28
     COUNT_ALL = 29
+    APPROX_COUNT_DISTINCT = 30
 
     _VALUES_TO_NAMES = {
         0: "COUNT",
@@ -193,6 +194,7 @@ class TAggregationType(object):
         27: "MIN",
         28: "MAX",
         29: "COUNT_ALL",
+        30: "APPROX_COUNT_DISTINCT",
     }
 
     _NAMES_TO_VALUES = {
@@ -226,6 +228,7 @@ class TAggregationType(object):
         "MIN": 27,
         "MAX": 28,
         "COUNT_ALL": 29,
+        "APPROX_COUNT_DISTINCT": 30,
     }
 
 
@@ -2533,13 +2536,15 @@ class TServiceProvider(object):
     Attributes:
      - endPoint
      - serviceType
+     - nodeId
 
     """
 
 
-    def __init__(self, endPoint=None, serviceType=None,):
+    def __init__(self, endPoint=None, serviceType=None, nodeId=None,):
         self.endPoint = endPoint
         self.serviceType = serviceType
+        self.nodeId = nodeId
 
     def read(self, iprot):
         if iprot._fast_decode is not None and isinstance(iprot.trans, TTransport.CReadableTransport) and self.thrift_spec is not None:
@@ -2561,6 +2566,11 @@ class TServiceProvider(object):
                     self.serviceType = iprot.readI32()
                 else:
                     iprot.skip(ftype)
+            elif fid == 3:
+                if ftype == TType.I32:
+                    self.nodeId = iprot.readI32()
+                else:
+                    iprot.skip(ftype)
             else:
                 iprot.skip(ftype)
             iprot.readFieldEnd()
@@ -2579,6 +2589,10 @@ class TServiceProvider(object):
             oprot.writeFieldBegin('serviceType', TType.I32, 2)
             oprot.writeI32(self.serviceType)
             oprot.writeFieldEnd()
+        if self.nodeId is not None:
+            oprot.writeFieldBegin('nodeId', TType.I32, 3)
+            oprot.writeI32(self.nodeId)
+            oprot.writeFieldEnd()
         oprot.writeFieldStop()
         oprot.writeStructEnd()
 
@@ -2587,6 +2601,8 @@ class TServiceProvider(object):
             raise TProtocolException(message='Required field endPoint is unset!')
         if self.serviceType is None:
             raise TProtocolException(message='Required field serviceType is unset!')
+        if self.nodeId is None:
+            raise TProtocolException(message='Required field nodeId is unset!')
         return
 
     def __repr__(self):
@@ -3265,6 +3281,7 @@ TServiceProvider.thrift_spec = (
     None,  # 0
     (1, TType.STRUCT, 'endPoint', [TEndPoint, None], None, ),  # 1
     (2, TType.I32, 'serviceType', None, None, ),  # 2
+    (3, TType.I32, 'nodeId', None, None, ),  # 3
 )
 all_structs.append(TSender)
 TSender.thrift_spec = (