diff --git a/db_load.py b/db_load.py new file mode 100644 index 0000000000000000000000000000000000000000..dbf60ca345383504817526fcbb4e7ee35e9978fd --- /dev/null +++ b/db_load.py @@ -0,0 +1,18 @@ +import os +from dotenv import load_dotenv +import psycopg2 +import pandas as pd + +def db_connect(db_name): + conn = psycopg2.connect( + host=os.getenv("DB_HOST"), + database=db_name, + user=os.getenv("DB_USER"), + password="Di&i$IM0815") + return(conn) + +def get_table(db_name, table_name): + conn = db_connect(db_name) + sql = f"select * from {table_name};" + dat = pd.io.sql.read_sql_query(sql, conn) + return(dat)