Added event to make sure session data (size, title, filename) is updated when annex updated

Author: gbastien

src/imio/esign/configure.zcml

@@ -8,6 +8,7 @@
 
   <i18n:registerTranslations directory="locales" />
 
+  <include file="events.zcml" />
   <include file="permissions.zcml" />
 
   <include package=".browser" />

src/imio/esign/events.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+
+from imio.esign.utils import get_sessions_for
+
+
+def on_categorized_annex_updated(annex, event):
+    '''When an annex is modified, update check if need to update esign session.'''
+    old_values = event.old_values
+    # we are creating a new annex, not in a session
+    if not old_values:
+        return
+
+    sessions = get_sessions_for(event.parent.UID())
+    if not sessions:
+        return
+
+    new_values = event.new_values
+    # if something usefull changed, we will update the session
+    update = False
+    checked_keys = ['title', 'filesize', 'relative_url']
+    for checked_key in checked_keys:
+        if new_values[checked_key] != old_values[checked_key]:
+            update = True
+            break
+    if update:
+        annex_uid = annex.UID()
+        for session in sessions:
+            # size
+            size_diff = new_values['filesize'] - old_values['filesize']
+            session['size'] += size_diff
+            # title and filename
+            for file_data in session['files']:
+                if file_data['uid'] == annex_uid:
+                    file_data['title'] = new_values['title']
+                    file_data['filename'] = annex.file.filename

src/imio/esign/events.zcml

@@ -0,0 +1,8 @@
+<configure
+    xmlns="http://namespaces.zope.org/zope">
+
+ <subscriber for="imio.annex.content.annex.IAnnex
+                  collective.iconifiedcategory.interfaces.ICategorizedElementUpdatedEvent"
+             handler=".events.on_categorized_annex_updated" />
+
+</configure>

src/imio/esign/utils.py

@@ -352,6 +352,17 @@ def get_session_info(session_id, portal=None):
         return annot['sessions'][session_id]
 
 
+def get_sessions_for(context_uid):
+    """ """
+    sessions = []
+    annot = get_session_annotation()
+    for f_uid in annot["c_uids"].get(context_uid, []):
+        session_id = annot["uids"].get(f_uid)
+        if session_id is not None:
+            sessions.append(annot["sessions"].get(session_id))
+    return sessions
+
+
 def remove_context_from_session(context_uids):
     """Remove all files from a session that are linked to the given context UIDs.