A warning is raised when adding a hook without attached module (for unload)
This commit is contained in:
parent
2fd395f4bc
commit
6c7be6761e
9
hooks.py
9
hooks.py
@ -44,11 +44,16 @@ class MessagesHook:
|
|||||||
|
|
||||||
def add_hook(self, store, hook, module_src=None):
|
def add_hook(self, store, hook, module_src=None):
|
||||||
"""Insert in the right place a hook into the given store"""
|
"""Insert in the right place a hook into the given store"""
|
||||||
|
if module_src is None:
|
||||||
|
print ("\033[1;35mWarning:\033[0m No source module was passed to "
|
||||||
|
"add_hook function, please fix it in order to be "
|
||||||
|
"compatible with unload feature")
|
||||||
|
|
||||||
if store in self.context.hooks_cache:
|
if store in self.context.hooks_cache:
|
||||||
del self.context.hooks_cache[store]
|
del self.context.hooks_cache[store]
|
||||||
|
|
||||||
if not hasattr(self, store):
|
if not hasattr(self, store):
|
||||||
print ("Warning: unrecognized hook store type")
|
print ("\033[1;35mWarning:\033[0m unrecognized hook store type")
|
||||||
return
|
return
|
||||||
attr = getattr(self, store)
|
attr = getattr(self, store)
|
||||||
|
|
||||||
@ -59,7 +64,7 @@ class MessagesHook:
|
|||||||
elif isinstance(attr, list):
|
elif isinstance(attr, list):
|
||||||
attr.append(hook)
|
attr.append(hook)
|
||||||
else:
|
else:
|
||||||
print ("Warning: unrecognized hook store type")
|
print ("\033[1;32mWarning:\033[0m unrecognized hook store type")
|
||||||
return
|
return
|
||||||
if module_src is not None:
|
if module_src is not None:
|
||||||
module_src.REGISTERED_HOOKS.append((store, hook))
|
module_src.REGISTERED_HOOKS.append((store, hook))
|
||||||
|
Loading…
Reference in New Issue
Block a user