Merge pull request #256 from dod-ccpo/environment-users

Show how many members are in an environment

atst/domain/environments.py

@@ -1,5 +1,6 @@
 from atst.database import db
 from atst.models.environment import Environment
+from atst.models.environment_role import EnvironmentRole, CSPRole
 
 
 class Environments(object):
@@ -16,3 +17,13 @@ class Environments(object):
             environment = Environment(project=project, name=name)
             db.session.add(environment)
         db.session.commit()
+
+    @classmethod
+    def add_member(cls, user, environment, member):
+        environment_user = EnvironmentRole(
+            user=member, environment=environment, role=CSPRole.NONSENSE_ROLE.value
+        )
+        db.session.add(environment_user)
+        db.session.commit()
+
+        return environment

atst/models/environment.py

@@ -14,3 +14,11 @@ class Environment(Base, TimestampsMixin):
 
     project_id = Column(ForeignKey("projects.id"))
     project = relationship("Project")
+
+    @property
+    def users(self):
+        return [r.user for r in self.roles]
+
+    @property
+    def num_users(self):
+        return len(self.users)

atst/models/environment_role.py

@@ -0,0 +1,32 @@
+from enum import Enum
+from sqlalchemy import Index, ForeignKey, Column, String
+from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.orm import relationship
+
+from atst.models import Base
+from .types import Id
+
+
+class CSPRole(Enum):
+    NONSENSE_ROLE = "nonesense_role"
+
+
+class EnvironmentRole(Base):
+    __tablename__ = "environment_roles"
+
+    id = Id()
+    environment_id = Column(UUID(as_uuid=True), ForeignKey("environments.id"))
+    environment = relationship("Environment", backref="roles")
+
+    role = Column(String())
+
+    user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"))
+    user = relationship("User", backref="environment_roles")
+
+
+Index(
+    "environments_role_user_environment",
+    EnvironmentRole.user_id,
+    EnvironmentRole.environment_id,
+    unique=True,
+)