import os
import pandas.io.sql as sqlio
import psycopg2

from dotenv import load_dotenv, find_dotenv

load_dotenv(find_dotenv())

SQL_USER = os.environ.get('SQL_USER')
SQL_PASS = os.environ.get('SQL_PASS')
SQL_HOST = os.environ.get('SQL_HOST')
SQL_PORT = os.environ.get('SQL_PORT')
SQL_DB = os.environ.get('SQL_DB')


conn = psycopg2.connect(
    dbname=SQL_DB,
    user=SQL_USER,
    password=SQL_PASS,
    port=SQL_PORT,
    host=SQL_HOST
)


SQL_QUERY = "SELECT * FROM test_table WHERE id = ANY(%s)"
test_ids = [1, 2, 3]

result_df = sqlio.read_sql_query(SQL_QUERY, params=(test_ids,), conn)