Add support for epub css files

2 files changed, 7 insertions, 1 deletions

diff --git a/panfry/document.py b/panfry/document.py
index 09a728d..5fdd05c 100755
--- a/panfry/document.py
+++ b/panfry/document.py
@@ -262,6 +262,10 @@ class Document(object):
         doc = pandoc.Document()
         doc.markdown = src
 
+        css = os.path.abspath(os.path.join(self.template_path, 'epub.css'))
+        if os.path.isfile(css):
+            print(css)
+            doc.add_argument('epub-stylesheet=%s' % css)
         print("epub_path: %s" % epub_path)
         pandoc.set_cwd(os.path.abspath(self.src_path))
         for option in self.pandoc_options:
diff --git a/panfry/templater.py b/panfry/templater.py
index 40fc17a..2cd9204 100755
--- a/panfry/templater.py
+++ b/panfry/templater.py
@@ -17,7 +17,9 @@ class Templater:
     def __init__(self, path):
         self.templates_path = path
         self.j2 = Environment(loader=FileSystemLoader(self.templates_path),
-                              line_statement_prefix='#')
+                              line_statement_prefix='#',
+                              extensions=['jinja2.ext.loopcontrols'],
+                              )
 
     def page_template(self, doc, page):
         '''