91 lines
3.2 KiB
Python
91 lines
3.2 KiB
Python
# Copyright (c) 2023-present Plane Software, Inc. and contributors
|
|
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# See the LICENSE file for details.
|
|
|
|
# Django imports
|
|
from django.utils import timezone
|
|
|
|
# Module imports
|
|
from plane.db.models import (
|
|
ProjectMember,
|
|
ProjectMemberInvite,
|
|
WorkspaceMember,
|
|
WorkspaceMemberInvite,
|
|
)
|
|
from plane.utils.cache import invalidate_cache_directly
|
|
from plane.bgtasks.event_tracking_task import track_event
|
|
from plane.utils.analytics_events import USER_JOINED_WORKSPACE
|
|
|
|
|
|
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,
|
|
)
|
|
|
|
for workspace_member_invite in workspace_member_invites:
|
|
invalidate_cache_directly(
|
|
path=f"/api/workspaces/{str(workspace_member_invite.workspace.slug)}/members/",
|
|
url_params=False,
|
|
user=False,
|
|
multiple=True,
|
|
)
|
|
track_event.delay(
|
|
user_id=user.id,
|
|
event_name=USER_JOINED_WORKSPACE,
|
|
slug=workspace_member_invite.workspace.slug,
|
|
event_properties={
|
|
"user_id": user.id,
|
|
"workspace_id": workspace_member_invite.workspace.id,
|
|
"workspace_slug": workspace_member_invite.workspace.slug,
|
|
"role": workspace_member_invite.role,
|
|
"joined_at": str(timezone.now().isoformat()),
|
|
},
|
|
)
|
|
|
|
# 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, 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, 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()
|