lps_gen.py: Update template_read.

* lps_gen.py (template_read): Add exception handling for non-existent
  templates.
* tests/test_lps_gen.py (TestTemplates.test_read): Add non-existent
  template test.

2 files changed, 10 insertions, 1 deletions

diff --git a/lps_gen.py b/lps_gen.py
index 13bf7a3..49bc477 100644
--- a/lps_gen.py
+++ b/lps_gen.py
@@ -99,7 +99,13 @@ def template_read(name):
     p = 'lpschedule_generator'
     r = 'data/{}'.format(name)
 
-    return pkgr.resource_string(p, r).decode('utf-8')
+    t = None
+    try:
+        t = pkgr.resource_string(p, r).decode('utf-8')
+    except Exception as e:
+        print(e, file=sys.stderr)
+
+    return t
 
 
 class LPiCal(object):
diff --git a/tests/test_lps_gen.py b/tests/test_lps_gen.py
index 26c6132..c4ff1dd 100644
--- a/tests/test_lps_gen.py
+++ b/tests/test_lps_gen.py
@@ -113,6 +113,9 @@ class TestTemplates(object):
         assert type(t) is str
         assert len(t) > 0
 
+        t = template_read('nonexistent.jinja2')
+        assert t is None
+
 
 class TestLPiCal(object):
     """