Merge pull request #206 from dod-ccpo/multiple-environments

Allow a user to add multiple environments when creating a project

atst/domain/environments.py

@@ -10,3 +10,9 @@ class Environments(object):
         db.session.commit()
         return environment
 
+    @classmethod
+    def create_many(cls, project, names):
+        for name in names:
+            environment = Environment(project=project, name=name)
+            db.session.add(environment)
+        db.session.commit()

atst/domain/projects.py

@@ -1,11 +1,13 @@
 from atst.database import db
 from atst.models.project import Project
+from atst.domain.environments import Environments
 
 
 class Projects(object):
     @classmethod
-    def create(cls, workspace, name, description):
+    def create(cls, workspace, name, description, environment_names):
         project = Project(workspace=workspace, name=name, description=description)
+        Environments.create_many(project, environment_names)
 
         db.session.add(project)
         db.session.commit()