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

Check for null values when setting attributes in IfcOpenShell-Python, also return value being set

Browse Source
This commit is contained in:
Thomas Krijnen
2015-06-18 13:15:33 +00:00
parent feaa13a809
commit 48febb7eff
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifcopenshell-python/ifcopenshell/__init__.py
+8 -4
View File
@@ -75,10 +75,14 @@ class entity_instance(object):
def __getitem__(self, key): def __getitem__(self, key):
return entity_instance.wrap_value(self.wrapped_data.get_argument(key)) return entity_instance.wrap_value(self.wrapped_data.get_argument(key))
def __setitem__(self, idx, value): def __setitem__(self, idx, value):
attr_type = self.attribute_type(idx).title().replace(' ', '') if value is None:
attr_type = attr_type.replace('Binary', 'String') self.wrapped_data.setArgumentAsNull(idx)
attr_type = attr_type.replace('Enumeration', 'String') else:
getattr(self.wrapped_data, "setArgumentAs%s" % attr_type)(idx, entity_instance.unwrap_value(value)) attr_type = self.attribute_type(idx).title().replace(' ', '')
attr_type = attr_type.replace('Binary', 'String')
attr_type = attr_type.replace('Enumeration', 'String')
getattr(self.wrapped_data, "setArgumentAs%s" % attr_type)(idx, entity_instance.unwrap_value(value))
return value
def __len__(self): return len(self.wrapped_data) def __len__(self): return len(self.wrapped_data)
def __repr__(self): return repr(self.wrapped_data) def __repr__(self): return repr(self.wrapped_data)
def is_a(self, *args): return self.wrapped_data.is_a(*args) def is_a(self, *args): return self.wrapped_data.is_a(*args)