Display members' count of environment roles

2018-09-24 14:45:03 -04:00
parent 5fec89d120
commit ac1c4cd5b7
2 changed files with 13 additions and 5 deletions
--- a/atst/models/workspace_user.py
+++ b/atst/models/workspace_user.py
@@ -41,8 +41,8 @@ class WorkspaceUser(object):
        return "active"

    @property
-    def has_environment_roles(self):
-        num_environment_roles = (
+    def num_environment_roles(self):
+        return (
            db.session.query(EnvironmentRole)
            .join(EnvironmentRole.environment)
            .join(Environment.project)
@@ -51,4 +51,7 @@ class WorkspaceUser(object):
            .filter(EnvironmentRole.user_id == self.user_id)
            .count()
        )
-        return num_environment_roles > 0
+
+    @property
+    def has_environment_roles(self):
+        return self.num_environment_roles > 0