335c8fe65b
# What this PR does ## Which issue(s) this PR fixes ## Checklist - [ ] Tests updated - [ ] Documentation added - [ ] `CHANGELOG.md` updated --------- Co-authored-by: Joey Orlando <joey.orlando@grafana.com>
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
from django.db.models import CharField
|
|
from django.db.models.functions import Cast
|
|
from django_filters import rest_framework as filters
|
|
from rest_framework import mixins
|
|
from rest_framework.permissions import IsAuthenticated
|
|
from rest_framework.viewsets import GenericViewSet
|
|
|
|
from apps.alerts.models import Alert
|
|
from apps.auth_token.auth import ApiTokenAuthentication
|
|
from apps.public_api.serializers.alerts import AlertSerializer
|
|
from apps.public_api.throttlers.user_throttle import UserThrottle
|
|
from common.api_helpers.mixins import RateLimitHeadersMixin
|
|
from common.api_helpers.paginators import FiftyPageSizePaginator
|
|
|
|
|
|
class AlertFilter(filters.FilterSet):
|
|
id = filters.CharFilter(field_name="public_primary_key")
|
|
|
|
|
|
class AlertView(RateLimitHeadersMixin, mixins.ListModelMixin, GenericViewSet):
|
|
authentication_classes = (ApiTokenAuthentication,)
|
|
permission_classes = (IsAuthenticated,)
|
|
|
|
throttle_classes = [UserThrottle]
|
|
|
|
model = Alert
|
|
serializer_class = AlertSerializer
|
|
pagination_class = FiftyPageSizePaginator
|
|
|
|
filter_backends = (filters.DjangoFilterBackend,)
|
|
filterset_class = AlertFilter
|
|
|
|
def get_queryset(self):
|
|
alert_group_id = self.request.query_params.get("alert_group_id", None)
|
|
search = self.request.query_params.get("search", None)
|
|
|
|
queryset = Alert.objects.filter(group__channel__organization=self.request.auth.organization)
|
|
|
|
if alert_group_id:
|
|
queryset = queryset.filter(group__public_primary_key=alert_group_id)
|
|
|
|
if search:
|
|
queryset = queryset.annotate(
|
|
raw_request_data_str=Cast("raw_request_data", output_field=CharField())
|
|
).filter(raw_request_data_str__icontains=search)
|
|
|
|
queryset = self.serializer_class.setup_eager_loading(queryset)
|
|
|
|
return queryset
|