288 lines
9.4 KiB
Python
288 lines
9.4 KiB
Python
# Python imports
|
|
import re
|
|
|
|
# Django imports
|
|
from django.db.models import Q
|
|
|
|
# Third party imports
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
from sentry_sdk import capture_exception
|
|
|
|
# Module imports
|
|
from .base import BaseAPIView
|
|
from plane.db.models import Workspace, Project, Issue, Cycle, Module, Page, IssueView
|
|
from plane.utils.issue_search import search_issues
|
|
|
|
|
|
class GlobalSearchEndpoint(BaseAPIView):
|
|
"""Endpoint to search across multiple fields in the workspace and
|
|
also show related workspace if found
|
|
"""
|
|
|
|
def filter_workspaces(self, query, slug, project_id, workspace_search):
|
|
fields = ["name"]
|
|
q = Q()
|
|
for field in fields:
|
|
q |= Q(**{f"{field}__icontains": query})
|
|
return (
|
|
Workspace.objects.filter(q, workspace_member__member=self.request.user)
|
|
.distinct()
|
|
.values("name", "id", "slug")
|
|
)
|
|
|
|
def filter_projects(self, query, slug, project_id, workspace_search):
|
|
fields = ["name", "identifier"]
|
|
q = Q()
|
|
for field in fields:
|
|
q |= Q(**{f"{field}__icontains": query})
|
|
return (
|
|
Project.objects.filter(
|
|
q,
|
|
Q(project_projectmember__member=self.request.user) | Q(network=2),
|
|
workspace__slug=slug,
|
|
)
|
|
.distinct()
|
|
.values("name", "id", "identifier", "workspace__slug")
|
|
)
|
|
|
|
def filter_issues(self, query, slug, project_id, workspace_search):
|
|
fields = ["name", "sequence_id", "project__identifier"]
|
|
q = Q()
|
|
for field in fields:
|
|
if field == "sequence_id":
|
|
sequences = re.findall(r"\d+\.\d+|\d+", query)
|
|
for sequence_id in sequences:
|
|
q |= Q(**{"sequence_id": sequence_id})
|
|
else:
|
|
q |= Q(**{f"{field}__icontains": query})
|
|
|
|
issues = Issue.issue_objects.filter(
|
|
q,
|
|
project__project_projectmember__member=self.request.user,
|
|
workspace__slug=slug,
|
|
)
|
|
|
|
if workspace_search == "false":
|
|
issues = issues.filter(project_id=project_id)
|
|
|
|
return issues.distinct().values(
|
|
"name",
|
|
"id",
|
|
"sequence_id",
|
|
"project__identifier",
|
|
"project_id",
|
|
"workspace__slug",
|
|
)
|
|
|
|
def filter_cycles(self, query, slug, project_id, workspace_search):
|
|
fields = ["name"]
|
|
q = Q()
|
|
for field in fields:
|
|
q |= Q(**{f"{field}__icontains": query})
|
|
|
|
cycles = Cycle.objects.filter(
|
|
q,
|
|
project__project_projectmember__member=self.request.user,
|
|
workspace__slug=slug,
|
|
)
|
|
|
|
if workspace_search == "false":
|
|
cycles = cycles.filter(project_id=project_id)
|
|
|
|
return cycles.distinct().values(
|
|
"name",
|
|
"id",
|
|
"project_id",
|
|
"workspace__slug",
|
|
)
|
|
|
|
def filter_modules(self, query, slug, project_id, workspace_search):
|
|
fields = ["name"]
|
|
q = Q()
|
|
for field in fields:
|
|
q |= Q(**{f"{field}__icontains": query})
|
|
|
|
modules = Module.objects.filter(
|
|
q,
|
|
project__project_projectmember__member=self.request.user,
|
|
workspace__slug=slug,
|
|
)
|
|
|
|
if workspace_search == "false":
|
|
modules = modules.filter(project_id=project_id)
|
|
|
|
return modules.distinct().values(
|
|
"name",
|
|
"id",
|
|
"project_id",
|
|
"workspace__slug",
|
|
)
|
|
|
|
def filter_pages(self, query, slug, project_id, workspace_search):
|
|
fields = ["name"]
|
|
q = Q()
|
|
for field in fields:
|
|
q |= Q(**{f"{field}__icontains": query})
|
|
|
|
pages = Page.objects.filter(
|
|
q,
|
|
project__project_projectmember__member=self.request.user,
|
|
workspace__slug=slug,
|
|
)
|
|
|
|
if workspace_search == "false":
|
|
pages = pages.filter(project_id=project_id)
|
|
|
|
return pages.distinct().values(
|
|
"name",
|
|
"id",
|
|
"project_id",
|
|
"workspace__slug",
|
|
)
|
|
|
|
def filter_views(self, query, slug, project_id, workspace_search):
|
|
fields = ["name"]
|
|
q = Q()
|
|
for field in fields:
|
|
q |= Q(**{f"{field}__icontains": query})
|
|
|
|
issue_views = IssueView.objects.filter(
|
|
q,
|
|
project__project_projectmember__member=self.request.user,
|
|
workspace__slug=slug,
|
|
)
|
|
|
|
if workspace_search == "false":
|
|
issue_views = issue_views.filter(project_id=project_id)
|
|
|
|
return issue_views.distinct().values(
|
|
"name",
|
|
"id",
|
|
"project_id",
|
|
"workspace__slug",
|
|
)
|
|
|
|
def get(self, request, slug, project_id):
|
|
try:
|
|
query = request.query_params.get("search", False)
|
|
workspace_search = request.query_params.get("workspace_search", "false")
|
|
|
|
if not query:
|
|
return Response(
|
|
{
|
|
"results": {
|
|
"workspace": [],
|
|
"project": [],
|
|
"issue": [],
|
|
"cycle": [],
|
|
"module": [],
|
|
"issue_view": [],
|
|
"page": [],
|
|
}
|
|
},
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
|
|
MODELS_MAPPER = {
|
|
"workspace": self.filter_workspaces,
|
|
"project": self.filter_projects,
|
|
"issue": self.filter_issues,
|
|
"cycle": self.filter_cycles,
|
|
"module": self.filter_modules,
|
|
"issue_view": self.filter_views,
|
|
"page": self.filter_pages,
|
|
}
|
|
|
|
results = {}
|
|
|
|
for model in MODELS_MAPPER.keys():
|
|
func = MODELS_MAPPER.get(model, None)
|
|
results[model] = func(query, slug, project_id, workspace_search)
|
|
return Response({"results": results}, 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 IssueSearchEndpoint(BaseAPIView):
|
|
def get(self, request, slug, project_id):
|
|
try:
|
|
query = request.query_params.get("search", False)
|
|
workspace_search = request.query_params.get("workspace_search", "false")
|
|
parent = request.query_params.get("parent", "false")
|
|
blocker_blocked_by = request.query_params.get("blocker_blocked_by", "false")
|
|
cycle = request.query_params.get("cycle", "false")
|
|
module = request.query_params.get("module", "false")
|
|
sub_issue = request.query_params.get("sub_issue", "false")
|
|
|
|
issue_id = request.query_params.get("issue_id", False)
|
|
|
|
issues = Issue.issue_objects.filter(
|
|
workspace__slug=slug,
|
|
project__project_projectmember__member=self.request.user,
|
|
)
|
|
|
|
if workspace_search == "false":
|
|
issues = issues.filter(project_id=project_id)
|
|
|
|
if query:
|
|
issues = search_issues(query, issues)
|
|
|
|
if parent == "true" and issue_id:
|
|
issue = Issue.issue_objects.get(pk=issue_id)
|
|
issues = issues.filter(
|
|
~Q(pk=issue_id), ~Q(pk=issue.parent_id), parent__isnull=True
|
|
).exclude(
|
|
pk__in=Issue.issue_objects.filter(parent__isnull=False).values_list(
|
|
"parent_id", flat=True
|
|
)
|
|
)
|
|
if blocker_blocked_by == "true" and issue_id:
|
|
issue = Issue.issue_objects.get(pk=issue_id)
|
|
issues = issues.filter(
|
|
~Q(pk=issue_id),
|
|
~Q(blocked_issues__block=issue),
|
|
~Q(blocker_issues__blocked_by=issue),
|
|
)
|
|
if sub_issue == "true" and issue_id:
|
|
issue = Issue.issue_objects.get(pk=issue_id)
|
|
issues = issues.filter(~Q(pk=issue_id), parent__isnull=True)
|
|
if issue.parent:
|
|
issues = issues.filter(~Q(pk=issue.parent_id))
|
|
|
|
if cycle == "true":
|
|
issues = issues.exclude(issue_cycle__isnull=False)
|
|
|
|
if module == "true":
|
|
issues = issues.exclude(issue_module__isnull=False)
|
|
|
|
return Response(
|
|
issues.values(
|
|
"name",
|
|
"id",
|
|
"sequence_id",
|
|
"project__name",
|
|
"project__identifier",
|
|
"project_id",
|
|
"workspace__slug",
|
|
"state__name",
|
|
"state__group",
|
|
"state__color",
|
|
),
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
except Issue.DoesNotExist:
|
|
return Response(
|
|
{"error": "Issue Does not exist"}, status=status.HTTP_400_BAD_REQUEST
|
|
)
|
|
except Exception as e:
|
|
print(e)
|
|
return Response(
|
|
{"error": "Something went wrong please try again later"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|