Insert record

insert_record.py

import bitdotio


columns = ['datetime', 'location', 'sensor_id', 'pm_2_5', 'pm_10']
record_list = [record[col] for col in columns]

bit = bitdotio.bitdotio(BITDOTIO_API_KEY)
# Replace the line below with your schema-qualified table name
qualified_table = '"air_quality_log_test/air_quality"."pm_measurements"'


def execute_sql(bitdotio, sql, params=None):
    """Run arbitrary sql with parameters on bit.io."""
    try:
        conn = bitdotio.get_connection()
        cur = conn.cursor()
        cur.execute(sql, params)
        cur.close()
        conn.commit()
    except Exception as e:
        logger.exception('A query error occurred')
        raise e
    finally:
        if conn is not None:
            conn.close()

            
def insert_record(bitdotio, record, qualified_table):
    """Inserts a single sensor measurement record.
    """
    sql = f'INSERT INTO {qualified_table} '
    sql += 'VALUES (' + ', '.join(['%s'] * len(record)) + ');'
    execute_sql(bitdotio, sql, record)
    
    
insert_record(bit, record_list, qualified_table)