fda05a6cc4
# What this PR does See [Slack discussion](https://raintank-corp.slack.com/archives/C06K1MQ07GS/p1732110700877869) for more context ## Checklist - [x] Unit, integration, and e2e (if applicable) tests updated - [x] Documentation added (or `pr:no public docs` PR label added if not required) - [x] Added the relevant release notes label (see labels prefixed w/ `release:`). These labels dictate how your PR will show up in the autogenerated release notes.
99 lines
3.4 KiB
Python
99 lines
3.4 KiB
Python
import logging
|
|
import typing
|
|
|
|
from apps.slack.client import SlackClient
|
|
from apps.slack.errors import (
|
|
SlackAPICantUpdateMessageError,
|
|
SlackAPIChannelArchivedError,
|
|
SlackAPIChannelInactiveError,
|
|
SlackAPIChannelNotFoundError,
|
|
SlackAPIInvalidAuthError,
|
|
SlackAPIMessageNotFoundError,
|
|
SlackAPIRatelimitError,
|
|
SlackAPITokenError,
|
|
)
|
|
|
|
if typing.TYPE_CHECKING:
|
|
from apps.alerts.models import AlertGroup
|
|
from apps.slack.models import SlackTeamIdentity
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class AlertGroupSlackService:
|
|
_slack_client: SlackClient
|
|
|
|
def __init__(
|
|
self,
|
|
slack_team_identity: "SlackTeamIdentity",
|
|
slack_client: typing.Optional[SlackClient] = None,
|
|
):
|
|
self.slack_team_identity = slack_team_identity
|
|
if slack_client is not None:
|
|
self._slack_client = slack_client
|
|
else:
|
|
self._slack_client = SlackClient(slack_team_identity)
|
|
|
|
def update_alert_group_slack_message(self, alert_group: "AlertGroup") -> None:
|
|
logger.info(f"Update message for alert_group {alert_group.pk}")
|
|
|
|
try:
|
|
self._slack_client.chat_update(
|
|
channel=alert_group.slack_message.channel_id,
|
|
ts=alert_group.slack_message.slack_id,
|
|
attachments=alert_group.render_slack_attachments(),
|
|
blocks=alert_group.render_slack_blocks(),
|
|
)
|
|
logger.info(f"Message has been updated for alert_group {alert_group.pk}")
|
|
except SlackAPIRatelimitError as e:
|
|
if not alert_group.channel.is_maintenace_integration:
|
|
if not alert_group.channel.is_rate_limited_in_slack:
|
|
alert_group.channel.start_send_rate_limit_message_task(e.retry_after)
|
|
logger.info(
|
|
f"Message has not been updated for alert_group {alert_group.pk} due to slack rate limit."
|
|
)
|
|
else:
|
|
raise
|
|
except (
|
|
SlackAPIMessageNotFoundError,
|
|
SlackAPICantUpdateMessageError,
|
|
SlackAPIChannelInactiveError,
|
|
SlackAPITokenError,
|
|
SlackAPIChannelNotFoundError,
|
|
):
|
|
pass
|
|
|
|
def publish_message_to_alert_group_thread(
|
|
self, alert_group: "AlertGroup", attachments=None, mrkdwn=True, unfurl_links=True, text=None
|
|
) -> None:
|
|
# TODO: refactor checking the possibility of sending message to slack
|
|
# do not try to post message to slack if integration is rate limited
|
|
if alert_group.channel.is_rate_limited_in_slack:
|
|
return
|
|
|
|
if attachments is None:
|
|
attachments = []
|
|
|
|
try:
|
|
result = self._slack_client.chat_postMessage(
|
|
channel=alert_group.slack_message.channel_id,
|
|
text=text,
|
|
attachments=attachments,
|
|
thread_ts=alert_group.slack_message.slack_id,
|
|
mrkdwn=mrkdwn,
|
|
unfurl_links=unfurl_links,
|
|
)
|
|
except (
|
|
SlackAPITokenError,
|
|
SlackAPIChannelArchivedError,
|
|
SlackAPIChannelNotFoundError,
|
|
SlackAPIInvalidAuthError,
|
|
):
|
|
return
|
|
|
|
alert_group.slack_messages.create(
|
|
slack_id=result["ts"],
|
|
organization=alert_group.channel.organization,
|
|
_slack_team_identity=self.slack_team_identity,
|
|
channel_id=alert_group.slack_message.channel_id,
|
|
)
|