mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-08-14 19:34:34 +00:00
rileywong311
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
import csv
|
|
import os
|
|
import json
|
|
#import pystache
|
|
import subprocess
|
|
import ifcopenshell
|
|
from pathlib import Path
|
|
|
|
class Generator():
|
|
def __init__(self):
|
|
self.file = ifcopenshell.file(schema='IFC4')
|
|
self.out_dir = './'
|
|
|
|
def generate(self):
|
|
classification = self.file.create_entity('IfcClassification', **{
|
|
'Source': 'VBIS',
|
|
'Edition': 'May-20',
|
|
'EditionDate': '2020-05-01',
|
|
'Name': 'VBIS',
|
|
'Description': '',
|
|
'Location': 'https://vbis.com.au/',
|
|
'ReferenceTokens': ['-']
|
|
})
|
|
# We assume all the Excel spreadsheets are re-exported in CSV format
|
|
for filename in Path('./').glob('*.csv'):
|
|
with open(filename, newline='') as csvfile:
|
|
reader = csv.reader(csvfile, delimiter=',', quotechar='"')
|
|
name = None
|
|
levels = [None, None, None, None]
|
|
ref = classification
|
|
for i, row in enumerate(reader):
|
|
if i <= 4:
|
|
continue
|
|
if row[0]:
|
|
name = row[0]
|
|
elif row[1]:
|
|
name = row[1]
|
|
elif row[2]:
|
|
name = row[2]
|
|
elif row[3]:
|
|
name = row[3]
|
|
ref = self.file.create_entity('IfcClassificationReference', **{
|
|
'Identification': row[4],
|
|
'Name': name
|
|
})
|
|
cur_level = len(row[4].split('-')) - 1
|
|
levels[cur_level] = ref
|
|
if cur_level == 0:
|
|
ref.ReferencedSource = classification
|
|
else:
|
|
ref.ReferencedSource = levels[cur_level-1]
|
|
self.file.write('VBIS.ifc')
|
|
|
|
generator = Generator()
|
|
generator.generate()
|