mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-19 14:41:25 +00:00
Allow global listeners (#1500)
* Allow global listeners Add "global" listeners when ifc_file is None, listeners are run whatever the file. This way allow to register listeners before a file exists, and "global" listeners. Add listener returns the ifc_key Remove listeners allow optional filtering by usecase_path / ifc_key.usecase_path. * Update __init__.py Fix was only removing the first listener found.
This commit is contained in:
@@ -14,6 +14,12 @@ def run(usecase_path, ifc_file=None, should_run_listeners=True, **settings):
|
|||||||
if should_run_listeners:
|
if should_run_listeners:
|
||||||
for listener in pre_listeners.get(".".join([ifc_key, usecase_path]), []):
|
for listener in pre_listeners.get(".".join([ifc_key, usecase_path]), []):
|
||||||
listener(usecase_path, ifc_file, **settings)
|
listener(usecase_path, ifc_file, **settings)
|
||||||
|
|
||||||
|
if None in registered_ifcs and ifc_key != registered_ifcs[None]:
|
||||||
|
global_key = registered_ifcs[None]
|
||||||
|
for listener in pre_listeners.get(".".join([global_key, usecase_path]), []):
|
||||||
|
listener(usecase_path, ifc_file, **settings)
|
||||||
|
|
||||||
importlib.import_module(f"ifcopenshell.api.{usecase_path}")
|
importlib.import_module(f"ifcopenshell.api.{usecase_path}")
|
||||||
module, usecase = usecase_path.split(".")
|
module, usecase = usecase_path.split(".")
|
||||||
usecase_class = getattr(getattr(getattr(ifcopenshell.api, module), usecase), "Usecase")
|
usecase_class = getattr(getattr(getattr(ifcopenshell.api, module), usecase), "Usecase")
|
||||||
@@ -26,34 +32,46 @@ def run(usecase_path, ifc_file=None, should_run_listeners=True, **settings):
|
|||||||
if should_run_listeners:
|
if should_run_listeners:
|
||||||
for listener in post_listeners.get(".".join([ifc_key, usecase_path]), []):
|
for listener in post_listeners.get(".".join([ifc_key, usecase_path]), []):
|
||||||
listener(usecase_path, ifc_file, **settings)
|
listener(usecase_path, ifc_file, **settings)
|
||||||
|
|
||||||
|
if None in registered_ifcs and ifc_key != registered_ifcs[None]:
|
||||||
|
global_key = registered_ifcs[None]
|
||||||
|
for listener in post_listeners.get(".".join([global_key, usecase_path]), []):
|
||||||
|
listener(usecase_path, ifc_file, **settings)
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
|
||||||
def add_pre_listener(usecase_path, ifc_file, callback):
|
def add_pre_listener(usecase_path, ifc_file, callback):
|
||||||
ifc_key = registered_ifcs.setdefault(ifc_file, ifcopenshell.guid.new())
|
ifc_key = registered_ifcs.setdefault(ifc_file, ifcopenshell.guid.new())
|
||||||
pre_listeners.setdefault(".".join([ifc_key, usecase_path]), set()).add(callback)
|
pre_listeners.setdefault(".".join([ifc_key, usecase_path]), set()).add(callback)
|
||||||
|
return ifc_key
|
||||||
|
|
||||||
|
|
||||||
def add_post_listener(usecase_path, ifc_file, callback):
|
def add_post_listener(usecase_path, ifc_file, callback):
|
||||||
ifc_key = registered_ifcs.setdefault(ifc_file, ifcopenshell.guid.new())
|
ifc_key = registered_ifcs.setdefault(ifc_file, ifcopenshell.guid.new())
|
||||||
post_listeners.setdefault(".".join([ifc_key, usecase_path]), set()).add(callback)
|
post_listeners.setdefault(".".join([ifc_key, usecase_path]), set()).add(callback)
|
||||||
|
return ifc_key
|
||||||
def remove_pre_listener(callback):
|
|
||||||
for ifc_key, callbacks in pre_listeners.items():
|
|
||||||
|
def remove_pre_listener(callback, usecase_path=""):
|
||||||
|
for listener_key, callbacks in pre_listeners.items():
|
||||||
|
if not listener_key.endswith(usecase_path):
|
||||||
|
continue
|
||||||
for fun in callbacks:
|
for fun in callbacks:
|
||||||
if fun == callback:
|
if fun == callback:
|
||||||
pre_listeners[ifc_key].remove(callback)
|
pre_listeners[listener_key].remove(callback)
|
||||||
return
|
|
||||||
|
|
||||||
def remove_post_listener(callback):
|
def remove_post_listener(callback, usecase_path=""):
|
||||||
for ifc_key, callbacks in post_listeners.items():
|
for listener_key, callbacks in post_listeners.items():
|
||||||
|
if not listener_key.endswith(usecase_path):
|
||||||
|
continue
|
||||||
for fun in callbacks:
|
for fun in callbacks:
|
||||||
if fun == callback:
|
if fun == callback:
|
||||||
post_listeners[ifc_key].remove(callback)
|
post_listeners[listener_key].remove(callback)
|
||||||
return
|
|
||||||
|
|
||||||
def remove_all_listeners():
|
def remove_all_listeners():
|
||||||
registered_ifcs.clear()
|
registered_ifcs.clear()
|
||||||
pre_listeners.clear()
|
pre_listeners.clear()
|
||||||
post_listeners.clear()
|
post_listeners.clear()
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user