Refactor code duplication

This commit is contained in:
Montana 2018-11-13 15:06:24 -05:00
parent e51de11a28
commit 3f6e13f33d
2 changed files with 3 additions and 3 deletions

View File

@ -101,9 +101,9 @@ class WorkspaceRoles(object):
return new_workspace_role return new_workspace_role
@classmethod @classmethod
def update_role(cls, member, workspace_id, role_name): def update_role(cls, member, role_name):
new_role = Roles.get(role_name) new_role = Roles.get(role_name)
workspace_role = WorkspaceRoles._get_workspace_role(member.user, workspace_id) workspace_role = WorkspaceRoles._get_workspace_role(member.user, member.workspace_id)
workspace_role.role = new_role workspace_role.role = new_role
db.session.add(workspace_role) db.session.add(workspace_role)

View File

@ -119,7 +119,7 @@ class Workspaces(object):
"edit workspace member", "edit workspace member",
) )
return WorkspaceRoles.update_role(member, workspace.id, role_name) return WorkspaceRoles.update_role(member, role_name)
@classmethod @classmethod
def _create_workspace_role( def _create_workspace_role(