oncall-engine/engine/apps/slack/utils.py
Joey Orlando 26946f0d43
fix: improve Slack rate limiting logic when updating alert groups (#5287)
# What this PR does

https://www.loom.com/share/1ac33822301444748133ffe72638ddc4

The two asks in the [original GH
issue](https://github.com/grafana/oncall-private/issues/2947) were:

> 1. Make the error message clearer. We can identify if it's delivering
or updating and being rate-limited. This is possible because Slack sets
limits per API method. Also, this limit is a per-slack channel while we
are posting messages & applying ratelimit per on-call integration, which
confuses customers.
> 2. Debounce update alert group message in Slack

Both of these have been addressed in this PR

## Which issue(s) this PR closes

Closes https://github.com/grafana/oncall-private/issues/2947

## 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.
2024-12-02 14:40:30 -05:00

107 lines
3.4 KiB
Python

import datetime
import enum
import typing
from apps.slack.client import SlackClient
from apps.slack.errors import (
SlackAPIChannelArchivedError,
SlackAPIChannelNotFoundError,
SlackAPIInvalidAuthError,
SlackAPITokenError,
)
if typing.TYPE_CHECKING:
from apps.user_management.models import Organization
class SlackDateFormat(enum.StrEnum):
"""
https://api.slack.com/reference/surfaces/formatting#date-formatting
"""
DATE_NUM = "date_num"
"""
Displayed as `2014-02-18`. It will include leading zeros before the month and date and is
probably best for more technical integrations that require a developer-friendly date format.
"""
DATE = "date"
"""
Displayed as `February 18th, 2014`. The year will be omitted if the date is less than six months in the past or future.
"""
DATE_SHORT = "date_short"
"""
Displayed as `Feb 18, 2014`. The year will be omitted if the date is less than six months in the past or future.
"""
DATE_LONG = "date_long"
"""
Displayed as `Tuesday, February 18th, 2014`. The year will be omitted if the date is less than six months in the past or future.
"""
DATE_PRETTY = "date_pretty"
"""
Displays the same as `{date}` but uses "yesterday", "today", or "tomorrow" where appropriate.
"""
DATE_SHORT_PRETTY = "date_short_pretty"
"""
Displays the same as `{date_short}` but uses "yesterday", "today", or "tomorrow" where appropriate.
"""
DATE_LONG_PRETTY = "date_long_pretty"
"""
Displays the same as `{date_long}` but uses "yesterday", "today", or "tomorrow" where appropriate.
"""
TIME = "time"
"""
Displayed as `6:39 AM` or `6:39 PM` in 12-hour format. If the client is set to show 24-hour format, it is displayed as `06:39` or `18:39`.
"""
TIME_SECS = "time_secs"
"""
Displayed as `6:39:45 AM` `6:39:42 PM` in 12-hour format. In 24-hour format it is displayed as `06:39:45` or `18:39:42`.
"""
def post_message_to_channel(organization: "Organization", channel_id: str, text: str) -> None:
if not organization.slack_team_identity:
return
slack_client = SlackClient(organization.slack_team_identity, enable_ratelimit_retry=True)
try:
slack_client.chat_postMessage(channel=channel_id, text=text)
except (
SlackAPITokenError,
SlackAPIInvalidAuthError,
SlackAPIChannelNotFoundError,
SlackAPIChannelArchivedError,
):
pass
def _format_datetime_to_slack(timestamp: float, format: str) -> str:
fallback = datetime.datetime.fromtimestamp(timestamp, datetime.UTC).strftime("%Y-%m-%d %H:%M (UTC)")
return f"<!date^{int(timestamp)}^{format}|{fallback}>"
def format_datetime_to_slack(timestamp: float, format: SlackDateFormat = SlackDateFormat.DATE_SHORT) -> str:
"""
See the docs [here](https://api.slack.com/reference/surfaces/formatting#date-formatting) for
more information
"""
return _format_datetime_to_slack(timestamp, f"{{{format}}}")
def format_datetime_to_slack_with_time(timestamp: float, format: SlackDateFormat = SlackDateFormat.DATE_SHORT) -> str:
"""
See the docs [here](https://api.slack.com/reference/surfaces/formatting#date-formatting) for
more information
"""
return _format_datetime_to_slack(timestamp, f"{{{format}}} {{time}}")
def get_populate_slack_channel_task_id_key(slack_team_identity_id: str) -> str:
return f"SLACK_CHANNELS_TASK_ID_TEAM_{slack_team_identity_id}"