Add slack notification for start/finish

.env.example

@@ -0,0 +1,4 @@
+# Optional channel, default to whatever is setup on webhook
+# SLACK_CHANNEL=
+# Optional webhook URL. If no webhook is provided, the command will not send any messages.
+# SLACK_WEBHOOK_URL=

.gitignore

@@ -1,9 +1,7 @@
-/.vscode/
+/*.egg-info
+/.env
 /.venv/
-
 /build/
 /dist/
-/*.egg-info
-__pycache__
-
 /output.json
+__pycache__

README.md

@@ -19,7 +19,9 @@ _Preview release, please do not use for anything important!_
 
 ```bash
 $ pip install terraecs
+$ cp .env.example .env
 ```
+Edit the `.env` file to your taste for Slack notifications.
 
 ## Usage
 

setup.py

@@ -22,6 +22,8 @@
     install_requires=[
       'boto3',
       'click',
+      'python-dotenv',
+      'requests',
     ],
     entry_points={
         'console_scripts': [

terraecs/__main__.py

@@ -2,8 +2,17 @@
 import click
 import json
 import logging
+import os
+import requests
 import time
 
+from dotenv import load_dotenv
+load_dotenv()
+
+
+SLACK_CHANNEL = os.getenv('SLACK_CHANNEL')
+SLACK_WEBHOOK_URL = os.getenv('SLACK_WEBHOOK_URL')
+
 
 class ContextFilter(logging.Filter):
     def filter(self, record):
@@ -26,6 +35,19 @@ def filter(self, record):
 
 output_file = None
 
+def notify_slack(text):
+    if not SLACK_WEBHOOK_URL:
+        return
+
+    payload = {
+        'text': text,
+    }
+
+    if SLACK_CHANNEL:
+        payload['channel'] = SLACK_CHANNEL
+
+    requests.post(SLACK_WEBHOOK_URL, json=payload)
+
 def log(message, level=logging.ERROR, **context):
     logger.log(level, message, extra={'context': context})
 
@@ -62,6 +84,7 @@ def run(command):
             log(f'could not find the {required_key}... Aborting')
             exit(1)
 
+    notify_slack(f'`terraecs run {" ".join(command)}` started')
 
     ecs_client = boto3.client('ecs')
     logs_client = boto3.client('logs')
@@ -142,7 +165,11 @@ def run(command):
 
         time.sleep(sleep_seconds)
 
-    exit(get_exit_code(response['tasks'][0]['containers']))
+    exit_code = get_exit_code(response['tasks'][0]['containers'])
+
+    notify_slack(f'`terraecs run {" ".join(command)}` finished with exit code {exit_code}')
+
+    exit(exit_code)
 
 if __name__ == "__main__":
     cli(None)