a1c67cdfe7
Related to https://github.com/grafana/oncall/issues/4496 [No warnings](https://github.com/grafana/oncall/actions/runs/10359091611/job/28674729718?pr=4681#step:5:305)!
112 lines
3.6 KiB
Python
112 lines
3.6 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_cache_key_update_incident_slack_message(alert_group_pk: str) -> str:
|
|
CACHE_KEY_PREFIX = "update_incident_slack_message"
|
|
return f"{CACHE_KEY_PREFIX}_{alert_group_pk}"
|
|
|
|
|
|
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}"
|