7aa78f5f73
Enables [flake8-bugbear](https://github.com/PyCQA/flake8-bugbear), checking for bugs/design problems, and [fixes the issues found](https://pastebin.com/fEDBz6Ta) (some interesting ones, particularly with mutable args). Related to https://github.com/grafana/oncall/pull/3448
40 lines
1.6 KiB
Python
40 lines
1.6 KiB
Python
from rest_framework import serializers
|
|
|
|
from apps.alerts.models import EscalationChain
|
|
from common.api_helpers.custom_fields import TeamPrimaryKeyRelatedField
|
|
from common.api_helpers.utils import CurrentOrganizationDefault, CurrentTeamDefault
|
|
|
|
|
|
class EscalationChainSerializer(serializers.ModelSerializer):
|
|
id = serializers.CharField(read_only=True, source="public_primary_key")
|
|
organization = serializers.HiddenField(default=CurrentOrganizationDefault())
|
|
team = TeamPrimaryKeyRelatedField(allow_null=True, default=CurrentTeamDefault())
|
|
|
|
class Meta:
|
|
model = EscalationChain
|
|
fields = ("id", "name", "organization", "team")
|
|
|
|
|
|
class EscalationChainListSerializer(EscalationChainSerializer):
|
|
number_of_integrations = serializers.SerializerMethodField()
|
|
number_of_routes = serializers.SerializerMethodField()
|
|
|
|
class Meta(EscalationChainSerializer.Meta):
|
|
fields = [*EscalationChainSerializer.Meta.fields, "number_of_integrations", "number_of_routes"]
|
|
|
|
def get_number_of_integrations(self, obj):
|
|
# num_integrations param added in queryset via annotate. Check EscalationChainViewSet.get_queryset
|
|
return getattr(obj, "num_integrations", 0)
|
|
|
|
def get_number_of_routes(self, obj):
|
|
# num_routes param added in queryset via annotate. Check EscalationChainViewSet.get_queryset
|
|
return getattr(obj, "num_routes", 0)
|
|
|
|
|
|
class FilterEscalationChainSerializer(serializers.ModelSerializer):
|
|
value = serializers.CharField(source="public_primary_key")
|
|
display_name = serializers.CharField(source="name")
|
|
|
|
class Meta:
|
|
model = EscalationChain
|
|
fields = ["value", "display_name"]
|