Manage file added again to same session, data is updated.

See #PARAF-359

Author: gbastien

CHANGES.rst

@@ -29,6 +29,8 @@ Changelog
   [chris-adam]
 - Added external watchers for esign sessions.
   [chris-adam, sgeulette]
+- Manage file added again to same session, data is updated.
+  [gbastien]
 
 1.0a2 (2026-02-06)
 ------------------

src/imio/esign/tests/test_utils.py

@@ -450,6 +450,30 @@ def test_add_files_with_duplicate_filenames(self):
         self.assertIn("same_filename-1.pdf", filenames)
         self.assertIn("same_filename-2.pdf", filenames)
 
+    def test_add_files_already_exist_is_updated(self):
+        """When adding a file to the same session, it is updated."""
+        annot = get_session_annotation()
+        self.assertEqual(len(annot["sessions"]), 0)
+
+        signers = [
+            ("user1", "user1@sign.com", "User 1", "Position 1"),
+        ]
+
+        annex_uid = self.uids[0]
+        annex = api.content.get(UID=annex_uid)
+
+        sid, session = add_files_to_session(signers, (annex_uid,))
+        self.assertEqual(len(session["files"]), 1)
+        self.assertEqual(session["files"][0]["filename"], "annex0.pdf")
+        self.assertEqual(session["files"][0]["title"], "Annex 0")
+        # edit annex and add again, still one annex in session and data are updated
+        annex.file.filename = u"new_filename.pdf"
+        annex.setTitle('New title')
+        sid, session = add_files_to_session(signers, (annex_uid,))
+        self.assertEqual(len(session["files"]), 1)
+        self.assertEqual(session["files"][0]["filename"], "new_filename.pdf")
+        self.assertEqual(session["files"][0]["title"], "New title")
+
     def test_remove_context_from_session(self):
         """Test removing a context from a session."""
         annot = get_session_annotation()

src/imio/esign/utils.py

@@ -83,6 +83,20 @@ def add_files_to_session(
         annex = uuidToObject(uuid=uid, unrestricted=True)
         context_uid_provider = getAdapter(annex, IContextUidProvider)
         context_uid = context_uid_provider.get_context_uid()
+        # update data if adding same file to same session
+        if annot['uids'].get(uid, -1) == session_id:
+            logger.info('File with UID %s is already in session_id %s and data were updated!', uid, session_id)
+            # remove old data from "files", existing_files and c_uids
+            #data_index = [file_data['uid'] for file_data in session["files"]].index(uid)
+            #old_filename = path.splitext([fn for fn in session["files"] if fn['uid'] == uid][0]['filename'])[0]
+            #existing_files.remove(old_filename)
+            #annot["c_uids"][context_uid].remove(uid)
+            #del session["files"][data_index]
+            old_filename = path.splitext([fn for fn in session["files"]
+                                          if fn['uid'] == uid][0]['filename'])[0]
+            existing_files.remove(old_filename)
+            remove_files_from_session([uid], remove_empty_session=False)
+
         filename, ext = path.splitext(annex.file.filename or "no_filename.pdf")
         new_filename = get_correct_id(existing_files, filename)
         session["files"].append(
@@ -353,7 +367,7 @@ def remove_context_from_session(context_uids):
         remove_files_from_session(list(c_uids[context_uid]))
 
 
-def remove_files_from_session(files_uids):
+def remove_files_from_session(files_uids, remove_empty_session=True):
     """Remove files from their corresponding sessions.
 
     :param files_uids: list of file UIDs to remove
@@ -386,7 +400,7 @@ def remove_files_from_session(files_uids):
             continue
 
         del session["files"][i]
-        if not session["files"]:
+        if not session["files"] and remove_empty_session:
             del sessions[session_id]
         else:
             session["last_update"] = datetime.now()

test-4.3.cfg

@@ -150,3 +150,8 @@ Pygments = 2.5.2
 # beautifulsoup
 soupsieve = 1.9.2
 backports.functools-lru-cache = 1.5
+
+# Required by:
+# mock
+mock = 3.0.5
+funcsigs = 1.0.2