Add query for getting environments pending baseline creation

2019-09-09 16:07:07 -04:00
parent 13861ad998
commit c00f13de2c
2 changed files with 47 additions and 0 deletions
--- a/atst/domain/environments.py
+++ b/atst/domain/environments.py
@@ -129,3 +129,13 @@ class Environments(object):
            .filter(Environment.root_user_info == text("'null'"))
        )
        return [environment_id for (environment_id,) in query.all()]
+
+    @classmethod
+    def get_environments_pending_baseline_creation(cls, now) -> [str]:
+        query = (
+            cls.base_provision_query(now)
+            .filter(Environment.cloud_id != None)
+            .filter(Environment.root_user_info != text("'null'"))
+            .filter(Environment.baseline_info == text("'null'"))
+        )
+        return [environment_id for (environment_id,) in query.all()]