apache-iotdb 1.3.4__py3-none-any.whl → 2.0.2__py3-none-any.whl

{apache_iotdb-1.3.4.dist-info → apache_iotdb-2.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apache-iotdb
-Version: 1.3.4
+Version: 2.0.2
 Summary: Apache IoTDB client API
 Author-email: Apache IoTDB <dev@iotdb.apache.org>
 License: Apache License, Version 2.0
@@ -17,7 +17,7 @@ Description-Content-Type: text/markdown
 Requires-Dist: thrift>=0.14.1
 Requires-Dist: pandas>=1.0.0
 Requires-Dist: numpy>=1.0.0
-Requires-Dist: sqlalchemy<1.5,>=1.4
+Requires-Dist: sqlalchemy>=1.4
 Requires-Dist: sqlalchemy-utils>=0.37.8
 
 <!--
@@ -69,9 +69,9 @@ You have to install thrift (>=0.14.1) before using the package.
 
 First, download the latest package: `pip3 install apache-iotdb`
 
-You can get an example of using the package to read and write data at here: [Example](https://github.com/apache/iotdb/blob/master/client-py/SessionExample.py)
+You can get an example of using the package to read and write data at here: [Example](https://github.com/apache/iotdb/blob/master/iotdb-client/client-py/SessionExample.py)
 
-An example of aligned timeseries: [Aligned Timeseries Session Example](https://github.com/apache/iotdb/blob/master/client-py/SessionAlignedTimeseriesExample.py)
+An example of aligned timeseries: [Aligned Timeseries Session Example](https://github.com/apache/iotdb/blob/master/iotdb-client/client-py/SessionAlignedTimeseriesExample.py)
 
 (you need to add `import iotdb` in the head of the file)
 

{apache_iotdb-1.3.4.dist-info → apache_iotdb-2.0.2.dist-info}/RECORD

@@ -1,21 +1,17 @@
-iotdb/IoTDBContainer.py,sha256=gmGMEzacytqvrpCXGn0xrm3ab4LDyToNCHPEFg82Mwg,2514
-iotdb/Session.py,sha256=kPT3lyLNP5AoJMVzHUlcOqsFQc9rkQb9MvTVP2Jn-gg,95232
-iotdb/SessionPool.py,sha256=SSB6ohCI-8cpXYneSByuckLQQGNig3ERVaABsMgFay0,6095
+iotdb/Session.py,sha256=x0vSwainD_xtL4uvYOBSEFcY5LofMaHH-HT57blCQTA,102965
+iotdb/SessionPool.py,sha256=-XXWDJ2Fpo1rceyc_Q-uv7XD5p69YdR1drD9tyBqUwQ,6621
 iotdb/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
+iotdb/table_session.py,sha256=Jav4TuvRl4PxOhNBaBqTvMGFWemu5YkAwRU_KXUy9wk,6173
+iotdb/table_session_pool.py,sha256=sqhceeGbr1bsLjYj6aEF_p3qzqiVhtXKAEjV9CY4pWI,5329
 iotdb/dbapi/Connection.py,sha256=t5CuwOTrMJxJ2YVxaEKi2fdaH9Vob-u7nQRCD_9vLcg,2569
 iotdb/dbapi/Cursor.py,sha256=CBFyHsk7cOxLJEwCWMJK0r4guXQ-OkNh39H54cyyH5k,9110
 iotdb/dbapi/Exceptions.py,sha256=X3qhGXQ_rNL7998DQwFeQc9cG5ByXlfhIQx2sBHOiuw,1278
 iotdb/dbapi/__init__.py,sha256=98jGNzw_AI1iqbYbUxGRg4Yd92Uo63S1v143_NnKKCs,953
-iotdb/dbapi/tests/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
-iotdb/dbapi/tests/test_connection.py,sha256=cq4NlkwUMEi22e2vpNuNZKwFPNC6mTc0y84qdBXAbac,1689
-iotdb/dbapi/tests/test_cursor.py,sha256=etg9sOpBLL6iIBZIoOnfPMGzMn__kwDZFnHT6ugpxRU,3943
-iotdb/sqlalchemy/IoTDBDialect.py,sha256=Pt8O-TjGN3DzO63szD6iYA73A7g3OhYSxf7SPqvp9WQ,4557
+iotdb/sqlalchemy/IoTDBDialect.py,sha256=047PV15HtnTej_cXkfwmzwp5IIOYyMco6bSeqHWDItI,4695
 iotdb/sqlalchemy/IoTDBIdentifierPreparer.py,sha256=NxOf58CXp6QYeifyF9KAZfEgPbNW0jIRZ6CwRvWcV-8,1089
 iotdb/sqlalchemy/IoTDBSQLCompiler.py,sha256=UuLRB7sEuy3YVVVU7oBeEKcJ9-OdLsbP7eiyIlja9kY,11790
 iotdb/sqlalchemy/IoTDBTypeCompiler.py,sha256=1ZrQvMt3lieN-HUjJLlmSVGy1DwBCWZMwC8zZjX6FOI,1416
 iotdb/sqlalchemy/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
-iotdb/sqlalchemy/tests/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
-iotdb/sqlalchemy/tests/test_dialect.py,sha256=MdrJXqVpwRznppYvqC3OfVUOjR0sbMU5ubpJacDxOAU,3083
 iotdb/template/MeasurementNode.py,sha256=O0-MVLN9a-Sap8uPriQxlmkh_NyUsiWGEoNQgmyBQ28,2315
 iotdb/template/Template.py,sha256=9zSaBsTwT_hWj-qf8Y_r9cIdd4WhdIrhxAl31Ejmo5E,3649
 iotdb/template/TemplateNode.py,sha256=6dR84yg7QV5A4nmGqws8cfV3GygfrR5HErQS5t2ORNw,1556
@@ -24,11 +20,11 @@ iotdb/template/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,78
 iotdb/thrift/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iotdb/thrift/common/__init__.py,sha256=YFcZTT8Cm-6Y4oTiCaqq0DT1lw2W09WqoEc5_pTAwW0,34
 iotdb/thrift/common/constants.py,sha256=IKRM8SpfhYWXERWBgUbUhbdFpn363hmZjXDhEYnVRbw,366
-iotdb/thrift/common/ttypes.py,sha256=CsmVig5NMdix7D2yddwdWj7ItBcmsM--8l_sVTXcOwA,115657
-iotdb/thrift/rpc/IClientRPCService.py,sha256=C9CZxONUf14jJgAjrgOXfxwK3rfwfVrnCDnucwpDkwI,416186
+iotdb/thrift/common/ttypes.py,sha256=L8cZ-biq0G2528LSKWNteVgFkWYTBMttyr6da_JXiIM,116582
+iotdb/thrift/rpc/IClientRPCService.py,sha256=gHrJArjwcFMkcKIR46CFiz_D5rhR-WetjQSI8jRLXeY,416186
 iotdb/thrift/rpc/__init__.py,sha256=69gRO2XxQu71SDeCRxiZFjK6Xw_61hdl5MdWceI-e1A,55
 iotdb/thrift/rpc/constants.py,sha256=IKRM8SpfhYWXERWBgUbUhbdFpn363hmZjXDhEYnVRbw,366
-iotdb/thrift/rpc/ttypes.py,sha256=hfsy0X7gLIizHVAqqMphbcgWwZPZNoHWaAvuCDwArak,295853
+iotdb/thrift/rpc/ttypes.py,sha256=SBn1WHuSwmEoZrO6Ke2Ddcty6z52Hu1lGJp0uDJ1z8Y,301147
 iotdb/tsfile/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
 iotdb/tsfile/common/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
 iotdb/tsfile/common/constant/TsFileConstant.py,sha256=uLSw9vnQZ6objfyhO3b9bHTip8yUW5aapPHEsXC0e7s,1165
@@ -38,31 +34,40 @@ iotdb/tsfile/utils/Pair.py,sha256=91ecijUAlEiMRLlH2hjRi_mT_xWp6Vm-R3n4Z4USFqQ,95
 iotdb/tsfile/utils/ReadWriteIOUtils.py,sha256=31CKH_1XWz8xc7DyInKaD-avkY-hVqNQILDb5PMucm4,2977
 iotdb/tsfile/utils/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
 iotdb/utils/BitMap.py,sha256=dGvruuvDJlT4KH7pIrkprZWsY_AG-P8HXe6JHBKDOpI,1436
-iotdb/utils/Field.py,sha256=zc4oJBcib8BPnVpLTq_3sbPfmGRAO7SmQiWeuPgaiGc,7303
+iotdb/utils/Field.py,sha256=gn2p7RBbn_Kr0r2jAGNK4TkHDOF5SrwxmeW4UgptRVQ,7302
 iotdb/utils/IoTDBConnectionException.py,sha256=D38r5MicwIdVsliCwtipICjx19wbvRI4ye0doAX7DPM,1179
 iotdb/utils/IoTDBConstants.py,sha256=mOXYQvaOccFS0cOUEqSXgZ3qSjhe6K8_sIjTT9S6qaI,1950
-iotdb/utils/IoTDBRpcDataSet.py,sha256=mYqfeQQQijr25mjarmPXyWhnyinBL9NNL3nSWKVo4yA,18411
-iotdb/utils/NumpyTablet.py,sha256=VQiuRb2vJSVgqpzod4OocehXh7JdUNBGH-MTV4lzg0g,6919
+iotdb/utils/IoTDBRpcDataSet.py,sha256=aFG1BGaXFYlivTmkx2exZh5lfCPWc7JDD0PC72SR5FQ,18429
+iotdb/utils/NumpyTablet.py,sha256=ImJ2y4QOQ36skWnfqVVxQHLmFh6kkik19EIAyyiQtWo,7950
 iotdb/utils/RowRecord.py,sha256=TIfQzMdi84GhGGIfVcZ-kvmm7aQwsDWge2kY00sf1lo,1690
 iotdb/utils/SessionDataSet.py,sha256=uS06ceDkhb8TocGM8MzdefofauoOrC4GWYi1TjWQS1w,5359
-iotdb/utils/Tablet.py,sha256=VM0Prem8xqLqV3YF9lbtGQ4N4pqQ1KvdbqrvczB6A2g,9035
+iotdb/utils/Tablet.py,sha256=WguKP2_Gu3UzaaKVt9wPR8kpib3Yd5iMONAzyAoIxbI,10197
 iotdb/utils/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
 tests/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
 tests/integration/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
+tests/integration/iotdb_container.py,sha256=gmGMEzacytqvrpCXGn0xrm3ab4LDyToNCHPEFg82Mwg,2514
 tests/integration/tablet_performance_comparison.py,sha256=cjWV-RGhp73JjzITyKwC7Abx5IRGg00cLrS3cH3BkOo,10724
-tests/integration/test_aligned_timeseries.py,sha256=hig7q7UEBEdYiQnCmW2uOUiVopBW9_WpUBWE8sWP8lk,9906
-tests/integration/test_dataframe.py,sha256=c3ATEbljj6NKFhbOOAFZX-WC4saoNUFNVfLfic9ogf8,2808
-tests/integration/test_delete_data.py,sha256=vDDNWJ_WUZihLOMdwRMyS4r9F6nm-SalVSx1hPXrDlE,5121
-tests/integration/test_new_data_types.py,sha256=qi1VKybP9S-KNRpXlnzgKYGoMEdRUayrr_WiXy7IN4A,6367
-tests/integration/test_one_device.py,sha256=QW6t8GfH4FnYxZODYztpHEt8m3NAC-AUcs18KKK4q0w,4734
-tests/integration/test_session.py,sha256=aLhjUxZJ1IX2DBPuh3h7E4vfJs2jTnBa4dSIDgnMLIo,14343
-tests/integration/test_session_pool.py,sha256=v4cbLLwob8JTG0vEV1zhr_YCHM7649JacBOg2Svcemo,3891
-tests/integration/test_tablet.py,sha256=m0jLWJPQpy0K8sgOMS-91MaIL8QnwI3u4SWHjcEWbDI,4438
-tests/integration/test_todf.py,sha256=MlntH0pnz9xXx7CJMNlb5Dm3MijKh6tpcQrr7jYTM9Q,8128
+tests/integration/test_aligned_timeseries.py,sha256=W6GG08YzYDfmF5rlvKmF_iAHxNoZIc9ETaAslUTAWu4,9902
+tests/integration/test_dataframe.py,sha256=6jnCqpvw-3uYLzCli9x9hnXTX8llnrLWdOzgOZ41OFA,2804
+tests/integration/test_delete_data.py,sha256=7ZSVZYLFChM4V7oeBKWyZeilr_HBaLgQ8hk6H7TpwEY,5117
+tests/integration/test_new_data_types.py,sha256=fjUccRuSGKHTQr03NBI3kVSqZ4IAgGZ3U_vJCoBiWIA,6363
+tests/integration/test_one_device.py,sha256=6Wab6CzHvoLdlCHgYBdBD7woZqOqQ57j4AWcVFd2q4M,4730
+tests/integration/test_relational_session.py,sha256=sZLZZj9IZSLVkGEPJGj2L_dmLmasm-kKwANERTPsnq8,4448
+tests/integration/test_session.py,sha256=7ewSmo7eMte8J7WX7t7cu70DOv644rbwzJjC9ElDwFo,16510
+tests/integration/test_session_pool.py,sha256=cldbmfO5NSK9XQipwiDgjjixpxuSoBTVYObhlFm-WRo,3887
+tests/integration/test_tablemodel_insert.py,sha256=NMF4f-FzhAIDc8qLBxI8rZEtQjfs0pDPykMWx19cjwI,54634
+tests/integration/test_tablet.py,sha256=j6LQJKVAZ9elZdHrv8O-yeb3q9yNIXOdjCexJqlLE24,4434
+tests/integration/test_todf.py,sha256=v4aF9SwU0gcKgIms161zFbMCCWzwGACfwZ16QPQLPb0,8124
+tests/integration/test_treemodel_insert.py,sha256=Td1pThwzjxg2JKSKKeMXsi9WYibDGa_eY9GlPRGajJo,195607
+tests/integration/dbapi/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
+tests/integration/dbapi/test_connection.py,sha256=FgM9Ama-YHcuZ-IaukAjoFcqnff9YEkAvVnXg9Y7AXg,1702
+tests/integration/dbapi/test_cursor.py,sha256=5pmvTMTfsBDn9hVbBEnbrB4Obknbx3zDHm2VA19yheQ,3956
+tests/integration/sqlalchemy/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
+tests/integration/sqlalchemy/test_dialect.py,sha256=QvhZ3h-gbXQHRLQglbIeBM-YaJxWjd8NFwCCtRZDCQg,3478
 tests/unit/__init__.py,sha256=vGVFsdMOp1iKaCQl5rfYcPVa0FV94d-a7MsMDQDhHMc,789
 tests/unit/test_numpy_tablet.py,sha256=w3lLK8Cft7lGsCj_6Z1Y7kDshltN0osSDJb-EESi6_Y,7235
-apache_iotdb-1.3.4.dist-info/METADATA,sha256=zL2_FMt-GbC9YkgWdAnj4vY0wWXJWS89EusBfyhuSnI,18771
-apache_iotdb-1.3.4.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-apache_iotdb-1.3.4.dist-info/entry_points.txt,sha256=BmQWy6NHyWXorQcVm8-VWpt9MSqe46UFyV2hnsl4yok,73
-apache_iotdb-1.3.4.dist-info/top_level.txt,sha256=B9jGGOaAAqF1WrWw2jX2K-naWAZn2nhyrL-6hM1qwyM,12
-apache_iotdb-1.3.4.dist-info/RECORD,,
+apache_iotdb-2.0.2.dist-info/METADATA,sha256=FRHjfXTDRW5l9tzgjeGCXdnkpxuuzk_3dUwfahHxA6E,18792
+apache_iotdb-2.0.2.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+apache_iotdb-2.0.2.dist-info/entry_points.txt,sha256=BmQWy6NHyWXorQcVm8-VWpt9MSqe46UFyV2hnsl4yok,73
+apache_iotdb-2.0.2.dist-info/top_level.txt,sha256=B9jGGOaAAqF1WrWw2jX2K-naWAZn2nhyrL-6hM1qwyM,12
+apache_iotdb-2.0.2.dist-info/RECORD,,

iotdb/Session.py

@@ -70,11 +70,12 @@ class Session(object):
     SUCCESS_STATUS = 200
     MULTIPLE_ERROR = 302
     REDIRECTION_RECOMMEND = 400
-    DEFAULT_FETCH_SIZE = 10000
+    DEFAULT_FETCH_SIZE = 5000
     DEFAULT_USER = "root"
     DEFAULT_PASSWORD = "root"
     DEFAULT_ZONE_ID = time.strftime("%z")
     RETRY_NUM = 3
+    SQL_DIALECT = "tree"
 
     def __init__(
         self,
@@ -108,6 +109,8 @@ class Session(object):
         self.__enable_redirection = enable_redirection
         self.__device_id_to_endpoint = None
         self.__endpoint_to_connection = None
+        self.sql_dialect = self.SQL_DIALECT
+        self.database = None
         self.__use_ssl = use_ssl
         self.__ca_certs = ca_certs
         self.__connection_timeout_in_ms = connection_timeout_in_ms
@@ -191,12 +194,15 @@ class Session(object):
         else:
             client = Client(TBinaryProtocol.TBinaryProtocolAccelerated(transport))
 
+        configuration = {"version": "V_1_0", "sql_dialect": self.sql_dialect}
+        if self.database is not None:
+            configuration["db"] = self.database
         open_req = TSOpenSessionReq(
             client_protocol=self.protocol_version,
             username=self.__user,
             password=self.__password,
             zoneId=self.__zone_id,
-            configuration={"version": "V_1_0"},
+            configuration=configuration,
         )
 
         try:
@@ -535,6 +541,15 @@ class Session(object):
             string_values = [string_values]
         if type(measurements) == str:
             measurements = [measurements]
+        if self.__has_none_value(string_values):
+            filtered_measurements, filtered_values = zip(
+                *[(m, v) for m, v in zip(measurements, string_values) if v is not None]
+            )
+            measurements = list(filtered_measurements)
+            values = list(filtered_values)
+            if len(measurements) == 0 or len(values) == 0:
+                logger.info("All inserting values are none!")
+                return
         request = self.gen_insert_str_record_req(
             device_id, timestamp, measurements, string_values
         )
@@ -566,6 +581,15 @@ class Session(object):
             string_values = [string_values]
         if type(measurements) == str:
             measurements = [measurements]
+        if self.__has_none_value(string_values):
+            filtered_measurements, filtered_values = zip(
+                *[(m, v) for m, v in zip(measurements, string_values) if v is not None]
+            )
+            measurements = list(filtered_measurements)
+            values = list(filtered_values)
+            if len(measurements) == 0 or len(values) == 0:
+                logger.info("All inserting values are none!")
+                return
         request = self.gen_insert_str_record_req(
             device_id, timestamp, measurements, string_values, True
         )
@@ -601,6 +625,20 @@ class Session(object):
         :param data_types: List of TSDataType, indicate the data type for each sensor
         :param values: List, values to be inserted, for each sensor
         """
+        if self.__has_none_value(values):
+            filtered_measurements, filtered_data_types, filtered_values = zip(
+                *[
+                    (m, d, v)
+                    for m, d, v in zip(measurements, data_types, values)
+                    if v is not None
+                ]
+            )
+            measurements = list(filtered_measurements)
+            data_types = list(filtered_data_types)
+            values = list(filtered_values)
+            if len(measurements) == 0 or len(data_types) == 0 or len(values) == 0:
+                logger.info("All inserting values are none!")
+                return
         request = self.gen_insert_record_req(
             device_id, timestamp, measurements, data_types, values
         )
@@ -634,6 +672,19 @@ class Session(object):
         :param types_lst: 2-D List of TSDataType, each element of outer list indicates sensor data types of a device
         :param values_lst: 2-D List, values to be inserted, for each device
         """
+        if self.__has_none_value(values_lst):
+            (
+                device_ids,
+                times,
+                measurements_lst,
+                types_lst,
+                values_lst,
+            ) = self.__filter_lists_by_values(
+                device_ids, times, measurements_lst, types_lst, values_lst
+            )
+            if len(device_ids) == 0:
+                logger.info("All inserting values are none!")
+                return
         if self.__enable_redirection:
             request_group = {}
             for i in range(len(device_ids)):
@@ -703,6 +754,20 @@ class Session(object):
         :param data_types: List of TSDataType, indicate the data type for each sensor
         :param values: List, values to be inserted, for each sensor
         """
+        if self.__has_none_value(values):
+            filtered_measurements, filtered_data_types, filtered_values = zip(
+                *[
+                    (m, d, v)
+                    for m, d, v in zip(measurements, data_types, values)
+                    if v is not None
+                ]
+            )
+            measurements = list(filtered_measurements)
+            data_types = list(filtered_data_types)
+            values = list(filtered_values)
+            if len(measurements) == 0 or len(data_types) == 0 or len(values) == 0:
+                logger.info("All inserting values are none!")
+                return
         request = self.gen_insert_record_req(
             device_id, timestamp, measurements, data_types, values, True
         )
@@ -736,6 +801,19 @@ class Session(object):
         :param types_lst: 2-D List of TSDataType, each element of outer list indicates sensor data types of a device
         :param values_lst: 2-D List, values to be inserted, for each device
         """
+        if self.__has_none_value(values_lst):
+            (
+                device_ids,
+                times,
+                measurements_lst,
+                types_lst,
+                values_lst,
+            ) = self.__filter_lists_by_values(
+                device_ids, times, measurements_lst, types_lst, values_lst
+            )
+            if len(device_ids) == 0:
+                logger.info("All inserting values are none!")
+                return
         if self.__enable_redirection:
             request_group = {}
             for i in range(len(device_ids)):
@@ -928,13 +1006,15 @@ class Session(object):
         """
         request = self.gen_insert_tablet_req(tablet)
         try:
-            connection = self.get_connection(tablet.get_device_id())
+            connection = self.get_connection(tablet.get_insert_target_name())
             request.sessionId = connection.session_id
             return Session.verify_success_with_redirection(
                 connection.client.insertTablet(request)
             )
         except RedirectException as e:
-            return self.handle_redirection(tablet.get_device_id(), e.redirect_node)
+            return self.handle_redirection(
+                tablet.get_insert_target_name(), e.redirect_node
+            )
         except TTransport.TException as e:
             if self.reconnect():
                 try:
@@ -953,14 +1033,14 @@ class Session(object):
         if self.__enable_redirection:
             request_group = {}
             for i in range(len(tablet_lst)):
-                connection = self.get_connection(tablet_lst[i].get_device_id())
+                connection = self.get_connection(tablet_lst[i].get_insert_target_name())
                 request = request_group.setdefault(
                     connection.client,
                     TSInsertTabletsReq(
                         connection.session_id, [], [], [], [], [], [], False
                     ),
                 )
-                request.prefixPaths.append(tablet_lst[i].get_device_id())
+                request.prefixPaths.append(tablet_lst[i].get_insert_target_name())
                 request.timestampsList.append(tablet_lst[i].get_binary_timestamps())
                 request.measurementsList.append(tablet_lst[i].get_measurements())
                 request.valuesList.append(tablet_lst[i].get_binary_values())
@@ -1019,13 +1099,15 @@ class Session(object):
         """
         request = self.gen_insert_tablet_req(tablet, True)
         try:
-            connection = self.get_connection(tablet.get_device_id())
+            connection = self.get_connection(tablet.get_insert_target_name())
             request.sessionId = connection.session_id
             return Session.verify_success_with_redirection(
                 connection.client.insertTablet(request)
             )
         except RedirectException as e:
-            return self.handle_redirection(tablet.get_device_id(), e.redirect_node)
+            return self.handle_redirection(
+                tablet.get_insert_target_name(), e.redirect_node
+            )
         except TTransport.TException as e:
             if self.reconnect():
                 try:
@@ -1044,14 +1126,14 @@ class Session(object):
         if self.__enable_redirection:
             request_group = {}
             for i in range(len(tablet_lst)):
-                connection = self.get_connection(tablet_lst[i].get_device_id())
+                connection = self.get_connection(tablet_lst[i].get_insert_target_name())
                 request = request_group.setdefault(
                     connection.client,
                     TSInsertTabletsReq(
                         connection.session_id, [], [], [], [], [], [], True
                     ),
                 )
-                request.prefixPaths.append(tablet_lst[i].get_device_id())
+                request.prefixPaths.append(tablet_lst[i].get_insert_target_name())
                 request.timestampsList.append(tablet_lst[i].get_binary_timestamps())
                 request.measurementsList.append(tablet_lst[i].get_measurements())
                 request.valuesList.append(tablet_lst[i].get_binary_values())
@@ -1096,6 +1178,36 @@ class Session(object):
                         self.connection_error_msg()
                     ) from None
 
+    def insert_relational_tablet(self, tablet):
+        """
+        insert one tablet, for example three column in the table1 can form a tablet:
+            timestamps,    id1,  attr1,    m1
+                     1,  id:1,  attr:1,   1.0
+                     2,  id:1,  attr:1,   2.0
+                     3,  id:2,  attr:2,   3.0
+        :param tablet: a tablet specified above
+        """
+        request = self.gen_insert_relational_tablet_req(tablet)
+        try:
+            connection = self.get_connection(tablet.get_insert_target_name())
+            request.sessionId = connection.session_id
+            return Session.verify_success_with_redirection(
+                connection.client.insertTablet(request)
+            )
+        except RedirectException as e:
+            return self.handle_redirection(
+                tablet.get_insert_target_name(), e.redirect_node
+            )
+        except TTransport.TException as e:
+            if self.reconnect():
+                try:
+                    request.sessionId = self.__session_id
+                    return Session.verify_success(self.__client.insertTablet(request))
+                except TTransport.TException as e1:
+                    raise IoTDBConnectionException(e1) from None
+            else:
+                raise IoTDBConnectionException(self.connection_error_msg()) from None
+
     def insert_records_of_one_device(
         self, device_id, times_list, measurements_list, types_list, values_list
     ):
@@ -1312,13 +1424,27 @@ class Session(object):
     def gen_insert_tablet_req(self, tablet, is_aligned=False):
         return TSInsertTabletReq(
             self.__session_id,
-            tablet.get_device_id(),
+            tablet.get_insert_target_name(),
+            tablet.get_measurements(),
+            tablet.get_binary_values(),
+            tablet.get_binary_timestamps(),
+            tablet.get_data_types(),
+            tablet.get_row_number(),
+            is_aligned,
+        )
+
+    def gen_insert_relational_tablet_req(self, tablet, is_aligned=False):
+        return TSInsertTabletReq(
+            self.__session_id,
+            tablet.get_insert_target_name(),
             tablet.get_measurements(),
             tablet.get_binary_values(),
             tablet.get_binary_timestamps(),
             tablet.get_data_types(),
             tablet.get_row_number(),
             is_aligned,
+            True,
+            tablet.get_column_categories(),
         )
 
     def gen_insert_tablets_req(self, tablet_lst, is_aligned=False):
@@ -1329,7 +1455,7 @@ class Session(object):
         type_lst = []
         size_lst = []
         for tablet in tablet_lst:
-            device_id_lst.append(tablet.get_device_id())
+            device_id_lst.append(tablet.get_insert_target_name())
             measurements_lst.append(tablet.get_measurements())
             values_lst.append(tablet.get_binary_values())
             timestamps_lst.append(tablet.get_binary_timestamps())
@@ -1402,6 +1528,18 @@ class Session(object):
             else:
                 raise IoTDBConnectionException(self.connection_error_msg()) from None
 
+        previous_db = self.database
+        if resp.database is not None:
+            self.database = resp.database
+        if previous_db != self.database and self.__endpoint_to_connection is not None:
+            iterator = iter(self.__endpoint_to_connection.items())
+            for entry in list(iterator):
+                endpoint, connection = entry
+                if connection != self.__default_connection:
+                    try:
+                        connection.change_database(sql)
+                    except Exception as e:
+                        self.__endpoint_to_connection.pop(endpoint)
         return Session.verify_success(resp.status)
 
     def execute_statement(self, sql: str, timeout=0):
@@ -1563,7 +1701,7 @@ class Session(object):
         ):
             return 0
 
-        raise RuntimeError(str(status.code) + ": " + status.message)
+        raise RuntimeError(f"{status.code}: {status.message}")
 
     @staticmethod
     def verify_success_by_list(status_list: list):
@@ -1571,14 +1709,15 @@ class Session(object):
         verify success of operation
         :param status_list: execution result status
         """
-        message = str(Session.MULTIPLE_ERROR) + ": "
-        for status in status_list:
-            if (
-                status.code != Session.SUCCESS_STATUS
-                and status.code != Session.REDIRECTION_RECOMMEND
-            ):
-                message += status.message + "; "
-        raise RuntimeError(message)
+        error_messages = [
+            status.message
+            for status in status_list
+            if status.code
+            not in {Session.SUCCESS_STATUS, Session.REDIRECTION_RECOMMEND}
+        ]
+        if error_messages:
+            message = f"{Session.MULTIPLE_ERROR}: {'; '.join(error_messages)}"
+            raise RuntimeError(message)
 
     @staticmethod
     def verify_success_with_redirection(status: TSStatus):
@@ -1879,6 +2018,49 @@ class Session(object):
         )
         return request
 
+    def __has_none_value(self, values_list) -> bool:
+        for item in values_list:
+            if isinstance(item, list):
+                if self.__has_none_value(item):
+                    return True
+            elif item is None:
+                return True
+        return False
+
+    @staticmethod
+    def __filter_lists_by_values(
+        device_lst, time_lst, measurements_lst, types_lst, values_lst
+    ):
+        filtered_devices = []
+        filtered_times = []
+        filtered_measurements = []
+        filtered_types = []
+        filtered_values = []
+
+        for device, time_, measurements, types, values in zip(
+            device_lst, time_lst, measurements_lst, types_lst, values_lst
+        ):
+            filtered_row = [
+                (m, t, v)
+                for m, t, v in zip(measurements, types, values)
+                if v is not None
+            ]
+            if filtered_row:
+                f_measurements, f_types, f_values = zip(*filtered_row)
+                filtered_measurements.append(list(f_measurements))
+                filtered_types.append(list(f_types))
+                filtered_values.append(list(f_values))
+                filtered_devices.append(device)
+                filtered_times.append(time_)
+
+        return (
+            filtered_devices,
+            filtered_times,
+            filtered_measurements,
+            filtered_types,
+            filtered_values,
+        )
+
     def create_schema_template(self, template: Template):
         warnings.warn(
             "The APIs about template are deprecated and will be removed in future versions. Use sql instead.",
@@ -2290,6 +2472,17 @@ class SessionConnection(object):
         self.session_id = session_id
         self.statement_id = statement_id
 
+    def change_database(self, sql):
+        try:
+            self.client.executeUpdateStatement(
+                TSExecuteStatementReq(self.session_id, sql, self.statement_id)
+            )
+        except TTransport.TException as e:
+            raise IoTDBConnectionException(
+                "failed to change database",
+                e,
+            ) from None
+
     def close_connection(self, req):
         try:
             self.client.closeSession(req)

iotdb/SessionPool.py

@@ -27,6 +27,7 @@ DEFAULT_MULTIPIE = 5
 DEFAULT_FETCH_SIZE = 5000
 DEFAULT_MAX_RETRY = 3
 DEFAULT_TIME_ZONE = "UTC+8"
+SQL_DIALECT = "tree"
 logger = logging.getLogger("IoTDB")
 
 
@@ -42,6 +43,7 @@ class PoolConfig(object):
         time_zone: str = DEFAULT_TIME_ZONE,
         max_retry: int = DEFAULT_MAX_RETRY,
         enable_compression: bool = False,
+        enable_redirection: bool = True,
         use_ssl: bool = False,
         ca_certs: str = None,
         connection_timeout_in_ms: int = None,
@@ -61,6 +63,7 @@ class PoolConfig(object):
         self.time_zone = time_zone
         self.max_retry = max_retry
         self.enable_compression = enable_compression
+        self.enable_redirection = enable_redirection
         self.use_ssl = use_ssl
         self.ca_certs = ca_certs
         self.connection_timeout_in_ms = connection_timeout_in_ms
@@ -77,6 +80,8 @@ class SessionPool(object):
         self.__queue = Queue(max_pool_size)
         self.__lock = Lock()
         self.__closed = False
+        self.sql_dialect = SQL_DIALECT
+        self.database = None
 
     def __construct_session(self) -> Session:
         if len(self.__pool_config.node_urls) > 0:
@@ -86,10 +91,13 @@ class SessionPool(object):
                 self.__pool_config.password,
                 self.__pool_config.fetch_size,
                 self.__pool_config.time_zone,
+                enable_redirection=self.__pool_config.enable_redirection,
                 use_ssl=self.__pool_config.use_ssl,
                 ca_certs=self.__pool_config.ca_certs,
                 connection_timeout_in_ms=self.__pool_config.connection_timeout_in_ms,
             )
+            session.sql_dialect = self.sql_dialect
+            session.database = self.database
 
         else:
             session = Session(
@@ -99,10 +107,13 @@ class SessionPool(object):
                 self.__pool_config.password,
                 self.__pool_config.fetch_size,
                 self.__pool_config.time_zone,
+                enable_redirection=self.__pool_config.enable_redirection,
                 use_ssl=self.__pool_config.use_ssl,
                 ca_certs=self.__pool_config.ca_certs,
                 connection_timeout_in_ms=self.__pool_config.connection_timeout_in_ms,
             )
+            session.sql_dialect = self.sql_dialect
+            session.database = self.database
 
         session.open(self.__pool_config.enable_compression)
         return session
@@ -114,6 +125,7 @@ class SessionPool(object):
         return q
 
     def get_session(self) -> Session:
+
         if self.__closed:
             raise ConnectionError("SessionPool has already been closed.")
 
@@ -148,6 +160,7 @@ class SessionPool(object):
         return session
 
     def put_back(self, session: Session):
+
         if self.__closed:
             raise ConnectionError(
                 "SessionPool has already been closed, please close the session manually."
@@ -160,6 +173,7 @@ class SessionPool(object):
                 self.__pool_size -= 1
 
     def close(self):
+
         with self.__lock:
             while not self.__queue.empty():
                 session = self.__queue.get(block=False)