From 542523c1de47be853363d465e07a69ad2f87e4fa Mon Sep 17 00:00:00 2001
From: Bastian Venthur <venthur@debian.org>
Date: Sat, 6 Feb 2021 14:07:03 +0100
Subject: [PATCH] added templatedir argument

---
 blag/blag.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/blag/blag.py b/blag/blag.py
index 78b4a2e..0e9686e 100644
--- a/blag/blag.py
+++ b/blag/blag.py
@@ -60,7 +60,11 @@ def parse_args(args):
             default='build',
             help='Ouptut directory (default: build)',
     )
-
+    build_parser.add_argument(
+            '-t', '--template-dir',
+            default='templates',
+            help='Template directory (default: templates)',
+    )
     return parser.parse_args()
 
 
@@ -85,14 +89,18 @@ def build(args):
             path = os.path.relpath(f'{root}/{dirname}', start=args.input_dir)
             os.makedirs(f'{args.output_dir}/{path}', exist_ok=True)
 
-    convert_to_html(convertibles, args.input_dir, args.output_dir)
+    convert_to_html(
+        convertibles, args.input_dir,
+        args.output_dir,
+        args.template_dir
+    )
 
 
-def convert_to_html(convertibles, input_dir, output_dir):
+def convert_to_html(convertibles, input_dir, output_dir, template_dir):
 
     env = Environment(
             loader=ChoiceLoader([
-                FileSystemLoader(['templates']),
+                FileSystemLoader([template_dir]),
                 PackageLoader('blag', 'templates'),
             ])
     )