Update send_slack_notification.py

writechoiceorg · Apr 11, 2024 · 8976f6c · 8976f6c
1 parent 5eae31b
commit 8976f6c
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/send_slack_notification.py b/send_slack_notification.py
@@ -1,9 +1,13 @@
 import requests
+import os
 import sys
 
 
-def send_slack_notification(message, assignee, origin_repo, webhook_url):
-    print(assignee)
+def send_slack_notification(message, assignee, origin_repo):
+    webhook_url = os.environ.get("SLACK_WEBHOOK_URL")
+    if webhook_url is None:
+        print("SLACK_WEBHOOK_URL environment variable is not set.")
+        sys.exit(1)
     data = {"Message": message, "Assignee": assignee, "Origin_Repo": origin_repo}
     print(data)
     response = requests.post(webhook_url, json=data)
@@ -21,7 +25,4 @@ def send_slack_notification(message, assignee, origin_repo, webhook_url):
     message = sys.argv[1]
     assignee = sys.argv[2]
     origin_repo = sys.argv[3]
-    webhook_url = sys.argv[
-        4
-    ]  # Assuming the webhook URL is passed as a command-line argument
-    send_slack_notification(message, assignee, origin_repo, webhook_url)
+    send_slack_notification(message, assignee, origin_repo)