# Licensed to the Apache Software Foundation (ASF) under one

# or more contributor license agreements. See the NOTICE file

# distributed with this work for additional information

# regarding copyright ownership. The ASF licenses this file

# to you under the Apache License, Version 2.0 (the

# "License"); you may not use this file except in compliance

# with the License. You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing,

# software distributed under the License is distributed on an

# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

# KIND, either express or implied. See the License for the

# specific language governing permissions and limitations

# under the License.

#

from iotdb.SessionPool import PoolConfig, SessionPool

from iotdb.Session import Session

from iotdb.table_session import TableSession, TableSessionConfig

ip = "127.0.0.1"

port_ = "6667"

username_ = "root"

password_ = "root"

# Configure SSL enabled

use_ssl = True

# Configure certificate path

ca_certs = "/path/server.crt"

# Configure client certificate and private key when the server enables mTLS.

client_cert = "/path/client.crt"

client_key = "/path/client.key"

def get_data():

session = Session(

ip,

port_,

username_,

password_,

use_ssl=use_ssl,

ca_certs=ca_certs,

client_cert=client_cert,

client_key=client_key,

)

session.open(False)

result = session.execute_query_statement("select * from root.eg.etth")

df = result.todf()

df.rename(columns={"Time": "date"}, inplace=True)

session.close()

return df

def get_data2():

pool_config = PoolConfig(

host=ip,

port=port_,

user_name=username_,

password=password_,

fetch_size=1024,

time_zone="Asia/Shanghai",

max_retry=3,

use_ssl=use_ssl,

ca_certs=ca_certs,

client_cert=client_cert,

client_key=client_key,

)

max_pool_size = 5

wait_timeout_in_ms = 3000

session_pool = SessionPool(pool_config, max_pool_size, wait_timeout_in_ms)

session = session_pool.get_session()

result = session.execute_query_statement("select * from root.eg.etth")

df = result.todf()

df.rename(columns={"Time": "date"}, inplace=True)

session_pool.put_back(session)

session_pool.close()

def get_table_data():

pool_config = TableSessionConfig(

node_urls=["127.0.0.1:6667"],

username=username_,

password=password_,

fetch_size=1024,

time_zone="Asia/Shanghai",

use_ssl=use_ssl,

ca_certs=ca_certs,

client_cert=client_cert,

client_key=client_key,

)

session = TableSession(pool_config)

result = session.execute_query_statement("select * from test")

df = result.todf()

session.close()

if __name__ == "__main__":

df = get_data()