pset.unshare_pset - fix bug where it would leaving an orphaned pset

It was leaving an orphaned pset if 'products' was a list of all elements pset is assigned to. Then it would create a new pset for each element leaving original pset orphaned.
This commit is contained in:
Andrej730
2024-09-05 18:40:53 +05:00
parent ffb2ea2be3
commit 6ebec7e0e5
3 changed files with 43 additions and 2 deletions
@@ -57,6 +57,22 @@ def unshare_pset(
assert new_pset != pset
assert ifcopenshell.util.element.get_elements_by_pset(new_pset) == {element2}
"""
if not products:
raise Exception("No products provided.")
# If pset has no other elements besides the provided products,
# then we skip the first product, so it won't get additional pset copy
# leaving the original pset orphaned.
pset_elements = ifcopenshell.util.element.get_elements_by_pset(pset)
products_original = products
if set(products) == pset_elements:
products = products[1:]
if not products:
raise Exception(f"Provided product is the only element to which pset is assigned: {products_original[0]}.")
products_occurrences: set[ifcopenshell.entity_instance] = set()
products_types: set[ifcopenshell.entity_instance] = set()
for product in products: