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

@@ -9,7 +9,6 @@ from atst.models.application_role import (
ApplicationRole,
Status as ApplicationRoleStatuses,
)
from atst.database import db
@@ -52,8 +51,9 @@ class Application(
def displayname(self):
return self.name
def has_member(self, user):
return user in self.users
def has_member(self, user_id):
user_ids = set(str(user.id) for user in self.users)
return str(user_id) in user_ids
def __repr__(self): # pragma: no cover
return "<Application(name='{}', description='{}', portfolio='{}', id='{}')>".format(