From acdd0659b31a9e126a4f5f689a91b1c420ed1e76 Mon Sep 17 00:00:00 2001
From: indar suthar <indrasuthar14@gmail.com>
Date: Sun, 9 Nov 2025 03:48:08 +0530
Subject: [PATCH] feat: ensure database connection and cursor are properly
 closed

---
 app/etl/load.py | 5 +++++
 1 file changed, 5 insertions(+)

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()