src/mdis/visitor.py

   1 import contextlib as _contextlib
   2
   3 from . import core as _core
   4 from . import dispatcher as _dispatcher
   5
   6
   7 class VisitorMeta(_dispatcher.DispatcherMeta):
   8     pass
   9
  10
  11 class Visitor(_dispatcher.Dispatcher, metaclass=VisitorMeta):
  12     @_contextlib.contextmanager
  13     @_core.hook(object)
  14     def dispatch_object(self, instance):
  15         yield instance