Create a count query class

ral-facilities · Aug 1, 2019 · b6897fb · b6897fb
1 parent 12affc7
commit b6897fb
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/common/database_helpers.py b/common/database_helpers.py
@@ -32,6 +32,20 @@ def commit_changes(self):
         self.session.close()
 
 
+
+class CountQuery(Query):
+
+    def __init__(self, table):
+        super().__init__(table)
+        self.include_related_entities = False
+
+    def execute_query(self):
+        self.commit_changes()
+
+    def get_count(self):
+        self.execute_query()
+        return self.base_query.count()
+
 class ReadQuery(Query):
 
     def __init__(self, table):