mirror of
https://github.com/IfcOpenShell/IfcOpenShell.git
synced 2026-09-01 09:26:26 +00:00
Dion Moult
68 lines
2.6 KiB
Python
68 lines
2.6 KiB
Python
import os
|
|
import re
|
|
import html
|
|
import json
|
|
from pathlib import Path
|
|
import ifcopenshell
|
|
|
|
class Describer():
|
|
def describe(self):
|
|
# BuildingSMART does not provide a computer interpretable set of
|
|
# descriptions. They provide HTML docs, which contained malformed /
|
|
# invalid HTML. Therefore, this dodgy hack was written.
|
|
schema = ifcopenshell.ifcopenshell_wrapper.schema_by_name('IFC4')
|
|
|
|
self.html_sources = {}
|
|
self.get_html_sources()
|
|
|
|
self.entity_descriptions = {}
|
|
self.enum_descriptions = {}
|
|
for entity in schema.entities():
|
|
name = entity.name()
|
|
self.get_entity_description(name)
|
|
for attribute in entity.attributes():
|
|
try:
|
|
attribute = attribute.type_of_attribute().declared_type()
|
|
except:
|
|
continue
|
|
if isinstance(attribute, str):
|
|
continue
|
|
if 'Enum' in attribute.name() and 'Enumeration' not in attribute.name():
|
|
self.get_enum_descriptions(attribute)
|
|
|
|
with open('entity_descriptions.json', 'w') as f:
|
|
f.write(json.dumps(self.entity_descriptions, indent=4))
|
|
with open('enum_descriptions.json', 'w') as f:
|
|
f.write(json.dumps(self.enum_descriptions, indent=4))
|
|
|
|
def get_html_sources(self):
|
|
html_dir = '/home/dion/Projects/IfcOpenShell/src/ifcblenderexport/descriptions/IFC4_3/RC1/HTML'
|
|
for filename in Path(html_dir).rglob('*.htm'):
|
|
if 'lexical' not in str(filename):
|
|
continue
|
|
name = os.path.basename(filename)[0:-4]
|
|
self.html_sources[name] = filename
|
|
|
|
def get_entity_description(self, name):
|
|
if name.lower() not in self.html_sources:
|
|
return
|
|
with open(self.html_sources[name.lower()]) as f:
|
|
for line in f:
|
|
if 'Entity definition' in line:
|
|
self.entity_descriptions[name] = html.unescape(re.sub('<.*?>', '', line.strip().replace('Entity definition', '')))
|
|
|
|
def get_enum_descriptions(self, enum):
|
|
if enum.name().lower() not in self.html_sources:
|
|
return
|
|
print(enum.name())
|
|
print(dir(enum))
|
|
for item in enum.enumeration_items():
|
|
with open(self.html_sources[enum.name().lower()]) as f:
|
|
for line in f:
|
|
if '<td>'+item+'</td>' in line:
|
|
self.enum_descriptions.setdefault(enum.name(), {})[item] = html.unescape(re.sub('<.*?>', '', line.strip().replace(item, '')))
|
|
|
|
|
|
describer = Describer()
|
|
describer.describe()
|