Use user.id to check if user is in application

This commit is contained in:
Montana
2019-04-29 11:04:17 -04:00
parent fec4687c02
commit 19a09b792e
5 changed files with 9 additions and 7 deletions

View File

@@ -7,7 +7,7 @@ from atst.models import EnvironmentRole
class EnvironmentRoles(object):
@classmethod
def create(cls, user, environment, role):
if environment.application.has_member(user):
if environment.application.has_member(user.id):
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 environment.application.has_member(user):
if environment.application.has_member(user.id):
if new_role is None:
updated = EnvironmentRoles.delete(user.id, environment.id)
else: