import os
import pandas as pd
from dotenv import load_dotenv, find_dotenv
from sqlalchemy.engine import URL, create_engine

load_dotenv(find_dotenv())

url = URL.create(
    drivername="postgresql+psycopg2",
    username=os.getenv("DATABASE_USERNAME"),
    password=os.getenv("DATABASE_PASSWORD"),
    host=os.getenv("DATABASE_IP"),
    port=int(os.getenv("DATABASE_PORT")),
    database=os.getenv("DATABASE_NAME"),
)

engine = create_engine(url)

query = """SELECT title, genres FROM audiobookshelf.ao3;"""

data = pd.read_sql_query(query, engine)
print(data.head())

# cursor.execute(Query)                                          
# row = cursor.fetchone()                                                      
# while row is not None:
#     print(row)
#     row = cursor.fetchone()