Add methods to fetch environments from a project

1 files changed, 9 insertions, 0 deletions

diff --git a/squad_client/core/models.py b/squad_client/core/models.py
index c64939b..3ffd037 100644
--- a/squad_client/core/models.py
+++ b/squad_client/core/models.py
@@ -239,6 +239,15 @@ class Project(SquadObject):
         objects = self.builds(count=1, **filters)
         return first(objects)
 
+    def environments(self, count=DEFAULT_COUNT, **filters):
+        filters.update({'project': self.id})
+        return self.__fetch__(Environment, filters, count)
+
+    def environment(self, slug):
+        filters = {'slug': slug}
+        objects = self.environments(count=1, **filters)
+        return first(objects)
+
     def suites(self, count=DEFAULT_COUNT, **filters):
         filters.update({'project': self.id})
         return self.__fetch__(Suite, filters, count)