diff --git a/src/blenderbim/blenderbim/bim/helper.py b/src/blenderbim/blenderbim/bim/helper.py
index f0dab98427..6b11944f87 100644
--- a/src/blenderbim/blenderbim/bim/helper.py
+++ b/src/blenderbim/blenderbim/bim/helper.py
@@ -38,11 +38,14 @@ def draw_attribute(attribute, layout, copy_operator=None):
if not value_name:
layout.label(text=attribute.name)
return
- layout.prop(
- attribute,
- value_name,
- text=attribute.name,
- )
+ if value_name == "enum_value":
+ prop_with_search(layout, attribute, "enum_value", text=attribute.name)
+ else:
+ layout.prop(
+ attribute,
+ value_name,
+ text=attribute.name,
+ )
if attribute.is_optional:
layout.prop(attribute, "is_null", icon="RADIOBUT_OFF" if attribute.is_null else "RADIOBUT_ON", text="")
if copy_operator:
@@ -105,6 +108,16 @@ def export_attributes(props, callback=None):
return attributes
+def prop_with_search(layout, data, prop_name, **kwargs):
+ # kwargs are layout.prop arguments (text, icon, etc.)
+ row = layout.row(align=True)
+ # Magick courtesy of https://blender.stackexchange.com/a/203443/86891
+ row.context_pointer_set(name="data", data=data)
+ row.prop(data, prop_name, **kwargs)
+ op = row.operator("bim.enum_property_search", text="", icon="VIEWZOOM")
+ op.prop_name = prop_name
+
+
class IfcHeaderExtractor:
def __init__(self, filepath: str):
self.filepath = filepath
diff --git a/src/blenderbim/blenderbim/bim/module/brick/ui.py b/src/blenderbim/blenderbim/bim/module/brick/ui.py
index f0b1cedce7..bb8898b2c0 100644
--- a/src/blenderbim/blenderbim/bim/module/brick/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/brick/ui.py
@@ -18,7 +18,7 @@
import blenderbim.tool as tool
from bpy.types import Panel, UIList
-from blenderbim.bim.ui import prop_with_search
+from blenderbim.bim.helper import prop_with_search
from blenderbim.bim.module.brick.data import BrickschemaData, BrickschemaReferencesData
diff --git a/src/blenderbim/blenderbim/bim/module/geometry/ui.py b/src/blenderbim/blenderbim/bim/module/geometry/ui.py
index 4ce050ffb9..5f7c918de0 100644
--- a/src/blenderbim/blenderbim/bim/module/geometry/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/geometry/ui.py
@@ -19,7 +19,7 @@
import bpy
from bpy.types import Panel
from blenderbim.bim.ifc import IfcStore
-from blenderbim.bim.ui import prop_with_search
+from blenderbim.bim.helper import prop_with_search
from blenderbim.bim.module.geometry.data import RepresentationsData, DerivedPlacementsData
diff --git a/src/blenderbim/blenderbim/bim/module/material/ui.py b/src/blenderbim/blenderbim/bim/module/material/ui.py
index f657ab0a40..cddd181189 100644
--- a/src/blenderbim/blenderbim/bim/module/material/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/material/ui.py
@@ -22,7 +22,7 @@ from ifcopenshell.api.material.data import Data
from ifcopenshell.api.profile.data import Data as ProfileData
from blenderbim.bim.ifc import IfcStore
from blenderbim.bim.helper import draw_attributes
-from blenderbim.bim.ui import prop_with_search
+from blenderbim.bim.helper import prop_with_search
from blenderbim.bim.module.material.data import MaterialsData, ObjectMaterialData
diff --git a/src/blenderbim/blenderbim/bim/module/patch/ui.py b/src/blenderbim/blenderbim/bim/module/patch/ui.py
index 9d58971d7d..9e7a241e51 100644
--- a/src/blenderbim/blenderbim/bim/module/patch/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/patch/ui.py
@@ -16,7 +16,7 @@
# You should have received a copy of the GNU General Public License
# along with BlenderBIM Add-on. If not, see .
-from blenderbim.bim.ui import prop_with_search
+from blenderbim.bim.helper import prop_with_search
import bpy
from blenderbim.bim.helper import draw_attributes
diff --git a/src/blenderbim/blenderbim/bim/module/project/ui.py b/src/blenderbim/blenderbim/bim/module/project/ui.py
index 8a0a21f109..4aff3bb1d3 100644
--- a/src/blenderbim/blenderbim/bim/module/project/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/project/ui.py
@@ -17,7 +17,7 @@
# along with BlenderBIM Add-on. If not, see .
import os
-from blenderbim.bim.ui import prop_with_search
+from blenderbim.bim.helper import prop_with_search
from bpy.types import Panel, UIList
from blenderbim.bim.ifc import IfcStore
from blenderbim.bim.module.project.data import ProjectData
diff --git a/src/blenderbim/blenderbim/bim/module/pset/ui.py b/src/blenderbim/blenderbim/bim/module/pset/ui.py
index 7f6f4d7cf0..b6c67b1611 100644
--- a/src/blenderbim/blenderbim/bim/module/pset/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/pset/ui.py
@@ -18,7 +18,7 @@
from bpy.types import Panel
from blenderbim.bim.ifc import IfcStore
-from blenderbim.bim.ui import prop_with_search
+from blenderbim.bim.helper import prop_with_search
from blenderbim.bim.module.pset.data import (
ObjectPsetsData,
ObjectQtosData,
diff --git a/src/blenderbim/blenderbim/bim/module/pset_template/ui.py b/src/blenderbim/blenderbim/bim/module/pset_template/ui.py
index 4fa0b67644..b15ca05bd4 100644
--- a/src/blenderbim/blenderbim/bim/module/pset_template/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/pset_template/ui.py
@@ -19,7 +19,7 @@
import bpy
from bpy.types import Panel
from blenderbim.bim.ifc import IfcStore
-from blenderbim.bim.ui import prop_with_search
+from blenderbim.bim.helper import prop_with_search
from blenderbim.bim.module.pset_template.prop import (
getPsetTemplates,
)
diff --git a/src/blenderbim/blenderbim/bim/module/root/ui.py b/src/blenderbim/blenderbim/bim/module/root/ui.py
index b6a5f99268..18ca900cea 100644
--- a/src/blenderbim/blenderbim/bim/module/root/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/root/ui.py
@@ -20,7 +20,7 @@ import bpy
import blenderbim.bim.module.root.prop as root_prop
from bpy.types import Panel
from blenderbim.bim.ifc import IfcStore
-from blenderbim.bim.ui import prop_with_search
+from blenderbim.bim.helper import prop_with_search
from blenderbim.bim.module.root.data import IfcClassData
diff --git a/src/blenderbim/blenderbim/bim/module/unit/ui.py b/src/blenderbim/blenderbim/bim/module/unit/ui.py
index 183ed5cbe7..10a3682296 100644
--- a/src/blenderbim/blenderbim/bim/module/unit/ui.py
+++ b/src/blenderbim/blenderbim/bim/module/unit/ui.py
@@ -19,7 +19,7 @@
import blenderbim.bim.helper
from bpy.types import Panel, UIList
from blenderbim.bim.ifc import IfcStore
-from blenderbim.bim.ui import prop_with_search
+from blenderbim.bim.helper import prop_with_search
from blenderbim.bim.module.unit.data import UnitsData
diff --git a/src/blenderbim/blenderbim/bim/prop.py b/src/blenderbim/blenderbim/bim/prop.py
index ffdfc8bb27..e86896fd39 100644
--- a/src/blenderbim/blenderbim/bim/prop.py
+++ b/src/blenderbim/blenderbim/bim/prop.py
@@ -240,6 +240,10 @@ class Attribute(PropertyGroup):
value = str(value)
setattr(self, self.get_value_name(), value)
+ getter_enum = {
+ "enum_value": getAttributeEnumValues,
+ }
+
class ModuleVisibility(PropertyGroup):
name: StringProperty(name="Name")
diff --git a/src/blenderbim/blenderbim/bim/ui.py b/src/blenderbim/blenderbim/bim/ui.py
index e04e05d5be..2457cd99bf 100644
--- a/src/blenderbim/blenderbim/bim/ui.py
+++ b/src/blenderbim/blenderbim/bim/ui.py
@@ -170,15 +170,6 @@ class BIM_ADDON_preferences(bpy.types.AddonPreferences):
row.operator("bim.configure_visibility")
-def prop_with_search(layout, data, prop_name, **kwargs): # Kwargs are layout.prop arguments (text, icon, etc.)
- row = layout.row(align=True)
- # Magick courtesy of https://blender.stackexchange.com/a/203443/86891
- row.context_pointer_set(name="data", data=data)
- row.prop(data, prop_name, **kwargs)
- op = row.operator("bim.enum_property_search", text="", icon="VIEWZOOM")
- op.prop_name = prop_name
-
-
def ifc_units(self, context):
scene = context.scene
props = scene.BIMProperties