1fc5d2bd45
* refactor: folder structure for urls * chore: deleted the urls file * chore: proper naming for urls * chore: reset password url * dev: create refresh token endpoint and endpoint to get settings for user * dev: workspace member me serializer * dev: remove extra fields from project list and retrieve endpoints * dev: update the project list endpoint with member details and deploy boolean * dev: enable user favorite project endpoint and remove is_favorite from project list * dev: analytics refactoring * dev: revert is_favorite settings * dev: create new serializer for project list and add pagination from projects * dev: fix analytics api * dev: module and cycle * dev: update error message, fix module analytics and add null check for labels * dev: member serializer * dev: dynamic base serializer * dev: remove view issues endpoint * dev: url pattern updates * dev: add comments to delete this file * dev: last workspace id * dev: analytics export * dev: export analytics validation * dev: update python runtime * dev: update notification endpoints * dev: cycle and validation fix * dev: issue activity validation when creating updating and deleting issue and comments * dev: update issue activity logging for link and reactions * dev: update module issue activity logging * dev: update module issue activity --------- Co-authored-by: NarayanBavisetti <narayan3119@gmail.com>
355 lines
14 KiB
Python
355 lines
14 KiB
Python
# Django imports
|
|
from django.db.models import Q
|
|
from django.utils import timezone
|
|
|
|
# Third party imports
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
from sentry_sdk import capture_exception
|
|
from plane.utils.paginator import BasePaginator
|
|
|
|
# Module imports
|
|
from .base import BaseViewSet, BaseAPIView
|
|
from plane.db.models import (
|
|
Notification,
|
|
IssueAssignee,
|
|
IssueSubscriber,
|
|
Issue,
|
|
WorkspaceMember,
|
|
)
|
|
from plane.api.serializers import NotificationSerializer
|
|
|
|
|
|
class NotificationViewSet(BaseViewSet, BasePaginator):
|
|
model = Notification
|
|
serializer_class = NotificationSerializer
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
super()
|
|
.get_queryset()
|
|
.filter(
|
|
workspace__slug=self.kwargs.get("slug"),
|
|
receiver_id=self.request.user.id,
|
|
)
|
|
.select_related("workspace", "project," "triggered_by", "receiver")
|
|
)
|
|
|
|
def list(self, request, slug):
|
|
try:
|
|
# Get query parameters
|
|
snoozed = request.GET.get("snoozed", "false")
|
|
archived = request.GET.get("archived", "false")
|
|
read = request.GET.get("read", "true")
|
|
type = request.GET.get("type", "all")
|
|
|
|
notifications = Notification.objects.filter(workspace__slug=slug, receiver_id=request.user.id) \
|
|
.select_related("workspace", "project", "triggered_by", "receiver") \
|
|
.order_by("snoozed_till", "-created_at")
|
|
|
|
# Filters based on query parameters
|
|
snoozed_filters = {
|
|
"true": Q(snoozed_till__lt=timezone.now()) | Q(snoozed_till__isnull=False),
|
|
"false": Q(snoozed_till__gte=timezone.now()) | Q(snoozed_till__isnull=True),
|
|
}
|
|
|
|
notifications = notifications.filter(snoozed_filters[snoozed])
|
|
|
|
archived_filters = {
|
|
"true": Q(archived_at__isnull=False),
|
|
"false": Q(archived_at__isnull=True),
|
|
}
|
|
|
|
notifications = notifications.filter(archived_filters[archived])
|
|
|
|
if read == "false":
|
|
notifications = notifications.filter(read_at__isnull=True)
|
|
|
|
# Subscribed issues
|
|
if type == "watching":
|
|
issue_ids = IssueSubscriber.objects.filter(
|
|
workspace__slug=slug, subscriber_id=request.user.id
|
|
).values_list("issue_id", flat=True)
|
|
notifications = notifications.filter(entity_identifier__in=issue_ids)
|
|
|
|
# Assigned Issues
|
|
if type == "assigned":
|
|
issue_ids = IssueAssignee.objects.filter(
|
|
workspace__slug=slug, assignee_id=request.user.id
|
|
).values_list("issue_id", flat=True)
|
|
notifications = notifications.filter(entity_identifier__in=issue_ids)
|
|
|
|
# Created issues
|
|
if type == "created":
|
|
if WorkspaceMember.objects.filter(
|
|
workspace__slug=slug, member=request.user, role__lt=15
|
|
).exists():
|
|
notifications = Notification.objects.none()
|
|
else:
|
|
issue_ids = Issue.objects.filter(
|
|
workspace__slug=slug, created_by=request.user
|
|
).values_list("pk", flat=True)
|
|
notifications = notifications.filter(
|
|
entity_identifier__in=issue_ids
|
|
)
|
|
|
|
# Pagination
|
|
if request.GET.get("per_page", False) and request.GET.get("cursor", False):
|
|
return self.paginate(
|
|
request=request,
|
|
queryset=(notifications),
|
|
on_results=lambda notifications: NotificationSerializer(
|
|
notifications, many=True
|
|
).data,
|
|
)
|
|
|
|
serializer = NotificationSerializer(notifications, many=True)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
except Exception as e:
|
|
capture_exception(e)
|
|
return Response(
|
|
{"error": "Something went wrong please try again later"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
def partial_update(self, request, slug, pk):
|
|
try:
|
|
notification = Notification.objects.get(
|
|
workspace__slug=slug, pk=pk, receiver=request.user
|
|
)
|
|
# Only read_at and snoozed_till can be updated
|
|
notification_data = {
|
|
"snoozed_till": request.data.get("snoozed_till", None),
|
|
}
|
|
serializer = NotificationSerializer(
|
|
notification, data=notification_data, partial=True
|
|
)
|
|
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
except Notification.DoesNotExist:
|
|
return Response(
|
|
{"error": "Notification does not exists"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
except Exception as e:
|
|
capture_exception(e)
|
|
return Response(
|
|
{"error": "Something went wrong please try again later"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
def mark_read(self, request, slug, pk):
|
|
try:
|
|
notification = Notification.objects.get(
|
|
receiver=request.user, workspace__slug=slug, pk=pk
|
|
)
|
|
notification.read_at = timezone.now()
|
|
notification.save()
|
|
serializer = NotificationSerializer(notification)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
except Notification.DoesNotExist:
|
|
return Response(
|
|
{"error": "Notification does not exists"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
except Exception as e:
|
|
capture_exception(e)
|
|
return Response(
|
|
{"error": "Something went wrong please try again later"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
def mark_unread(self, request, slug, pk):
|
|
try:
|
|
notification = Notification.objects.get(
|
|
receiver=request.user, workspace__slug=slug, pk=pk
|
|
)
|
|
notification.read_at = None
|
|
notification.save()
|
|
serializer = NotificationSerializer(notification)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
except Notification.DoesNotExist:
|
|
return Response(
|
|
{"error": "Notification does not exists"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
except Exception as e:
|
|
capture_exception(e)
|
|
return Response(
|
|
{"error": "Something went wrong please try again later"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
def archive(self, request, slug, pk):
|
|
try:
|
|
notification = Notification.objects.get(
|
|
receiver=request.user, workspace__slug=slug, pk=pk
|
|
)
|
|
notification.archived_at = timezone.now()
|
|
notification.save()
|
|
serializer = NotificationSerializer(notification)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
except Notification.DoesNotExist:
|
|
return Response(
|
|
{"error": "Notification does not exists"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
except Exception as e:
|
|
capture_exception(e)
|
|
return Response(
|
|
{"error": "Something went wrong please try again later"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
def unarchive(self, request, slug, pk):
|
|
try:
|
|
notification = Notification.objects.get(
|
|
receiver=request.user, workspace__slug=slug, pk=pk
|
|
)
|
|
notification.archived_at = None
|
|
notification.save()
|
|
serializer = NotificationSerializer(notification)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
except Notification.DoesNotExist:
|
|
return Response(
|
|
{"error": "Notification does not exists"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
except Exception as e:
|
|
capture_exception(e)
|
|
return Response(
|
|
{"error": "Something went wrong please try again later"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
|
|
class UnreadNotificationEndpoint(BaseAPIView):
|
|
def get(self, request, slug):
|
|
try:
|
|
# Watching Issues Count
|
|
watching_issues_count = Notification.objects.filter(
|
|
workspace__slug=slug,
|
|
receiver_id=request.user.id,
|
|
read_at__isnull=True,
|
|
archived_at__isnull=True,
|
|
entity_identifier__in=IssueSubscriber.objects.filter(
|
|
workspace__slug=slug, subscriber_id=request.user.id
|
|
).values_list("issue_id", flat=True),
|
|
).count()
|
|
|
|
# My Issues Count
|
|
my_issues_count = Notification.objects.filter(
|
|
workspace__slug=slug,
|
|
receiver_id=request.user.id,
|
|
read_at__isnull=True,
|
|
archived_at__isnull=True,
|
|
entity_identifier__in=IssueAssignee.objects.filter(
|
|
workspace__slug=slug, assignee_id=request.user.id
|
|
).values_list("issue_id", flat=True),
|
|
).count()
|
|
|
|
# Created Issues Count
|
|
created_issues_count = Notification.objects.filter(
|
|
workspace__slug=slug,
|
|
receiver_id=request.user.id,
|
|
read_at__isnull=True,
|
|
archived_at__isnull=True,
|
|
entity_identifier__in=Issue.objects.filter(
|
|
workspace__slug=slug, created_by=request.user
|
|
).values_list("pk", flat=True),
|
|
).count()
|
|
|
|
return Response(
|
|
{
|
|
"watching_issues": watching_issues_count,
|
|
"my_issues": my_issues_count,
|
|
"created_issues": created_issues_count,
|
|
},
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
except Exception as e:
|
|
capture_exception(e)
|
|
return Response(
|
|
{"error": "Something went wrong please try again later"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
|
|
class MarkAllReadNotificationViewSet(BaseViewSet):
|
|
def create(self, request, slug):
|
|
try:
|
|
snoozed = request.data.get("snoozed", False)
|
|
archived = request.data.get("archived", False)
|
|
type = request.data.get("type", "all")
|
|
|
|
notifications = (
|
|
Notification.objects.filter(
|
|
workspace__slug=slug,
|
|
receiver_id=request.user.id,
|
|
read_at__isnull=True,
|
|
)
|
|
.select_related("workspace", "project", "triggered_by", "receiver")
|
|
.order_by("snoozed_till", "-created_at")
|
|
)
|
|
|
|
# Filter for snoozed notifications
|
|
if snoozed:
|
|
notifications = notifications.filter(
|
|
Q(snoozed_till__lt=timezone.now()) | Q(snoozed_till__isnull=False)
|
|
)
|
|
else:
|
|
notifications = notifications.filter(
|
|
Q(snoozed_till__gte=timezone.now()) | Q(snoozed_till__isnull=True),
|
|
)
|
|
|
|
# Filter for archived or unarchive
|
|
if archived:
|
|
notifications = notifications.filter(archived_at__isnull=False)
|
|
else:
|
|
notifications = notifications.filter(archived_at__isnull=True)
|
|
|
|
# Subscribed issues
|
|
if type == "watching":
|
|
issue_ids = IssueSubscriber.objects.filter(
|
|
workspace__slug=slug, subscriber_id=request.user.id
|
|
).values_list("issue_id", flat=True)
|
|
notifications = notifications.filter(entity_identifier__in=issue_ids)
|
|
|
|
# Assigned Issues
|
|
if type == "assigned":
|
|
issue_ids = IssueAssignee.objects.filter(
|
|
workspace__slug=slug, assignee_id=request.user.id
|
|
).values_list("issue_id", flat=True)
|
|
notifications = notifications.filter(entity_identifier__in=issue_ids)
|
|
|
|
# Created issues
|
|
if type == "created":
|
|
if WorkspaceMember.objects.filter(
|
|
workspace__slug=slug, member=request.user, role__lt=15
|
|
).exists():
|
|
notifications = Notification.objects.none()
|
|
else:
|
|
issue_ids = Issue.objects.filter(
|
|
workspace__slug=slug, created_by=request.user
|
|
).values_list("pk", flat=True)
|
|
notifications = notifications.filter(
|
|
entity_identifier__in=issue_ids
|
|
)
|
|
|
|
updated_notifications = []
|
|
for notification in notifications:
|
|
notification.read_at = timezone.now()
|
|
updated_notifications.append(notification)
|
|
Notification.objects.bulk_update(
|
|
updated_notifications, ["read_at"], batch_size=100
|
|
)
|
|
return Response({"message": "Successful"}, status=status.HTTP_200_OK)
|
|
except Exception as e:
|
|
capture_exception(e)
|
|
return Response(
|
|
{"error": "Something went wrong please try again later"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|