Make it easier for sublcasses to modify the email footer

mythmon · Aug 16, 2011 · c199a15 · c199a15
1 parent 59c0055
commit c199a15
Showing 1 changed file with 18 additions and 8 deletions.
diff --git a/python/nagcat/plugins/notify_smtp.py b/python/nagcat/plugins/notify_smtp.py
@@ -100,14 +100,21 @@ def headers(self):
 
     def body(self):
         text = super(EmailNotification, self).body()
-        if self.format == "long":
+        text += self.footer()
-            urls = self.urls()
+        return text
-            if urls:
+
-                text += "\n"
+    def footer(self):
-                if 'nagios' in urls:
+        urls = self.urls()
-                    text += "Nagios: %s\n" % urls['nagios']
+
-                if 'graphs' in urls:
+        if urls:
-                    text += "Graphs: %s\n" % urls['graphs']
+            text = "\n"
+            if 'nagios' in urls:
+                text += "Nagios: %s\n" % urls['nagios']
+            if 'graphs' in urls:
+                text += "Graphs: %s\n" % urls['graphs']
+        else:
+            text = ""
+
         return text
 
     def send(self):
@@ -169,6 +176,9 @@ def headers(self):
         headers['To'] = self.macros['CONTACTPAGER']
         return headers
 
+    def footer():
+        return ""
+
     def graph(self):
         return None