Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-08-11 18:16:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

ifccsv - support importing enum properties

Browse Source
This commit is contained in:
Andrej730
2024-05-06 18:50:36 +05:00
parent 28d205b4a3
commit 8f34752141
2 changed files with 23 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/blenderbim/blenderbim/bim/module/csv/operator.py
+1
View File
@@ -282,6 +282,7 @@ class ImportIfcCsv(bpy.types.Operator):
empty=props.empty_value,
bool_true=props.true_value,
bool_false=props.false_value,
concat=props.concat_value
)
if not props.should_load_from_memory:
ifc_file.write(props.csv_ifc_file)
src/ifccsv/ifccsv.py
+22 -11
View File
@@ -391,15 +391,16 @@ class IfcCsv:
empty: str = "",
bool_true: str = "YES",
bool_false: str = "NO",
concat: str = ", ",
) -> None:
ext = table.split(".")[-1].lower()
if ext == "csv":
self.import_csv(ifc_file, table, attributes, delimiter, null, empty, bool_true, bool_false)
self.import_csv(ifc_file, table, attributes, delimiter, null, empty, bool_true, bool_false, concat)
elif ext == "ods":
self.import_ods(ifc_file, table, attributes, null, empty, bool_true, bool_false)
self.import_ods(ifc_file, table, attributes, null, empty, bool_true, bool_false, concat)
elif ext == "xlsx":
self.import_xlsx(ifc_file, table, attributes, null, empty, bool_true, bool_false)
self.import_xlsx(ifc_file, table, attributes, null, empty, bool_true, bool_false, concat)
def import_csv(
self,
@@ -411,6 +412,7 @@ class IfcCsv:
empty: str = "",
bool_true: str = "YES",
bool_false: str = "NO",
concat: str = ", ",
) -> None:
with open(table, newline="", encoding="utf-8") as f:
reader = csv.reader(f, delimiter=delimiter)
@@ -423,17 +425,17 @@ class IfcCsv:
elif len(attributes) == len(headers) - 1:
attributes.insert(0, "") # The GlobalId column
continue
self.process_row(ifc_file, row, headers, attributes, null, empty, bool_true, bool_false)
self.process_row(ifc_file, row, headers, attributes, null, empty, bool_true, bool_false, concat)
def import_xlsx(self, ifc_file, table, attributes, null, empty, bool_true, bool_false):
def import_xlsx(self, ifc_file, table, attributes, null, empty, bool_true, bool_false, concat):
df = pd.read_excel(table)
self.import_pd(ifc_file, df, attributes, null, empty, bool_true, bool_false)
self.import_pd(ifc_file, df, attributes, null, empty, bool_true, bool_false, concat)
def import_ods(self, ifc_file, table, attributes, null, empty, bool_true, bool_false):
def import_ods(self, ifc_file, table, attributes, null, empty, bool_true, bool_false, concat):
df = pd.read_excel(table, engine="odf")
self.import_pd(ifc_file, df, attributes, null, empty, bool_true, bool_false)
self.import_pd(ifc_file, df, attributes, null, empty, bool_true, bool_false, concat)
def import_pd(self, ifc_file, df, attributes=None, null="-", empty="", bool_true="YES", bool_false="NO"):
def import_pd(self, ifc_file, df, attributes=None, null="-", empty="", bool_true="YES", bool_false="NO", concat=", "):
headers = df.columns.tolist()
if not attributes:
@@ -442,7 +444,7 @@ class IfcCsv:
attributes.insert(0, "") # The GlobalId column
for _, row in df.iterrows():
self.process_row(ifc_file, row.tolist(), headers, attributes, null, empty, bool_true, bool_false)
self.process_row(ifc_file, row.tolist(), headers, attributes, null, empty, bool_true, bool_false, concat)
def process_row(
self,
@@ -454,6 +456,7 @@ class IfcCsv:
empty: str,
bool_true: str,
bool_false: str,
concat: str
) -> None:
try:
element = ifc_file.by_guid(row[0])
@@ -472,7 +475,14 @@ class IfcCsv:
elif value == bool_false:
value = False
key = attributes[i] or headers[i]
ifcopenshell.util.selector.set_element_value(ifc_file, element, key, value)
try:
ifcopenshell.util.selector.set_element_value(ifc_file, element, key, value)
except ValueError as e:
if "enum property" in e.args[0]:
value = value.split(concat)
ifcopenshell.util.selector.set_element_value(ifc_file, element, key, value)
else:
raise e
if __name__ == "__main__":
@@ -534,5 +544,6 @@ if __name__ == "__main__":
delimiter=args.delimiter,
null=args.null,
empty=args.empty,
concat=args.concat
)
ifc_file.write(args.ifc)