Add and list environment members
This commit is contained in:
@@ -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
|
||||
|
@@ -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]
|
||||
|
Reference in New Issue
Block a user