Fix #6640. Fix invalid association of annotations to grids.

There was already some half written code with this intent it seems, so I
just finished it up.

src/ifcopenshell-python/ifcopenshell/api/drawing/assign_product.py

@@ -59,49 +59,47 @@ def assign_product(
         ifcopenshell.api.drawing.assign_product(model,
             relating_product=furniture, related_object=annotation)
     """
-    is_grid_axis = relating_product.is_a("IfcGridAxis")
-
-    if is_grid_axis:
-        if related_object.HasAssignments:
-            for rel in related_object.HasAssignments:
-                if rel.is_a("IfcRelAssignsToProduct") and rel.Name == relating_product.AxisTag:
-                    return
-    elif related_object.HasAssignments:
-        for rel in related_object.HasAssignments:
-            if rel.is_a("IfcRelAssignsToProduct") and rel.RelatingProduct == relating_product:
-                return
-
-    referenced_by = None
-
-    if is_grid_axis:
+    if relating_product.is_a("IfcGridAxis"):
         axis = relating_product
         grid = None
         for attribute in ("PartOfW", "PartOfV", "PartOfU"):
             if getattr(axis, attribute, None):
                 grid = getattr(axis, attribute)[0]
-        for rel in grid.ReferencedBy:
-            if rel.Name == axis.AxisTag:
-                referenced_by = rel
                 break
-    elif relating_product.ReferencedBy:
+        for rel in grid.ReferencedBy or []:
+            if rel.Name == axis.AxisTag and related_object in rel.RelatedObjects:
+                return
+
+        return file.create_entity(
+            "IfcRelAssignsToProduct",
+            GlobalId=ifcopenshell.guid.new(),
+            Name=axis.AxisTag,
+            OwnerHistory=ifcopenshell.api.owner.create_owner_history(file),
+            RelatedObjects=[related_object],
+            RelatingProduct=grid,
+        )
+
+    if related_object.HasAssignments:
+        for rel in related_object.HasAssignments:
+            if rel.is_a("IfcRelAssignsToProduct") and rel.RelatingProduct == relating_product:
+                return
+
+    referenced_by = None
+    if relating_product.ReferencedBy:
         referenced_by = relating_product.ReferencedBy[0]
 
     if referenced_by:
         related_objects = list(referenced_by.RelatedObjects)
         related_objects.append(related_object)
         referenced_by.RelatedObjects = related_objects
-        ifcopenshell.api.owner.update_owner_history(file, **{"element": referenced_by})
+        ifcopenshell.api.owner.update_owner_history(file, element=referenced_by)
     else:
         referenced_by = file.create_entity(
             "IfcRelAssignsToProduct",
-            **{
-                "GlobalId": ifcopenshell.guid.new(),
-                "OwnerHistory": ifcopenshell.api.owner.create_owner_history(file),
-                "RelatedObjects": [related_object],
-                "RelatingProduct": relating_product,
-            },
+            GlobalId=ifcopenshell.guid.new(),
+            OwnerHistory=ifcopenshell.api.owner.create_owner_history(file),
+            RelatedObjects=[related_object],
+            RelatingProduct=relating_product,
         )
 
-    if is_grid_axis:
-        referenced_by.Name = axis.AxisTag
     return referenced_by

src/ifcopenshell-python/ifcopenshell/api/drawing/unassign_product.py

@@ -51,6 +51,14 @@ def unassign_product(
         ifcopenshell.api.drawing.unassign_product(model,
             relating_product=furniture, related_object=annotation)
     """
+    if relating_product.is_a("IfcGridAxis"):
+        grid = None
+        for attribute in ("PartOfW", "PartOfV", "PartOfU"):
+            if getattr(relating_product, attribute, None):
+                grid = getattr(relating_product, attribute)[0]
+                break
+        relating_product = grid
+
     for rel in related_object.HasAssignments or []:
         if not rel.is_a("IfcRelAssignsToProduct") or rel.RelatingProduct != relating_product:
             continue