add feature to update duration

flomo/cli.py

@@ -132,13 +132,14 @@ def delete(session_ids: Tuple):
 @click.argument("session_id")
 @click.option("-t", "--tag", help="Session Tag")
 @click.option("-n", "--name", help="Session Name")
-def update(session_id: str, tag: str | None, name: str | None):
+@click.option("-d", "--duration", help="Total Duration")
+def update(session_id: str, tag: str | None, name: str | None, duration: str | None):
     """
     Update session data.
     """
     try:
         db = tracker.Tracker()
-        db.update_session(session_id, tag, name)
+        db.update_session(session_id, tag, name, duration)
         db.conn.close()
     except (
         errors.DBFileNotFoundError,

flomo/tracker.py

@@ -92,7 +92,7 @@ def delete_session(self, session_ids: Tuple[str] | Tuple):
         )
         self.conn.commit()
 
-    def update_session(self, session_id: str, tag: str | None, name: str | None):
+    def update_session(self, session_id: str, tag: str | None, name: str | None, duration: str | None):
         if not self.get_session(session_id):
             raise errors.NoSessionError(session_id)
 
@@ -106,6 +106,11 @@ def update_session(self, session_id: str, tag: str | None, name: str | None):
                 "UPDATE sessions SET name = ? WHERE id = ?", (name, session_id)
             )
             print(f'Name updated to "{name}" for session {session_id}')
+        if duration:
+            self.cursor.execute(
+                "UPDATE sessions SET total_time = ? WHERE id = ?", (duration, session_id)
+            )
+            print(f'Duration updated to "{duration}" for session {session_id}')
         self.conn.commit()