Upgrade command create_notification_user

- Add remarks parameter. Better with a comment to have information about the telefon used when creating a SMS notification user.
volter · Mar 30, 2016 · 7e685aa · 7e685aa
1 parent 2508bcc
commit 7e685aa
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/zabbix_cli/cli.py b/zabbix_cli/cli.py
@@ -2969,7 +2969,7 @@ def do_create_notification_user(self,args):
         COMMAND:
         create_notification_user [sendto]
                                  [mediatype]
-                    
+                                 [remarks]
         [sendto]
         --------
         E-mail address or SMS number
@@ -2979,6 +2979,11 @@ def do_create_notification_user(self,args):
         One of the media types names defined in your Zabbix
         installation, e.g.  Email, SMS
 
+        [remarks]
+        ---------
+        Comments about this user. e.g. Johns cellphone.
+        Max lenght is 20 characters.
+
         '''
 
         # Default: md5 value of a random int >1 and <1000000 
@@ -3016,6 +3021,7 @@ def do_create_notification_user(self,args):
                 print '--------------------------------------------------------'
                 sendto = raw_input('# SendTo []: ').strip()
                 mediatype = raw_input('# Media type []: ').strip()
+                remarks = raw_input('# Remarks []: ').strip()
                 print '--------------------------------------------------------'
 
             except Exception as e:
@@ -3027,10 +3033,11 @@ def do_create_notification_user(self,args):
         # Command with parameters
         #
 
-        elif len(arg_list) == 2:
+        elif len(arg_list) == 3:
 
             sendto = arg_list[0].strip()
             mediatype = arg_list[1].strip()
+            remarks = arg_list[2].strip()
 
         #
         # Command with the wrong number of parameters
@@ -3052,7 +3059,11 @@ def do_create_notification_user(self,args):
             self.generate_feedback('Error','Media type is empty')
             return False
 
-        alias = 'notification-user-' + sendto.replace('.','-')
+        if remarks.strip() == '':
+            alias = 'notification-user-' + sendto.replace('.','-')
+        else:
+            alias = 'notification-user-' + remarks.strip()[:20].replace(' ','_') + '-' + sendto.replace('.','-')
+
         passwd = passwd_default
         type = type_default
         autologin = autologin_default