From 38c14b940c4bed5ee9e5256272f2afea7030d84a Mon Sep 17 00:00:00 2001
From: Julien Riou <julien@riou.xyz>
Date: Thu, 26 Nov 2020 08:59:14 +0100
Subject: [PATCH] Move Jinja templates to files

Signed-off-by: Julien Riou <julien@riou.xyz>
---
 .pre-commit-config.yaml |  1 -
 notify-by-telegram.py   | 30 +++++++++---------------------
 templates/host.md.j2    |  7 +++++++
 templates/service.md.j2 |  9 +++++++++
 4 files changed, 25 insertions(+), 22 deletions(-)
 create mode 100644 templates/host.md.j2
 create mode 100644 templates/service.md.j2

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 3880860..503fbbb 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -26,4 +26,3 @@ repos:
     hooks:
       - id: pydocstyle
         args: ['--config=.pydocstyle', '--match="(?!test_).*\.py"']
-  - repo: https://github.com/jumanjihouse/pre-commit-hooks
diff --git a/notify-by-telegram.py b/notify-by-telegram.py
index 7abe5cc..8410ac2 100755
--- a/notify-by-telegram.py
+++ b/notify-by-telegram.py
@@ -5,7 +5,7 @@ import logging
 import os
 
 import requests
-from jinja2 import Template
+from jinja2 import Environment, FileSystemLoader
 
 logger = logging.getLogger(__name__)
 
@@ -78,15 +78,10 @@ def markdown_escape(text):
 
 def generate_host_payload(chat_id, args):
     payload = {'chat_id': chat_id, 'parse_mode': 'MarkdownV2'}
-    text = r"""\*\*\*\*\* Nagios \*\*\*\*\*
-*Notification Type:* {{notification_type}}
-*Host:* {{host_name}}
-*State:* {{host_state}}
-*Address:* {{host_address}}
-*Info:* {{host_output}}
-*Date/Time:* {{long_date_time}}
-"""
-    template = Template(text)
+    absolute_path = os.path.split(os.path.abspath(__file__))[0]
+    loader = FileSystemLoader(os.path.join(absolute_path, 'templates'))
+    env = Environment(loader=loader)
+    template = env.get_template('host.md.j2')
     text = template.render(notification_type=markdown_escape(args.notification_type),
                            host_name=markdown_escape(args.host_name), host_state=markdown_escape(args.host_state),
                            host_address=markdown_escape(args.host_address),
@@ -98,17 +93,10 @@ def generate_host_payload(chat_id, args):
 
 def generate_service_payload(chat_id, args):
     payload = {'chat_id': chat_id, 'parse_mode': 'MarkdownV2'}
-    text = r"""\*\*\*\*\* Nagios \*\*\*\*\*
-*Notification Type:* {{notification_type}}
-*Service:* {{service_desc}}
-*Host:* {{host_alias}}
-*Address:* {{host_address}}
-*State:* {{service_state}}
-*Date/Time:* {{long_date_time}}
-*Additional Info:*
-{{service_output}}
-"""
-    template = Template(text)
+    absolute_path = os.path.split(os.path.abspath(__file__))[0]
+    loader = FileSystemLoader(os.path.join(absolute_path, 'templates'))
+    env = Environment(loader=loader)
+    template = env.get_template('service.md.j2')
     text = template.render(notification_type=markdown_escape(args.notification_type),
                            service_desc=markdown_escape(args.service_desc), host_alias=markdown_escape(args.host_alias),
                            host_address=markdown_escape(args.host_address),
diff --git a/templates/host.md.j2 b/templates/host.md.j2
new file mode 100644
index 0000000..3a97671
--- /dev/null
+++ b/templates/host.md.j2
@@ -0,0 +1,7 @@
+\*\*\*\*\* Nagios \*\*\*\*\*
+*Notification Type:* {{notification_type}}
+*Host:* {{host_name}}
+*State:* {{host_state}}
+*Address:* {{host_address}}
+*Info:* {{host_output}}
+*Date/Time:* {{long_date_time}}
diff --git a/templates/service.md.j2 b/templates/service.md.j2
new file mode 100644
index 0000000..1e3ad7b
--- /dev/null
+++ b/templates/service.md.j2
@@ -0,0 +1,9 @@
+\*\*\*\*\* Nagios \*\*\*\*\*
+*Notification Type:* {{notification_type}}
+*Service:* {{service_desc}}
+*Host:* {{host_alias}}
+*Address:* {{host_address}}
+*State:* {{service_state}}
+*Date/Time:* {{long_date_time}}
+*Additional Info:*
+{{service_output}}