import authz domain module

atst/domain/workspace_users.py

@@ -0,0 +1,64 @@
+from sqlalchemy.orm.exc import NoResultFound
+from sqlalchemy.dialects.postgresql import insert
+
+from atst.models import User, WorkspaceRole, Role
+from .exceptions import NotFoundError
+
+class WorkspaceUsers(object):
+
+    def __init__(self, db_session):
+        self.db_session = db_session
+
+    def get(self, workspace_id, user_id):
+        try:
+            user = User.query.filter_by(id=user_id).one()
+        except NoResultFound:
+            raise NotFoundError("user")
+
+        try:
+            workspace_role = (
+                WorkspaceRole.query.join(User)
+                .filter(User.id == user_id, WorkspaceRole.workspace_id == workspace_id)
+                .one()
+            )
+        except NoResultFound:
+            workspace_role = None
+
+        return WorkspaceUser(user, workspace_role)
+
+    def add_many(self, workspace_id, workspace_user_dicts):
+        workspace_users = []
+
+        for user_dict in workspace_user_dicts:
+            try:
+                user = User.query.filter_by(id=user_dict["id"]).one()
+            except NoResultFound:
+                default_role = Role.query.filter_by(name="developer").one_or_none()
+                user = User(id=user_dict["id"], atat_role=default_role)
+
+            try:
+                role = Role.query.filter_by(name=user_dict["workspace_role"]).one()
+            except NoResultFound:
+                raise NotFoundError("role")
+
+            try:
+                existing_workspace_role = WorkspaceRole.query.filter(
+                    WorkspaceRole.user == user,
+                    WorkspaceRole.workspace_id == workspace_id,
+                ).one()
+                new_workspace_role = existing_workspace_role
+                new_workspace_role.role = role
+            except NoResultFound:
+                new_workspace_role = WorkspaceRole(
+                    user=user, role_id=role.id, workspace_id=workspace_id
+                )
+
+            user.workspace_roles.append(new_workspace_role)
+            workspace_user = WorkspaceUser(user, new_workspace_role)
+            workspace_users.append(workspace_user)
+
+            self.db_session.add(user)
+
+        self.db_session.commit()
+
+        return workspace_users