Files
IfcOpenShell/src/ifcopenshell-python/ifcopenshell/util/attribute.py
T

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

36 lines
1.4 KiB
Python
Raw Normal View History

def get_primitive_type(attribute_or_data_type):
if hasattr(attribute_or_data_type, "type_of_attribute"):
data_type = str(attribute_or_data_type.type_of_attribute())
else:
data_type = str(attribute_or_data_type)
if data_type.find("<type") == 0:
return get_primitive_type(data_type[data_type[1:].find("<")+1:])
elif data_type.find("<list") == 0:
return ("list", get_primitive_type(data_type[data_type[1:].find("<")+1:]))
elif data_type.find("<set") == 0:
return ("set", get_primitive_type(data_type[data_type[1:].find("<")+1:]))
elif data_type.find("<select") == 0:
2021-08-22 11:04:39 +10:00
select_definition = data_type[data_type.find("(")+1:data_type.find(")")].split("|")
select_types = [get_primitive_type(d.strip()) for d in select_definition]
return ("select", tuple(select_types))
elif "<entity" in data_type:
return "entity"
elif "<string>" in data_type:
return "string"
elif "<real>" in data_type:
return "float"
elif "<number>" in data_type or "<integer>" in data_type:
return "integer"
elif "<boolean>" in data_type or "<logical>" in data_type:
return "boolean"
elif "<enumeration" in data_type:
return "enum"
def get_enum_items(attribute):
return attribute.type_of_attribute().declared_type().enumeration_items()
def get_select_items(attribute):
return attribute.type_of_attribute().declared_type().select_list()