Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-09-11 14:22:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

More advanced entity_instance.get_info()

Browse Source
This commit is contained in:
Thomas Krijnen
2017-08-04 16:30:43 +02:00
parent c67de4bbb3
commit 85524170dd
1 changed files with 28 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/entity_instance.py
+28 -20
View File
@@ -124,27 +124,35 @@ class entity_instance(object):
map(str, self.wrapped_data.get_inverse_attribute_names()) map(str, self.wrapped_data.get_inverse_attribute_names())
))) )))
def get_info(self, include_identifier=True, recursive=False): def get_info(self, include_identifier=True, recursive=False, return_type=dict, ignore=()):
_info = {} def _():
try:
if include_identifier:
_info["id"] = self.id()
_info["type"] = self.is_a()
except:
logging.exception("unhandled exception while getting id / type info on {}".format(self))
for i in range(len(self)):
try: try:
attr_value = self[i] if include_identifier:
if recursive: yield "id", self.id()
is_instance = lambda e: isinstance(e, entity_instance) yield "type", self.is_a()
get_info = lambda inst: entity_instance.get_info(inst,
include_identifier=include_identifier,
recursive=recursive
)
attr_value = entity_instance.walk(is_instance, get_info, attr_value)
_info[self.attribute_name(i)] = attr_value
except: except:
logging.exception("unhandled exception occured setting attribute name for {}".format(self)) logging.exception("unhandled exception while getting id / type info on {}".format(self))
return _info for i in range(len(self)):
try:
if self.wrapped_data.get_attribute_names()[i] in ignore:
continue
attr_value = self[i]
if recursive:
is_instance = lambda e: isinstance(e, entity_instance)
def get_info_(inst):
# for ty in ignore:
# if inst.is_a(ty):
# return None
return entity_instance.get_info(inst,
include_identifier=include_identifier,
recursive=recursive,
return_type=return_type,
ignore=ignore
)
attr_value = entity_instance.walk(is_instance, get_info_, attr_value)
yield self.attribute_name(i), attr_value
except:
logging.exception("unhandled exception occured setting attribute name for {}".format(self))
return return_type(_())
__dict__ = property(get_info) __dict__ = property(get_info)