diff --git a/app/etl/load.py b/app/etl/load.py index 4ae4589..e7ef3d7 100644 --- a/app/etl/load.py +++ b/app/etl/load.py @@ -23,6 +23,7 @@ def load(df: pd.DataFrame, db_path: str = "etl_data.db", table_name: str = "proc if db_dir and not os.path.exists(db_dir): os.makedirs(db_dir) conn = None + cursor = None try: # Connect to database conn = sqlite3.connect(db_path) @@ -59,5 +60,9 @@ def load(df: pd.DataFrame, db_path: str = "etl_data.db", table_name: str = "proc if conn: conn.rollback() finally: + # Ensure cursor is closed + if cursor: + cursor.close() + # Ensure database connection is properly closed if conn: conn.close()