Make Brick "root_classes" one source of truth

This commit is contained in:
Trashman247
2023-07-18 16:47:04 -07:00
committed by Riley Wong
parent d834d94fd9
commit 4665b44ded
2 changed files with 4 additions and 7 deletions
@@ -52,11 +52,7 @@ def get_brick_entity_classes(self, context):
def get_brick_roots(self, context):
return [("System", "System", ""),
("Location", "Location", ""),
("Equipment", "Equipment", ""),
("Point", "Point", "")
]
return BrickStore.root_classes
class Brick(PropertyGroup):
+3 -2
View File
@@ -345,6 +345,7 @@ class BrickStore:
current_changesets = 0
history_size = 64
namespaces = []
root_classes = [("Equipment", "Equipment", ""), ("Location", "Location", ""), ("System", "System", ""), ("Point", "Point", "")]
entity_classes = {}
@staticmethod
@@ -374,8 +375,8 @@ class BrickStore:
@classmethod
def load_entity_classes(cls):
root_classes = ["System", "Location", "Equipment", "Point"]
for root_class in root_classes:
for root_class, __, __ in BrickStore.root_classes:
query = BrickStore.graph.query(
"""
PREFIX brick: <https://brickschema.org/schema/Brick#>