fix(translator): specify the translator to apply for each callback

Author: Justineantoine

trame_server/state.py

@@ -294,13 +294,9 @@ def flush(self):
                 self._pushed_state.update(self._pending_update)
                 self._pending_update.clear()
 
-                reverse_translated_state = self.translator.reverse_translate_dict(
-                    self._pushed_state
-                )
-
                 # Execute state listeners
                 self._state_listeners.add_all(_keys)
-                for fn in self._state_listeners:
+                for fn, translator in self._state_listeners:
                     if isinstance(fn, weakref.WeakMethod):
                         callback = fn()
                         if callback is None:
@@ -312,6 +308,7 @@ def flush(self):
                         if not inspect.iscoroutinefunction(callback):
                             callback = reload(callback)
 
+                    reverse_translated_state = translator.reverse_translate_dict(self._pushed_state)
                     coroutine = callback(**reverse_translated_state)
                     if inspect.isawaitable(coroutine):
                         asynchronous.create_task(coroutine)
@@ -366,7 +363,7 @@ def register_change_callback(func):
                 if name not in self._change_callbacks:
                     self._change_callbacks[name] = []
 
-                self._change_callbacks[name].append(func)
+                self._change_callbacks[name].append((func, self._translator))
             return func
 
         return register_change_callback