Skip to content

Commit

Permalink
Make ui_handler publicly available again
Browse files Browse the repository at this point in the history
  • Loading branch information
mdickinson committed Jan 24, 2025
1 parent 70b9bbb commit bc9ca95
Showing 1 changed file with 10 additions and 8 deletions.
18 changes: 10 additions & 8 deletions traits/trait_notifiers.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,31 +29,33 @@

# The currently active handler for notifications that must be run on the UI
# thread, or None if no handler has been set.
_ui_handler = None
# Note: the Pyface library current accesses the `ui_handler` attribute
# directly, so we can't make it private yet.
ui_handler = None


def get_ui_handler():
"""
Return the current user interface thread handler.
"""
return _ui_handler
return ui_handler


def set_ui_handler(handler):
""" Sets up the user interface thread handler.
"""
global _ui_handler
global ui_handler

_ui_handler = handler
ui_handler = handler


def ui_dispatch(handler, *args, **kw):
if threading.current_thread() == threading.main_thread():
handler(*args, **kw)
elif _ui_handler is None:
elif ui_handler is None:
raise RuntimeError("no UI handler registered for dispatch='ui'")
else:
_ui_handler(handler, *args, **kw)
ui_handler(handler, *args, **kw)


class NotificationExceptionHandlerState(object):
Expand Down Expand Up @@ -616,10 +618,10 @@ class FastUITraitChangeNotifyWrapper(TraitChangeNotifyWrapper):
def dispatch(self, handler, *args):
if threading.current_thread() == threading.main_thread():
handler(*args)
elif _ui_handler is None:
elif ui_handler is None:
raise RuntimeError("no UI handler registered for dispatch='ui'")
else:
_ui_handler(handler, *args)
ui_handler(handler, *args)


class NewTraitChangeNotifyWrapper(TraitChangeNotifyWrapper):
Expand Down

0 comments on commit bc9ca95

Please sign in to comment.