forks/sigal
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

More configurable writer (add template_file and output_file).

Browse Source
This commit is contained in:
Simon Conseil
2013-04-16 23:25:37 +02:00
parent fff1e6e2b4
commit 920c706e8d
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
sigal/writer.py
View File

@@ -37,7 +37,6 @@ from .image import Image
from .settings import get_thumb
from .pkgmeta import __url__ as sigal_link
INDEX_PAGE = "index.html"
THEMES_PATH = os.path.normpath(os.path.join(
os.path.abspath(os.path.dirname(__file__)), 'themes'))
@@ -45,14 +44,17 @@ THEMES_PATH = os.path.normpath(os.path.join(
class Writer(object):
"""Generate html pages for each directory of images."""
def __init__(self, settings, output_dir, theme=None):
def __init__(self, settings, output_dir, theme=None,
template_file="index.html", output_file="index.html"):
self.settings = settings
self.output_dir = os.path.abspath(output_dir)
self.theme = theme or settings['theme']
self.template_file = template_file
self.output_file = output_file
self.logger = logging.getLogger(__name__)
# optionally add index.html to the URLs
self.url_ext = INDEX_PAGE if settings['index_in_url'] else ''
self.url_ext = self.output_file if settings['index_in_url'] else ''
# search the theme in sigal/theme if the given one does not exists
if not os.path.exists(self.theme):
@@ -75,7 +77,7 @@ class Writer(object):
)
try:
self.template = env.get_template(INDEX_PAGE)
self.template = env.get_template(self.template_file)
except TemplateNotFound:
self.logger.error(colored.red('ERROR:') +
' The index.html template was not found.')
@@ -164,6 +166,6 @@ class Writer(object):
ctx = self.generate_context(paths, relpath)
page = self.template.render(paths[relpath], **ctx)
path = os.path.join(self.output_dir, relpath)
with codecs.open(os.path.join(path, INDEX_PAGE), 'w', 'utf-8') as f:
output_file = os.path.join(self.output_dir, relpath, self.output_file)
with codecs.open(output_file, 'w', 'utf-8') as f:
f.write(page)