83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
from plane.db.models import (
|
|
ProjectMember,
|
|
ProjectMemberInvite,
|
|
WorkspaceMember,
|
|
WorkspaceMemberInvite,
|
|
)
|
|
from plane.utils.cache import invalidate_cache_directly
|
|
|
|
|
|
def process_workspace_project_invitations(user):
|
|
"""This function takes in User and adds him to all workspace and projects that the user has accepted invited of"""
|
|
|
|
# Check if user has any accepted invites for workspace and add them to workspace
|
|
workspace_member_invites = WorkspaceMemberInvite.objects.filter(
|
|
email=user.email, accepted=True
|
|
)
|
|
|
|
WorkspaceMember.objects.bulk_create(
|
|
[
|
|
WorkspaceMember(
|
|
workspace_id=workspace_member_invite.workspace_id,
|
|
member=user,
|
|
role=workspace_member_invite.role,
|
|
)
|
|
for workspace_member_invite in workspace_member_invites
|
|
],
|
|
ignore_conflicts=True,
|
|
)
|
|
|
|
[
|
|
invalidate_cache_directly(
|
|
path=f"/api/workspaces/{str(workspace_member_invite.workspace.slug)}/members/",
|
|
url_params=False,
|
|
user=False,
|
|
multiple=True,
|
|
)
|
|
for workspace_member_invite in workspace_member_invites
|
|
]
|
|
|
|
# Check if user has any project invites
|
|
project_member_invites = ProjectMemberInvite.objects.filter(
|
|
email=user.email, accepted=True
|
|
)
|
|
|
|
# Add user to workspace
|
|
WorkspaceMember.objects.bulk_create(
|
|
[
|
|
WorkspaceMember(
|
|
workspace_id=project_member_invite.workspace_id,
|
|
role=(
|
|
project_member_invite.role
|
|
if project_member_invite.role in [5, 10, 15]
|
|
else 15
|
|
),
|
|
member=user,
|
|
created_by_id=project_member_invite.created_by_id,
|
|
)
|
|
for project_member_invite in project_member_invites
|
|
],
|
|
ignore_conflicts=True,
|
|
)
|
|
|
|
# Now add the users to project
|
|
ProjectMember.objects.bulk_create(
|
|
[
|
|
ProjectMember(
|
|
workspace_id=project_member_invite.workspace_id,
|
|
role=(
|
|
project_member_invite.role
|
|
if project_member_invite.role in [5, 10, 15]
|
|
else 15
|
|
),
|
|
member=user,
|
|
created_by_id=project_member_invite.created_by_id,
|
|
)
|
|
for project_member_invite in project_member_invites
|
|
],
|
|
ignore_conflicts=True,
|
|
)
|
|
|
|
# Delete all the invites
|
|
workspace_member_invites.delete()
|
|
project_member_invites.delete()
|