Update task_operations.py

task_operations.py

@@ -14,7 +14,6 @@ class TaskManager:
         self.create_table()
 
     def create_table(self):
-        # Create tasks table with unique id
         query = """
         CREATE TABLE IF NOT EXISTS tasks (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -29,14 +28,17 @@ class TaskManager:
         self.conn.commit()
 
     def load_tasks(self):
-        # Load tasks from the database
         query = "SELECT * FROM tasks;"
         df = pd.read_sql_query(query, self.conn)
-        
+
+        # Debug: Show the columns of the DataFrame immediately after loading
+        print("Loaded columns before processing:", df.columns.tolist())
+
         if not df.empty:
             # Ensure 'Task Time' is in datetime format and id is an integer
             df['Task Time'] = pd.to_datetime(df['task_time'])  # Convert task_time to datetime
             df['id'] = df['id'].astype(int)  # Ensure id is an integer
+
             df = df.rename(columns={
                 'task_name': 'Task Name', 
                 'task_time': 'Task Time',
@@ -45,12 +47,15 @@ class TaskManager:
                 'category': 'Category', 
                 'id': 'Task ID'  # Rename id to 'Task ID'
             })
+
+            # Debug: Show the columns after renaming
+            print("Loaded columns after processing:", df.columns.tolist())
+
             return df.to_dict(orient='records')
         else:
             return []
 
     def save_task(self, task):
-        # Save task into the database
         query = """
         INSERT INTO tasks (task_name, task_time, task_duration_hours, task_duration_minutes, category)
         VALUES (?, ?, ?, ?, ?);
@@ -72,12 +77,22 @@ class TaskManager:
         st.session_state.tasks.append(task_entry)
 
     def delete_task_by_id(self, task_id):
-        # Delete task by its ID
         query = "DELETE FROM tasks WHERE id = ?;"
         self.conn.execute(query, (task_id,))
         self.conn.commit()
         return True
 
+    def update_task(self, task_id, task_name, task_time, task_duration_hours, task_duration_minutes, task_category):
+        task_time_full = datetime.combine(datetime.today(), task_time)
+        query = """
+        UPDATE tasks
+        SET task_name = ?, task_time = ?, task_duration_hours = ?, task_duration_minutes = ?, category = ?
+        WHERE id = ?;
+        """
+        self.conn.execute(query, (task_name, task_time_full, task_duration_hours, task_duration_minutes, task_category, task_id))
+        self.conn.commit()
+        return True
+
     def generate_report(self, timeframe):
         df = pd.DataFrame(self.load_tasks())
         if df.empty: