Add and list environment members

This commit is contained in:
richard-dds
2018-09-06 13:53:19 -04:00
parent e5a0d69844
commit 3edbfa3129
3 changed files with 31 additions and 0 deletions

View File

@@ -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

View File

@@ -14,3 +14,7 @@ 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]