datastore: override put method to add changes to db history

inveniosoftware · Nov 8, 2023 · 6595d6b · 6595d6b
1 parent 05657e0
commit 6595d6b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/invenio_accounts/datastore.py b/invenio_accounts/datastore.py
@@ -37,6 +37,12 @@ def commit(self):
         datastore_post_commit.send(session=self.db.session)
         current_db_change_history.clear_dirty_sets(self.db.session)
 
+    def put(self, model):
+        """Put a user to its session."""
+        res = super().put(model)
+        self.mark_changed(id(self.db.session), uid=model.id)
+        return res
+
     def mark_changed(self, sid, uid=None, rid=None):
         """Save a user to the changed history."""
         if uid: