pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Factor out _create_workspace_role

Browse Source
This commit is contained in:
richard-dds 2018-08-22 11:24:55 -04:00
parent 50e2666c3c
commit 287fcf7e3e
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
atst/domain/workspaces.py
View File

@ -14,14 +14,9 @@ class Workspaces(object):
def create(cls, request, name=None): def create(cls, request, name=None):
name = name or request.id name = name or request.id
workspace = Workspace(request=request, name=name) workspace = Workspace(request=request, name=name)
Workspaces._create_workspace_role(request.creator, workspace, "owner")
role = Roles.get("owner")
workspace_role = WorkspaceRole(
user=request.creator, role=role, workspace=workspace
)
db.session.add(workspace) db.session.add(workspace)
db.session.add(workspace_role)
db.session.commit() db.session.commit()
return workspace return workspace
@ -67,6 +62,13 @@ class Workspaces(object):
return workspaces return workspaces
@classmethod @classmethod
def _create_workspace_role(cls, user, workspace, role_name):
role = Roles.get(role_name)
workspace_role = WorkspaceRole(
user=user, role=role, workspace=workspace
)
db.session.add(workspace_role)
return workspace_role
class Members(object): class Members(object):