diff --git a/python_startup/01_load_db.py b/python_startup/01_load_db.py index e51eac923fa6fc31eb8d87b3aa5be42bf51f4e70..077ad64c0aea674a07ddca341233643e61ebe7e6 100644 --- a/python_startup/01_load_db.py +++ b/python_startup/01_load_db.py @@ -12,6 +12,10 @@ class Database: def __getattr__(self, table_name): return pd.read_sql_table(table_name, self.conn) + def list_tables(self): + table_names = self.conn.table_names() + return table_names + def db_connect(db_name): hostname=os.getenv("DB_HOST") user=os.getenv("DB_USER")