u

frontend/amundsen_application/api/file_upload/v0.py

@@ -89,6 +89,9 @@ def complete_multipart_upload():
             }
         )
 
+        message = f"*{app.config['HOST_ID']} - Metadata File Upload*\n{file_name}"
+        _send_notification(message=message)
+
         return make_response(jsonify(response), HTTPStatus.OK)
 
     except Exception as e:
@@ -112,4 +115,11 @@ def _get_s3_client():
     return s3_client
 
 def _get_s3_file_key(file_name: str) -> str:
-    return f'upload/{file_name}'
+    return f'upload/{file_name}'
+
+def _send_notification(message: str):
+    if app.config.get('SLACK_CLIENT'):
+        try:
+            app.config['SLACK_CLIENT'].post(channel=app.config['SLACK_SUPPORT_CHANNEL'], message=message)
+        except Exception as e:
+            LOGGER.exception(f'Failed to send Slack Notification')

frontend/amundsen_application/client/slack/slack_client.py

@@ -0,0 +1,46 @@
+# https://slack.dev/python-slack-sdk/web/index.html#messaging
+# https://pandeynandancse.medium.com/slack-send-attachments-using-python-webclient-baa39974ceea
+from pathlib import Path
+import logging
+
+from slack_sdk import WebClient
+from slack_sdk.errors import SlackApiError
+
+
+LOGGER = logging.getLogger(__name__)
+
+
+class SlackClient():
+    def __init__(self,
+                 token: str) -> None:
+
+        self.client = WebClient(token=token)
+
+
+    def post(self,
+             channel: str,
+             message: str):
+        try:
+            response = self.client.chat_postMessage(
+                channel=channel,
+                text=message
+            )
+        except SlackApiError as e:
+            LOGGER.exception(e)
+            raise e
+
+    def upload(self,
+               channel: str, # Must use the channel ID not the name
+               file_path: str,
+               title: str = None,
+               comment: str = None):
+        try:
+            response = self.client.files_upload_v2(
+                channel=channel,
+                file=file_path,
+                title=title,
+                initial_comment=comment
+            )
+        except SlackApiError as e:
+            LOGGER.exception(e)
+            raise e

frontend/amundsen_application/config.py

@@ -10,6 +10,8 @@
 
 from amundsen_application.tests.test_utils import get_test_user
 
+from amundsen_application.client.slack.slack_client import SlackClient
+
 
 class MatchRuleObject:
     def __init__(self,
@@ -157,6 +159,12 @@ class Config:
     AWS_ACCESS_KEY_ID = os.getenv('AWS_ACCESS_KEY_ID')
     AWS_SECRET_ACCESS_KEY = os.getenv('AWS_SECRET_ACCESS_KEY')
 
+    SLACK_TOKEN = os.getenv('SLACK_TOKEN')
+    SLACK_SUPPORT_CHANNEL = os.getenv('SLACK_SUPPORT_CHANNEL')
+
+    if SLACK_TOKEN:
+        SLACK_CLIENT = SlackClient(token=SLACK_TOKEN)
+
 
 class LocalConfig(Config):
     DEBUG = False