9dde1805aa
# What this PR does - Adds [`mypy` static type checking](https://mypy-lang.org/) to our CI pipeline. Currently there is still a **ton** of errors being returned by the tool, as we'll need to fix pre-existing errors. I think we can slowly chip away at these errors in small PRs, doing them all in one large PR is likely very risky. - Also, this PR starts chipping away at one of the main type errors that we have which is accessing the `datetime` class (from the `datetime` library) or `timedelta` function on the `django.utils.timezone` module. Basically we should be instead accessing these two objects from the native `datetime` module. This makes sense because the [`__all__` attribute](https://github.com/django/django/blob/main/django/utils/timezone.py#L14-L30) in `django.utils.timezone` does not re-export `datetime` or `timedelta`. - splits `engine` dependencies out into `requirements.txt` and `requirements-dev.txt` ## Checklist - [ ] Unit, integration, and e2e (if applicable) tests updated (N/A) - [ ] Documentation added (or `pr:no public docs` PR label added if not required) (N/A) - [ ] `CHANGELOG.md` updated (or `pr:no changelog` PR label added if not required) (N/A)
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from django.db.models import Prefetch
|
|
from rest_framework import serializers
|
|
|
|
from apps.alerts.models import AlertGroup
|
|
from apps.telegram.models.message import TelegramMessage
|
|
from common.api_helpers.mixins import EagerLoadingMixin
|
|
from common.constants.alert_group_restrictions import IS_RESTRICTED_TITLE
|
|
|
|
|
|
class IncidentSerializer(EagerLoadingMixin, serializers.ModelSerializer):
|
|
id = serializers.CharField(read_only=True, source="public_primary_key")
|
|
integration_id = serializers.CharField(source="channel.public_primary_key")
|
|
route_id = serializers.SerializerMethodField()
|
|
created_at = serializers.DateTimeField(source="started_at")
|
|
alerts_count = serializers.SerializerMethodField()
|
|
title = serializers.SerializerMethodField()
|
|
state = serializers.SerializerMethodField()
|
|
|
|
SELECT_RELATED = ["channel", "channel_filter", "slack_message", "channel__organization"]
|
|
PREFETCH_RELATED = [
|
|
"alerts",
|
|
Prefetch(
|
|
"telegram_messages",
|
|
TelegramMessage.objects.filter(chat_id__startswith="-", message_type=TelegramMessage.ALERT_GROUP_MESSAGE),
|
|
to_attr="prefetched_telegram_messages",
|
|
),
|
|
]
|
|
|
|
class Meta:
|
|
model = AlertGroup
|
|
fields = [
|
|
"id",
|
|
"integration_id",
|
|
"route_id",
|
|
"alerts_count",
|
|
"state",
|
|
"created_at",
|
|
"resolved_at",
|
|
"acknowledged_at",
|
|
"title",
|
|
"permalinks",
|
|
]
|
|
|
|
def get_title(self, obj):
|
|
return IS_RESTRICTED_TITLE if obj.is_restricted else obj.web_title_cache
|
|
|
|
def get_alerts_count(self, obj):
|
|
return len(obj.alerts.all())
|
|
|
|
def get_state(self, obj):
|
|
return obj.state
|
|
|
|
def get_route_id(self, obj):
|
|
if obj.channel_filter is not None:
|
|
return obj.channel_filter.public_primary_key
|
|
else:
|
|
return None
|