Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-21 10:23:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix sequence.get_related_products bug after 845d772

Browse Source
This commit is contained in:
Andrej730
2024-09-10 17:27:12 +05:00
parent 2d0b262351
commit 6fd4d8b1ba
1 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/util/sequence.py
+5 -7
View File
@@ -459,12 +459,9 @@ def get_related_products(
"""Gets the related products being output by a task """Gets the related products being output by a task
:param relating_product: One of the products already output by the task. :param relating_product: One of the products already output by the task.
:type relating_product: ifcopenshell.entity_instance, optional
:param related_object: The IfcTask that you want to get all the related :param related_object: The IfcTask that you want to get all the related
products for. products for.
:type related_object: ifcopenshell.entity_instance, optional
:return: A set of IfcProducts output by the IfcTask. :return: A set of IfcProducts output by the IfcTask.
:rtype: set[ifcopenshell.entity_instance]
Example: Example:
@@ -489,17 +486,18 @@ def get_related_products(
products = ifcopenshell.util.sequence.get_related_products(related_object=task) products = ifcopenshell.util.sequence.get_related_products(related_object=task)
""" """
assert relating_product or related_object, "Either relating_product or related_object must be provided."
products = set() products = set()
related_object = None if not related_object and relating_product:
if related_object:
related_object = related_object
elif relating_product:
for reference in relating_product.ReferencedBy: for reference in relating_product.ReferencedBy:
if reference.is_a("IfcRelAssignsToProduct"): if reference.is_a("IfcRelAssignsToProduct"):
related_object = reference.RelatedObjects[0] related_object = reference.RelatedObjects[0]
if related_object: if related_object:
assignments = related_object.HasAssignments assignments = related_object.HasAssignments
for assignment in assignments: for assignment in assignments:
if assignment.is_a("IfcRelAssignsToProduct"): if assignment.is_a("IfcRelAssignsToProduct"):
products.add(assignment.RelatingProduct.id()) products.add(assignment.RelatingProduct.id())
return products return products