Update functions

python_startup/01_load_db.py

 import os
 from dotenv import load_dotenv
-import psycopg2
 import pandas as pd
+import sqlalchemy
+from urllib.parse import quote_plus
 
 load_dotenv("/usr/src/app/.env")
 
+class Database:
+    def __init__(self, db_name):
+        self.conn = db_connect(db_name)
+
+    def __getattr__(self, table_name):
+        return pd.read_sql_table(table_name, self.conn)
+
 def db_connect(db_name):
-    conn = psycopg2.connect(
-    host=os.getenv("DB_HOST"),
-    database=db_name,
-    user=os.getenv("DB_USER"),
-    password=os.getenv("DB_PASSWORD"))
-    return(conn)
+    hostname = "wi-sql.fh-muenster.de"
+    user = 'digibim'
+    password = quote_plus("Di&i$IM0815")
+    conn = sqlalchemy.create_engine(f'postgresql+psycopg2://{user}:{password}@{hostname}/{db_name}')
+    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)
+    dat = pd.read_sql_table(table_name, conn)
+    return dat
+
+def get_all_tables(db_name):
+    db_obj = Database(db_name)
+    return db_obj
+

requirements.txt

@@ -3,3 +3,4 @@ jupyterlab-gitlab==3.0.0
 psycopg2-binary
 psycopg2
 python-dotenv
+sqlalchemy
\ No newline at end of file