Run black

This commit is contained in:
Thomas Krijnen
2025-09-09 13:26:01 +02:00
parent a0b93265c2
commit 62295beab2
+17 -11
View File
@@ -307,30 +307,31 @@ binary_deserializers = (
), ),
) )
@functools.cache @functools.cache
def attribute_lookup(schema_name, entity_name): def attribute_lookup(schema_name, entity_name):
decl = ifcopenshell_wrapper.schema_by_name(schema_name).declaration_by_name(entity_name) decl = ifcopenshell_wrapper.schema_by_name(schema_name).declaration_by_name(entity_name)
attributes = ( attributes = decl.as_entity().all_attributes()
decl di = {v: k for k, v in enumerate(a.name() for a in attributes)}
.as_entity()
.all_attributes()
)
di = {v:k for k, v in enumerate(a.name() for a in attributes)}
all_inverses = decl.as_entity().all_inverse_attributes() all_inverses = decl.as_entity().all_inverse_attributes()
for inv in all_inverses: for inv in all_inverses:
def visit(decl): def visit(decl):
yield decl.index_in_schema() yield decl.index_in_schema()
for ty in decl.subtypes(): for ty in decl.subtypes():
yield from visit(ty) yield from visit(ty)
attribute_index = inv.entity_reference().attribute_index(inv.attribute_reference()) attribute_index = inv.entity_reference().attribute_index(inv.attribute_reference())
entity_indices = list(visit(inv.entity_reference())) entity_indices = list(visit(inv.entity_reference()))
di[inv.name()] = (entity_indices, attribute_index) di[inv.name()] = (entity_indices, attribute_index)
return di return di
@functools.cache @functools.cache
def entity_name_lookup(schema_name, index): def entity_name_lookup(schema_name, index):
return ifcopenshell_wrapper.schema_by_name(schema_name).declarations()[index].name() return ifcopenshell_wrapper.schema_by_name(schema_name).declarations()[index].name()
class rocksdb_lazy_instance: class rocksdb_lazy_instance:
__slots__ = ("storage", "name") __slots__ = ("storage", "name")
@@ -357,10 +358,12 @@ class rocksdb_lazy_instance:
return self[attr] return self[attr]
else: else:
entity_indices, attribute_index = attr entity_indices, attribute_index = attr
def _(): def _():
for index_in_schema in entity_indices: for index_in_schema in entity_indices:
buffer = self.storage.read(f'v|{self.name[2:]}|{index_in_schema}|{attribute_index}') or b'' buffer = self.storage.read(f"v|{self.name[2:]}|{index_in_schema}|{attribute_index}") or b""
yield from map(self.storage.by_id, struct.unpack('<' + 'I' * (len(buffer) // 4), buffer)) yield from map(self.storage.by_id, struct.unpack("<" + "I" * (len(buffer) // 4), buffer))
return list(_()) return list(_())
def __getitem__(self, index): def __getitem__(self, index):
@@ -450,16 +453,19 @@ class rocksdb_file_storage:
raise KeyError(f"Instance with name {name} not found in file") raise KeyError(f"Instance with name {name} not found in file")
return inst return inst
def by_type(self, ty:str): def by_type(self, ty: str):
def visit(decl): def visit(decl):
yield decl.index_in_schema() yield decl.index_in_schema()
for ty in decl.subtypes(): for ty in decl.subtypes():
yield from visit(ty) yield from visit(ty)
decl = ifcopenshell_wrapper.schema_by_name(self.schema_identifier).declaration_by_name(ty) decl = ifcopenshell_wrapper.schema_by_name(self.schema_identifier).declaration_by_name(ty)
def _(): def _():
for index in visit(decl): for index in visit(decl):
buff = self.read(f't|{index}') or b'' buff = self.read(f"t|{index}") or b""
yield from map(self.by_id, struct.unpack('@' + 'q' * (len(buff) // 8), buff)) yield from map(self.by_id, struct.unpack("@" + "q" * (len(buff) // 8), buff))
return list(_()) return list(_())
__getitem__ = by_id __getitem__ = by_id