Glassvision/IfcOpenShell
1
0
Fork 0
mirror of https://github.com/IfcOpenShell/IfcOpenShell.git synced 2026-08-10 09:48:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

ifc5d csv2ifc - autodetect schedule of rates by the header structure

Browse Source
This commit is contained in:
Andrej730
2025-04-16 11:14:59 +05:00
parent 7acd13cb82
commit ddd7dff75e
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ifc5d/ifc5d/csv2ifc.py
+13 -5
View File
@@ -84,6 +84,8 @@ class Csv2Ifc:
parents: dict[int, CostItem] = {} parents: dict[int, CostItem] = {}
locale.setlocale(locale.LC_ALL, "") # set the system locale locale.setlocale(locale.LC_ALL, "") # set the system locale
non_sor_fields = {"Property", "Query"}
with open(self.csv, "r", encoding="utf-8") as csv_file: with open(self.csv, "r", encoding="utf-8") as csv_file:
reader = csv.reader(csv_file) reader = csv.reader(csv_file)
for row in reader: for row in reader:
@@ -102,12 +104,18 @@ class Csv2Ifc:
# validate header # validate header
mandatory_fields = {"Name", "Quantity", "Unit"} mandatory_fields = {"Name", "Quantity", "Unit"}
if not self.is_schedule_of_rates: if not self.is_schedule_of_rates:
mandatory_fields.update({"Property", "Query"}) mandatory_fields.update(non_sor_fields)
available_fields = set(self.headers.keys()) available_fields = set(self.headers.keys())
if not mandatory_fields.issubset(available_fields): missing_fields = mandatory_fields - available_fields
raise Exception( if missing_fields:
f"Missing mandatory fields in CSV header: {', '.join(mandatory_fields-available_fields)}" if missing_fields == non_sor_fields and not self.is_schedule_of_rates:
) self.is_schedule_of_rates = True
print(
"WARNING. Assumed the imported cost schedule is a schedule of rates "
f"because the following fields are missing: {', '.join(non_sor_fields)}."
)
else:
raise Exception(f"Missing mandatory fields in CSV header: {', '.join(missing_fields)}")
continue continue
cost_data = self.get_row_cost_data(row) cost_data = self.get_row_cost_data(row)