Check for member in application function, not user function

This commit is contained in:
Montana
2019-04-24 09:09:29 -04:00
parent 6822680bc8
commit fec4687c02
6 changed files with 19 additions and 18 deletions

View File

@@ -7,7 +7,7 @@ from atst.models import EnvironmentRole
class EnvironmentRoles(object):
@classmethod
def create(cls, user, environment, role):
if user.is_app_member(environment.application):
if environment.application.has_member(user):
env_role = EnvironmentRole(user=user, environment=environment, role=role)
if not user.cloud_id:
user.cloud_id = app.csp.cloud.create_user(user)

View File

@@ -68,7 +68,7 @@ class Environments(object):
def update_env_role(cls, environment, user, new_role):
updated = False
if user.is_app_member(environment.application):
if environment.application.has_member(user):
if new_role is None:
updated = EnvironmentRoles.delete(user.id, environment.id)
else: