dbconn = pq.dbconnect('dw_2')

# Use this connection to read data
rows = dbconn.fetch('db_name', 'schema_name', 'table_name')

rows = dbconn.fetch('db_name', 'schema_name', 'table_name')

# Read data using custom query
rows = dbconn.fetch('db_name', query='SELECT * FROM schema_name.table_name')

# Get result as a data frame
df = dbconn.fetch('db_name', 'schema_name', 'table_name', df=True)

# Options:
# df = True: set to True to return a Dataframe instead of a list of objects
# fillna_with = '': replace Na (not a number) with a given value, e.g. in empty string
# fillnat_with= '': replace NaT (not a date)
# enable_python_types = True: use Python types in response
# enable_datetime_as_string = True: return datetime columns as string
# tz='UTC': timezone for datetimes

dbconn.insert('db_name', 'schema_name', 'table_name', record_dict)

dbconn.update('db_name', 'schema_name', 'table_name', 'row_pk_value', record_dict)

dbconn.upsert('db_name', 'schema_name', 'table_name', 'row_pk_value', record_dict)

dbconn.write('schema_name', 'table_name', [{'id': 1, 'name': 'John'}], pk='id')

# Write with a schema provided
object_schema = {
	'properties': {
		'id': {'type': 'integer'},
		'name': {'type': 'string'}
	}
}
dbconn.write('schema_name', 'table_name', [{'id': 1, 'name': 'John'}], object_schema, pk='id')

dbconn.execute('db_name', query='TRUNCATE TABLE schema_name.table_name')

trinoconn = pq.trinoconnect()

rows = trinoconn.fetch('db_or_catalog_name', 'schema_name', 'table_name')

# Read data using custom query
rows = trinoconn.fetch('db_name', query='SELECT * FROM catalog.schema.table_name')

# Get result as a data frame
df = trinoconn.fetch('db_name', 'schema_name', 'table_name', df=True)

apiconn = pq.connect('connection_name')

apiconn.add('object_type', object)

apiconn.update('object_type', object)

apiconn.upsert('contact', contact, searchfield='email', searchterm='john@doe.com')

apiconn.delete('object_type', id=1)

apiconn.get('object_type', id=1)

apiconn.findone('object_type', searchfield='email', searchterm='john@doe.com')

apiconn.list('object_type')

apiconn.copy('object_type', id=1)

apiconn.rename('object_type', id=1, name='new name')

apiconn.apicall('path', method='get', body={})

sftpconn = pq.sftpconnect('connection_name')

result = sftpconn.dir('.')

result = sftpconn.read_file('./file.json')

pq.get_state()

pq.set_state(new_state)

databases = pq.list_databases()
st.json(databases)

schemas = pq.list_schemas(db_name)
st.json(schemas)

databases = pq.list_databases()
for db in databases:
	for table in db['tables']:
		st.text(table['name'])

scripts = pq.list_scripts()
st.json(scripts)

db_meta_data = pq.get_database(database_id=37734)
st.json(db_meta_data)

schema_meta_data = pq.get_schema(schema_id=37734)
st.json(db_meta_data)

table_meta_data = pq.get_table(table_id=37734)
st.json(table_meta_data)

script_meta_data = pq.get_script(script_id=745)
st.json(script_meta_data)

db_meta_data = pq.update_database(database_id=1603, description="This is a database description.")
st.json(db_meta_data)

schema_meta_data = pq.update_schema(schema_id=1603, name="my_schema")
st.json(schema_meta_data)

table_meta_data = pq.update_table(table_id=37734, description="This is a table description.")
st.json(table_meta_data)

field_meta_data = pq.update_field(field_id=829137, description="This is a field description.")
st.json(field_meta_data)

script_meta_data = pq.update_script(script_id=745, name="Create and hydrate subaccount")
st.json(script_meta_data)

pq.run_pipeline(connection_name = 'Hubspot', is_async = False)

pq.run_pipeline(connection_name = 'Hubspot', tables = "deals")

pq.run_pipeline(connection_name = 'Hubspot', tables = ["deals", "contacts"])

pq.get_pipeline_runs(source_connection_id)

pq.get_pipeline_logs(pipeline_run_id)

pq.refresh_database(database_name = 'dw_12345', is_async = False)

pq.refresh_schema(database_name = 'dw_12345', schema_name = 'CRM', is_async = False)

pq.refresh_table(database_name = 'dw_12345', schema_name = 'CRM', table_name = 'customers', is_async = False)