Fix username for slack bot

[perso/Immae/Projets/Puppet.git] / modules / role / files / cryptoportfolio / slack-notify.py

diff --git a/modules/role/files/cryptoportfolio/slack-notify.py b/modules/role/files/cryptoportfolio/slack-notify.py
index fe64fb68784fc22df1dc2eef9c8f9ebb12309973..1b3cd51006598e91f4a2276093bc7e27d125bc94 100644 (file)
--- a/modules/role/files/cryptoportfolio/slack-notify.py
+++ b/modules/role/files/cryptoportfolio/slack-notify.py
@@ -1,12 +1,18 @@
 #!/usr/bin/env python3
 
+import os
 import sys
 import json
 import urllib3
 
-project = sys.argv[1]
-url     = sys.argv[2]
-version = sys.argv[3]
+project = os.environ["P_PROJECT"]
+url     = os.environ["P_WEBHOOK"]
+version = os.environ["P_VERSION"]
+host    = os.environ["P_HOST"]
+if os.environ["P_HTTPS"] == "true":
+    scheme = "https://"
+else:
+    scheme = "http://"
 
 def post(url, data):
     urllib3.disable_warnings()
@@ -19,8 +25,14 @@ def post(url, data):
             body=encoded)
 
 data = {
+        "username": "Puppet",
         "icon_url": "https://learn.puppet.com/static/images/logos/Puppet-Logo-Mark-Amber.png",
-        "text": "Deployed {} on {}".format(version, project),
+        "text": "Deployed {} of {} on {}{}".format(
+            version,
+            project,
+            scheme,
+            host,
+            ),
         }
 
 json_data = json.dumps(data)