From dbf318b6908347286a221fc1d50946fc17b5fa8b Mon Sep 17 00:00:00 2001
From: nemesgyadam <nemesgyadam@gmail.com>
Date: Mon, 21 Jun 2021 10:34:06 +0200
Subject: [PATCH] option for addition parameters

One can add a dictinionary with additional infos to show
---
 knockknock/slack_sender.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/knockknock/slack_sender.py b/knockknock/slack_sender.py
index 7ca63f3..36a93d7 100644
--- a/knockknock/slack_sender.py
+++ b/knockknock/slack_sender.py
@@ -9,7 +9,7 @@
 
 DATE_FORMAT = "%Y-%m-%d %H:%M:%S"
 
-def slack_sender(webhook_url: str, channel: str, user_mentions: List[str] = []):
+def slack_sender(webhook_url: str, channel: str, user_mentions: List[str] = [], host=None, infos = {}):
     """
     Slack sender wrapper: execute func, send a Slack notification with the end status
     (sucessfully finished or crashed) at the end. Also send a Slack notification before
@@ -35,7 +35,10 @@ def decorator_sender(func):
         def wrapper_sender(*args, **kwargs):
 
             start_time = datetime.datetime.now()
-            host_name = socket.gethostname()
+            if host is None:	
+                host_name = socket.gethostname()	
+            else:
+                host_name = str(host)
             func_name = func.__name__
 
             # Handling distributed training edge case.
@@ -54,6 +57,8 @@ def wrapper_sender(*args, **kwargs):
                             'Machine name: %s' % host_name,
                             'Main call: %s' % func_name,
                             'Starting date: %s' % start_time.strftime(DATE_FORMAT)]
+                for (key, value) in infos.items():	
+                    contents.append(key+': '+str(value))
                 contents.append(' '.join(user_mentions))
                 dump['text'] = '\n'.join(contents)
                 dump['icon_emoji'] = ':clapper:'
@@ -71,7 +76,8 @@ def wrapper_sender(*args, **kwargs):
                                 'Starting date: %s' % start_time.strftime(DATE_FORMAT),
                                 'End date: %s' % end_time.strftime(DATE_FORMAT),
                                 'Training duration: %s' % str(elapsed_time)]
-
+                    for (key, value) in infos.items():	
+                        contents.append(key+': '+str(value))
                     try:
                         str_value = str(value)
                         contents.append('Main call returned value: %s'% str_value)
@@ -98,6 +104,8 @@ def wrapper_sender(*args, **kwargs):
                             '%s\n\n' % ex,
                             "Traceback:",
                             '%s' % traceback.format_exc()]
+                for (key, value) in infos.items():	
+                    contents.append(key+': '+str(value))
                 contents.append(' '.join(user_mentions))
                 dump['text'] = '\n'.join(contents)
                 dump['icon_emoji'] = ':skull_and_crossbones:'