add alert details using slack block kit (#1751)

* add alert details using slack block kit

* tidy up

* get rid of fstrings

metaflow/plugins/argo/argo_workflows.py

@@ -1540,7 +1540,6 @@ def _container_templates(self):
 
  # Return exit hook templates for workflow execution notifications.
  def _exit_hook_templates(self):
- # TODO: Add details to slack message
  templates = []
  if self.notify_on_error:
  templates.append(self._slack_error_template())
@@ -1649,36 +1648,100 @@ def _pager_duty_notification_links(self):
 
  return links
 
+ def _get_slack_blocks(self, message):
+ """
+ Use Slack's Block Kit to add general information about the environment and
+ execution metadata, including a link to the UI and an optional message.
+ """
+ ui_link = "%s%s/argo-{{workflow.name}}" % (UI_URL, self.flow.name)
+ # fmt: off
+ if getattr(current, "project_name", None):
+ # Add @project metadata when available.
+ environment_details_block = {
+ "type": "section",
+ "text": {
+ "type": "mrkdwn",
+ "text": ":metaflow: Environment details"
+ },
+ "fields": [
+ {
+ "type": "mrkdwn",
+ "text": "*Project:* %s" % current.project_name 
+ },
+ {
+ "type": "mrkdwn",
+ "text": "*Project Branch:* %s" % current.branch_name
+ }
+ ]
+ }
+ else:
+ environment_details_block = {
+ "type": "section",
+ "text": {
+ "type": "mrkdwn",
+ "text": ":metaflow: Environment details"
+ }
+ }
+
+ blocks = [
+ environment_details_block,
+ {
+ "type": "context",
+ "elements": [
+ {
+ "type": "mrkdwn",
+ "text": " :information_source: *<%s>*" % ui_link,
+ }
+ ],
+ },
+ {
+ "type": "divider"
+ },
+ ]
+
+ if message:
+ blocks += [
+ {
+ "type": "section",
+ "text": {
+ "type": "mrkdwn",
+ "text": message
+ }
+ }
+ ]
+ # fmt: on
+ return blocks
+
  def _slack_error_template(self):
  if self.notify_slack_webhook_url is None:
  return None
+
+ message = (
+ ":rotating_light: _%s/argo-{{workflow.name}}_ failed!" % self.flow.name
+ )
+ payload = {"text": message}
+ if UI_URL:
+ blocks = self._get_slack_blocks(message)
+ payload = {"text": message, "blocks": blocks}
+
  return Template("notify-slack-on-error").http(
- Http("POST")
- .url(self.notify_slack_webhook_url)
- .body(
- json.dumps(
- {
- "text": ":rotating_light: _%s/argo-{{workflow.name}}_ failed!"
- % self.flow.name
- }
- )
- )
+ Http("POST").url(self.notify_slack_webhook_url).body(json.dumps(payload))
  )
 
  def _slack_success_template(self):
  if self.notify_slack_webhook_url is None:
  return None
+
+ message = (
+ ":white_check_mark: _%s/argo-{{workflow.name}}_ succeeded!" % self.flow.name
+ )
+ payload = {"text": message}
+ if UI_URL:
+ blocks = self._get_slack_blocks(message)
+ payload = {"text": message, "blocks": blocks}
+
  return Template("notify-slack-on-success").http(
- Http("POST")
- .url(self.notify_slack_webhook_url)
- .body(
- json.dumps(
- {
- "text": ":white_check_mark: _%s/argo-{{workflow.name}}_ succeeded!"
- % self.flow.name
- }
- )
- )
+ Http("POST").url(self.notify_slack_webhook_url).body(json.dumps(payload))
  )
 
  def _compile_sensor(self):