From 0c7ecd383bfeffa123def5e1715d56bb8f1fb0d0 Mon Sep 17 00:00:00 2001 From: Tyler Goodlet Date: Tue, 17 Apr 2018 17:17:08 -0400 Subject: [PATCH] Monkey patch broker mods with a name attr --- piker/brokers/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/piker/brokers/__init__.py b/piker/brokers/__init__.py index 8ae3f1dc..2e12b3ea 100644 --- a/piker/brokers/__init__.py +++ b/piker/brokers/__init__.py @@ -13,7 +13,10 @@ __brokers__ = [ def get_brokermod(brokername: str) -> ModuleType: """Return the imported broker module by name. """ - return import_module('.' + brokername, 'piker.brokers') + module = import_module('.' + brokername, 'piker.brokers') + # we only allows monkeys because it's for internal keying + module.name = module.__name__.split('.')[-1] + return module def iter_brokermods():