Merge "Test resource templates now use the same lab as the image generated alongside it."

1 files changed, 42 insertions, 0 deletions

diff --git a/src/dashboard/utils.py b/src/dashboard/utils.py
new file mode 100644
index 0000000..3d63366
--- /dev/null
+++ b/src/dashboard/utils.py
@@ -0,0 +1,42 @@
+##############################################################################
+# Copyright (c) 2020 Parker Berberian and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+from django.core.exceptions import ObjectDoesNotExist
+
+
+class AbstractModelQuery():
+    """
+    This is a class made for querying abstract models.
+
+    This class is itself abstract. create subclasses to
+    query your own abstract models.
+    """
+
+    model_list = []
+
+    @classmethod
+    def filter(cls, *args, **kwargs):
+        """
+        Query all concrete model classes.
+
+        Iterates over the model list and returns a list of all
+        matching models from the classes given.
+        Filter queries are given here as normal and are passed into the Django ORM
+        for each concrete model
+        """
+        result = []
+        for model in cls.model_list:
+            result += list(model.objects.filter(*args, **kwargs))
+
+    @classmethod
+    def get(cls, *args, **kwargs):
+        try:
+            return cls.filter(*args, **kwargs)[0]
+        except IndexError:
+            raise ObjectDoesNotExist()