|
1 | 1 | from typing import NotRequired, TypedDict
|
2 | 2 |
|
3 | 3 | from drf_spectacular.utils import extend_schema
|
4 |
| -from rest_framework import serializers, status |
5 |
| -from rest_framework.response import Response |
| 4 | +from rest_framework import serializers |
6 | 5 |
|
7 | 6 | from sentry.api.api_owners import ApiOwner
|
8 | 7 | from sentry.api.api_publish_status import ApiPublishStatus
|
|
19 | 18 | from sentry.apidocs.parameters import GlobalParams
|
20 | 19 | from sentry.apidocs.utils import inline_sentry_response_serializer
|
21 | 20 | from sentry.workflow_engine.endpoints.serializers import DataConditionHandlerSerializer
|
22 |
| -from sentry.workflow_engine.models.data_condition import IGNORED_CONDITIONS |
| 21 | +from sentry.workflow_engine.models.data_condition import LEGACY_CONDITIONS |
23 | 22 | from sentry.workflow_engine.registry import condition_handler_registry
|
24 | 23 | from sentry.workflow_engine.types import DataConditionHandler
|
25 | 24 |
|
26 | 25 |
|
27 |
| -class DataConditionRequestSerializer(serializers.Serializer): |
28 |
| - type = serializers.ChoiceField( |
29 |
| - choices=[type.value for type in DataConditionHandler.Type], |
30 |
| - ) |
31 |
| - |
32 |
| - |
33 | 26 | class DataConditionHandlerResponse(TypedDict):
|
34 | 27 | condition_id: str
|
35 | 28 | type: str
|
@@ -63,17 +56,16 @@ def get(self, request, organization):
|
63 | 56 | """
|
64 | 57 | Returns a list of data conditions for a given org
|
65 | 58 | """
|
66 |
| - serializer = DataConditionRequestSerializer(data=request.GET) |
67 |
| - if not serializer.is_valid(): |
68 |
| - return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) |
69 |
| - serialized = serializer.validated_data |
70 |
| - |
71 |
| - type = serialized.get("type") |
| 59 | + type = request.GET.get("type") |
| 60 | + try: |
| 61 | + DataConditionHandler.Type(type) |
| 62 | + except ValueError: |
| 63 | + raise serializers.ValidationError("Invalid group") |
72 | 64 |
|
73 | 65 | data_conditions = []
|
74 | 66 |
|
75 | 67 | for condition, handler in condition_handler_registry.registrations.items():
|
76 |
| - if condition not in IGNORED_CONDITIONS and handler.type == type: |
| 68 | + if condition not in LEGACY_CONDITIONS and handler.type == type: |
77 | 69 | condition_json = {"condition_id": condition}
|
78 | 70 |
|
79 | 71 | condition_json.update(
|
|
0 commit comments