catch exception at entry point

Author: BruceZhang1993

index_generator/__main__.py

@@ -6,7 +6,9 @@
 import jinja2
 import argparse
 
+from jinja2.exceptions import TemplateNotFound
 from index_generator.models.entries import Entry
+from index_generator.models.exceptions import IndexGeneratorTemplateNotFound
 from . import *
 
 indexIgnore = ('index.html', 'images', 'favicon.ico')
@@ -29,9 +31,14 @@ def main():
         parser.add_argument('--human', action='store_true', default=False, help='Make size human readable.')
         parser.add_argument('path', type=str, default='', help='Path', nargs='?')
         arguments = parser.parse_args()
-        app(arguments)
-    except Exception as e:
-        print(e)
+        try:
+            app(arguments)
+        except TemplateNotFound as e:
+            raise IndexGeneratorTemplateNotFound(str(e))
+    except BaseException as e:
+        print('[Exception] ' + e.__class__.__name__ + ': ' + str(e))
+        if hasattr(e, 'hint'):
+            print(e.hint)
 
 
 def app(args):

index_generator/models/exceptions.py

@@ -1,2 +1,9 @@
 class IndexGeneratorBaseException(BaseException):
-    pass
+    def __init__(self, message, hint='Something went wrong.'):
+        super().__init__(message)
+        self.hint = hint
+
+
+class IndexGeneratorTemplateNotFound(IndexGeneratorBaseException):
+    def __init__(self, message):
+        return super().__init__(message, hint='Template folder does not exists or template file is not found.')