Pass in commands to get df rather than getting db conn

src/dbt_data_client.py

@@ -23,6 +23,11 @@ def run_dbt_build():
     st.text("\n".join(stdout.decode().split("\n")[1:][:-1]))
 
 
-def get_current_tables(duckdb_conn=get_db_conn()) -> list[str]:
-    current_tables_df = duckdb_conn.sql("SHOW TABLES").df()
+def execute_db_command_df(db_command: str):
+    with get_db_conn() as db_conn:
+        return db_conn.sql(db_command).df()
+
+
+def get_current_tables() -> list[str]:
+    current_tables_df = execute_db_command_df("SHOW TABLES")
     return current_tables_df["name"].tolist()

src/pages/98_Load_Data copy.py

@@ -3,7 +3,7 @@ import streamlit as st
 from config import DEFAULT_ICON
 from shared_page import common_page_config
 
-from dbt_data_client import get_db_conn, run_dbt_build, get_current_tables
+from dbt_data_client import run_dbt_build, get_current_tables, execute_db_command_df
 
 
 def get_page():
@@ -12,15 +12,13 @@ def get_page():
     common_page_config()
     st.title(page_title)
 
-    with get_db_conn() as duckdb_conn:
-        current_tables_list = get_current_tables(duckdb_conn)
-
     if st.button("Refresh Data"):
         run_dbt_build()
 
+    current_tables_list = get_current_tables()
+
     if selected_table := st.selectbox("Describe a table:", current_tables_list, index=0):
-        with get_db_conn() as duckdb_conn:
-            describe_df = duckdb_conn.sql(f"DESCRIBE {selected_table}").df()
+        describe_df = execute_db_command_df(f"DESCRIBE {selected_table}")
         st.dataframe(
             describe_df,
             hide_index=True,
@@ -28,8 +26,7 @@ def get_page():
         )
 
         if st.checkbox("Explore data"):
-            with get_db_conn() as duckdb_conn:
-                st.dataframe(duckdb_conn.sql(f"SELECT * FROM {selected_table} LIMIT 50").df())
+            st.dataframe(execute_db_command_df(f"SELECT * FROM {selected_table} LIMIT 50"))
 
 
 if __name__ == "__main__":