Guard against double init hooks

This commit is contained in:
Florian Bruhin 2018-12-10 15:24:15 +01:00
parent 93a36c8572
commit 6001640a8a

View File

@ -33,5 +33,7 @@ class init: # noqa: N801,N806 pylint: disable=invalid-name
def __call__(self, func: typing.Callable) -> typing.Callable:
module = importlib.import_module(func.__module__)
info = loader.add_module_info(module)
if info.init_hook is not None:
raise ValueError("init hook is already registered!")
info.init_hook = func
return func