Check for Tautulli footer in newsletters

plexpy/newsletters.py

@@ -19,6 +19,7 @@ from itertools import groupby
 from mako.lookup import TemplateLookup
 from mako import exceptions
 import os
+import re
 
 import plexpy
 import common
@@ -420,7 +421,7 @@ class Newsletter(object):
 
         self.retrieve_data()
 
-        return serve_template(
+        newsletter_rendered = serve_template(
             templatename=self._TEMPLATE,
             uuid=self.uuid,
             subject=self.subject_formatted,
@@ -431,6 +432,25 @@ class Newsletter(object):
             preview=self.is_preview
         )
 
+        # Force Tautulli footer
+        if '<!-- FOOTER MESSAGE - DO NOT REMOVE -->' in newsletter_rendered:
+            newsletter_rendered = newsletter_rendered.replace(
+                '<!-- FOOTER MESSAGE - DO NOT REMOVE -->',
+                'Newsletter generated by <a href="https://tautulli.com" target="_blank" '
+                'style="text-decoration: underline;color: #fff;font-size: 12px;">Tautulli</a>.'
+            )
+            return newsletter_rendered
+        else:
+            msg = ('<div style="text-align: center;padding-top: 100px;padding-bottom: 100px;">'
+                   '<p style="font-family: \'Open Sans\', Helvetica, Arial, sans-serif;color: #282A2D;'
+                   'font-size: 18px;line-height: 30px;">'
+                   'The Tautulli newsletter footer was removed from the newsletter template.<br>'
+                   'Please leave the footer in place as it is unobtrusive and supports '
+                   '<a href="https://tautulli.com" target="_blank">Tautulli</a>.<br>Thank you.'
+                   '</p></div>')
+            newsletter_rendered = re.sub(r'(<body.*?>)', r'\1' + msg, newsletter_rendered)
+            return newsletter_rendered
+
     def send(self):
         self.newsletter = self.generate_newsletter()