Upload utilities.py

utilities.py

@@ -24,7 +24,8 @@ import sqlite3
 import datetime
 from scenario import numerize
 import psycopg2
-# 
+
+#
 import re
 import bcrypt
 import os
@@ -78,7 +79,7 @@ color_palette = [
 
 
 CURRENCY_INDICATOR = "$"
-db_cred=None
+db_cred = None
 # database_file = r"DB/User.db"
 
 # conn = sqlite3.connect(database_file, check_same_thread=False)  # connection with sql db
@@ -145,15 +146,11 @@ db_cred=None
 #         conn.close()
 
 
-db_path=os.path.join('imp_db.db')
+db_path = os.path.join("imp_db.db")
+
 
 def query_excecuter_postgres(
-    query,
-    db_path=r'db\imp_db.db',
-    params=None,
-    insert=True,
-    insert_retrieve=False,
-    db_cred=None
+    query, db_path, params=None, insert=True, insert_retrieve=False, db_cred=None
 ):
     """
     Executes a SQL query on a SQLite database, handling both insert and select operations.
@@ -167,8 +164,13 @@ def query_excecuter_postgres(
 
     """
     try:
+        # Construct a cross-platform path to the database
+        db_dir = os.path.join("db")
+        os.makedirs(db_dir, exist_ok=True)  # Make sure the directory exists
+        db_path = os.path.join(db_dir, "imp_db.db")
+
         # Establish connection to the SQLite database
-        conn = sqlite3.connect(r'db\imp_db.db')
+        conn = sqlite3.connect(db_path)
     except sqlite3.Error as e:
         st.warning(f"Unable to connect to the SQLite database: {e}")
         st.stop()
@@ -179,7 +181,7 @@ def query_excecuter_postgres(
     # Prepare the query with proper placeholders
     if params:
         # Handle the `IN (?)` clause dynamically
-        query = query.replace('IN (?)', f"IN ({','.join(['?' for _ in params])})")
+        query = query.replace("IN (?)", f"IN ({','.join(['?' for _ in params])})")
         c.execute(query, params)
     else:
         c.execute(query)
@@ -203,9 +205,6 @@ def query_excecuter_postgres(
         conn.close()
 
 
-
-
-
 def update_summary_df():
     """
     Updates the 'project_summary_df' in the session state with the latest project