From: Samuel Colvin <s@muelcolvin.com>
Date: Mon, 19 Jun 2017 16:15:36 +0000 (+0100)
Subject: add custom exception: ChevronError
X-Git-Url: https://git.devinivas.org/?a=commitdiff_plain;h=460a62b547bb3f3099c6c77c062d641db8fc38c0;p=chevron.git

add custom exception: ChevronError
---

diff --git a/chevron/__init__.py b/chevron/__init__.py
index 0d4ad77..d34cdac 100644
--- a/chevron/__init__.py
+++ b/chevron/__init__.py
@@ -1,4 +1,5 @@
 from .main import main, cli_main
 from .renderer import render
+from .tokenizer import ChevronError
 
-__all__ = ['main', 'render', 'cli_main']
+__all__ = ['main', 'render', 'cli_main', 'ChevronError']
diff --git a/chevron/tokenizer.py b/chevron/tokenizer.py
index 44f4eac..ea3f6b2 100644
--- a/chevron/tokenizer.py
+++ b/chevron/tokenizer.py
@@ -3,10 +3,14 @@ _CURRENT_LINE = 1
 _LAST_TAG_LINE = None
 
 
+class ChevronError(SyntaxError):
+    pass
+
 #
 # Helper functions
 #
 
+
 def grab_literal(template, l_del):
     """Parse a literal from the template"""
 
@@ -95,8 +99,8 @@ def parse_tag(template, l_del, r_del):
 
         # Otherwise we should complain
         else:
-            raise SyntaxError('unclosed set delimiter tag\n'
-                              'at line {0}'.format(_CURRENT_LINE))
+            raise ChevronError('unclosed set delimiter tag\n'
+                               'at line {0}'.format(_CURRENT_LINE))
 
     # If we might be a no html escape tag
     elif tag_type == 'no escape?':
@@ -200,11 +204,11 @@ def tokenize(template, def_ldel='{{', def_rdel='}}'):
             last_section = open_sections.pop()
             if tag_key != last_section:
                 # Otherwise we need to complain
-                raise SyntaxError('Trying to close tag "{0}"\n'
-                                  'last open tag is "{1}"\n'
-                                  'line {2}'
-                                  .format(tag_key, last_section,
-                                          _CURRENT_LINE + 1))
+                raise ChevronError('Trying to close tag "{0}"\n'
+                                   'last open tag is "{1}"\n'
+                                   'line {2}'
+                                   .format(tag_key, last_section,
+                                           _CURRENT_LINE + 1))
 
         # Do the second check to see if we're a standalone
         is_standalone = r_sa_check(template, tag_type, is_standalone)
@@ -231,7 +235,7 @@ def tokenize(template, def_ldel='{{', def_rdel='}}'):
     # If there are any open sections when we're done
     if open_sections:
         # Then we need to complain
-        raise SyntaxError('Unexpected EOF\n'
-                          'the tag "{0}" was never closed\n'
-                          'was opened at line {1}'
-                          .format(open_sections[-1], _LAST_TAG_LINE))
+        raise ChevronError('Unexpected EOF\n'
+                           'the tag "{0}" was never closed\n'
+                           'was opened at line {1}'
+                           .format(open_sections[-1], _LAST_TAG_LINE))
diff --git a/test_spec.py b/test_spec.py
index 8d1c662..4fb5d14 100755
--- a/test_spec.py
+++ b/test_spec.py
@@ -61,8 +61,10 @@ class ExpandedCoverage(unittest.TestCase):
             'template': '{{# section }} end of file'
         }
 
+        self.assertRaises(chevron.ChevronError, chevron.render, **test1)
+        self.assertRaises(chevron.ChevronError, chevron.render, **test2)
+        # check SyntaxError still catches ChevronError:
         self.assertRaises(SyntaxError, chevron.render, **test1)
-        self.assertRaises(SyntaxError, chevron.render, **test2)
 
     def test_bad_set_delimiter_tag(self):
         args = {