Ticket #130: send-count.diff

foolscap/call.py

@@ -4 +4 @@
 
 from foolscap import copyable, slicer, tokens
 from foolscap.copyable import AttributeDictConstraint
-from foolscap.constraint import ByteStringConstraint
+from foolscap.constraint import ByteStringConstraint, IntegerConstraint
 from foolscap.slicers.list import ListConstraint
 from tokens import BananaError, Violation
 from foolscap.util import AsyncAND
@@ -23 +23 @@
                  ('value', ByteStringConstraint(1000)),
                  ('traceback', ByteStringConstraint(2000)),
                  ('parents', ListConstraint(ByteStringConstraint(200))),
+                 ('count', IntegerConstraint()),
                  ]
         AttributeDictConstraint.__init__(self, *attrs)
 
@@ -801 +802 @@
         # Exception.__repr__ returns.
         state['value'] = truncate(state['value'], 1000)
         state['type'] = truncate(state['type'], 200)
+        if hasattr(obj, "count"):
+            state["count"] = obj.count
 
         if broker.unsafeTracebacks:
             if isinstance(obj.type, str):