From e575ffc4e92a825c7117ae81e4db50803149f274 Mon Sep 17 00:00:00 2001 From: giannibalistreri Date: Thu, 28 Dec 2023 00:20:06 +0100 Subject: [PATCH] Refactoring --- .../v1/slack_alerting/task_container/src/task.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/kubeflow_templates/kubeflow_pipelines/task_pool/v1/slack_alerting/task_container/src/task.py b/kubeflow_templates/kubeflow_pipelines/task_pool/v1/slack_alerting/task_container/src/task.py index 99f7619..221f417 100644 --- a/kubeflow_templates/kubeflow_pipelines/task_pool/v1/slack_alerting/task_container/src/task.py +++ b/kubeflow_templates/kubeflow_pipelines/task_pool/v1/slack_alerting/task_container/src/task.py @@ -9,6 +9,7 @@ import json import urllib3 +from custom_logger import Log from typing import NamedTuple PARSER = argparse.ArgumentParser(description="slack alerting") @@ -46,6 +47,12 @@ def slack_alerting(msg: str, -> abort: Abort Kubeflow Pipeline -> succeed: End Kubeflow Pipline successfully + :param aws_region: str + Code of the AWS region + + :param slack_channel: str + Name of the Slack channel + :return: NamedTuple Status code of the request """ @@ -83,7 +90,7 @@ def slack_alerting(msg: str, "elements": [ { "type": "mrkdwn", - "text": "Region: *eu-central-1*" + "text": f"Region: *{aws_region}*" } ] } @@ -165,6 +172,7 @@ def slack_alerting(msg: str, raise SlackAlertingException(f'Pipeline status ({pipeline_status}) not supported') _encoded_msg: bytes = json.dumps(_slack_msg).encode('utf-8') _response = _http.request(method='POST', url=_slack_url, body=_encoded_msg) + Log().log(msg=f'Slack message send to channel "{slack_channel}" with response status "{_response.status}": {_slack_msg}') return [_slack_msg, _response.status]