Merge pull request #450 from guidow/log_assign_change

Log changes of agent assignments to tasks
pyfarm · May 9, 2015 · 4158926 · 4158926
2 parents af2271b + 6d965fa
commit 4158926
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pyfarm/models/task.py b/pyfarm/models/task.py
@@ -112,6 +112,11 @@ def increment_attempts(target, new_value, old_value, initiator):
         if new_value is not None and new_value != old_value:
             target.attempts += 1
 
+    @staticmethod
+    def log_assign_change(target, new_value, old_value, initiator):
+        logger.debug("Agent change for task %s: old %s new: %s",
+                     target.id, old_value, new_value)
+
     @staticmethod
     def update_failures(target, new_value, old_value, initiator):
         if new_value == WorkState.FAILED and new_value != old_value:
@@ -152,5 +157,6 @@ def clear_error_state(target, new_value, old_value, initiator):
 event.listen(Task.state, "set", Task.update_failures)
 event.listen(Task.state, "set", Task.set_progress_on_success)
 event.listen(Task.agent_id, "set", Task.increment_attempts)
+event.listen(Task.agent_id, "set", Task.log_assign_change)
 event.listen(Task.state, "set", Task.reset_agent_if_failed_and_retry,
              retval=True)