visitor: support additional arguments

[mdis.git] / src / mdis / visitor.py

diff --git a/src/mdis/visitor.py b/src/mdis/visitor.py
index ff2617297877fb0fbfab4683e7f95339b8f38c96..d7422681592dfba7d4ba34c40c97dd292e567748 100644 (file)
--- a/src/mdis/visitor.py
+++ b/src/mdis/visitor.py
@@ -15,12 +15,12 @@ class VisitorMeta(dispatcher.DispatcherMeta):
 
 class Visitor(dispatcher.Dispatcher, metaclass=VisitorMeta):
     @dispatcher.Hook(object)
-    def dispatch_object(self, node):
+    def dispatch_object(self, node, *arguments):
         return node
 
 
 class ContextVisitor(Visitor):
     @dispatcher.Hook(object)
     @contextlib.contextmanager
-    def dispatch_object(self, node):
-        yield super().__call__(node)
+    def dispatch_object(self, node, *arguments):
+        yield super().__call__(node, *arguments)