Refactor BrickStore.root_classes

This commit is contained in:
rileywong311
2023-07-24 21:23:40 -07:00
parent 546f8d6fe7
commit bc71ccd9f1
2 changed files with 6 additions and 6 deletions
@@ -47,13 +47,13 @@ def get_namespaces(self, context):
def get_brick_entity_classes(self, context):
entity = context.scene.BIMBrickProperties.brick_entity_create_type
entity = self.brick_entity_create_type
return BrickStore.entity_classes[entity]
def get_brick_roots(self, context):
return BrickStore.root_classes
def get_brick_roots(self, context):
return [(root, root, "") for root in BrickStore.root_classes]
class Brick(PropertyGroup):
name: StringProperty(name="Name")
+2 -2
View File
@@ -345,7 +345,7 @@ class BrickStore:
current_changesets = 0
history_size = 64
namespaces = []
root_classes = [("Equipment", "Equipment", ""), ("Location", "Location", ""), ("System", "System", ""), ("Point", "Point", "")]
root_classes = ["Equipment", "Location", "System", "Point"]
entity_classes = {}
@staticmethod
@@ -376,7 +376,7 @@ class BrickStore:
@classmethod
def load_entity_classes(cls):
for root_class, __, __ in BrickStore.root_classes:
for root_class in BrickStore.root_classes:
query = BrickStore.graph.query(
"""
PREFIX brick: <https://brickschema.org/schema/Brick#>