From d26f23f26fdbc55fc7532a3faccf9d8cb140d13c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Michael=20B=C3=BCcker?= <mb366767@fh-muenster.de>
Date: Thu, 28 Sep 2023 15:52:12 +0000
Subject: [PATCH] Add new file
---
db_load.py | 18 ++++++++++++++++++
1 file changed, 18 insertions(+)
create mode 100644 db_load.py
diff --git a/db_load.py b/db_load.py
new file mode 100644
index 0000000..dbf60ca
--- /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)
--
GitLab