freyjagp/blag
1
0
Fork 0
forked from github.com/blag
Code Actions Packages Releases Activity

Generate needed dirs and files if not present, added Logger.

Browse Source
This commit is contained in:
Bastian Venthur
2010-08-29 12:02:00 +02:00
parent 20c9ad2ecf
commit f2ad8de6ae
1 changed files with 27 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
sg.py
View File

@@ -22,6 +22,7 @@ import shutil
import string import string
import codecs import codecs
import re import re
import logging
import markdown import markdown
@@ -30,15 +31,35 @@ LAYOUTS_DIR = '_layouts'
RESULT_DIR = '_site' RESULT_DIR = '_site'
STATIC_DIR = '_static' STATIC_DIR = '_static'
DEFAULT_LAYOUT = os.path.sep.join([LAYOUTS_DIR, 'default.html'])
DEFAULT_LAYOUT_HTML = """
<html>
<head></head>
<body>$content</body>
</html>
"""
def prepare_site(): def prepare_site():
"""Prepare site generation.""" """Prepare site generation."""
# check if all needed files are available logging.info("Checking if all needed dirs and files are available.")
# check if all needed dirs and files are available
for directory in LAYOUTS_DIR, STATIC_DIR:
if not os.path.exists(directory):
logging.warning("Directory '%s' does not exist, creating it." % directory)
os.mkdir(directory)
if not os.path.exists(DEFAULT_LAYOUT):
logging.warning("File '%s' does not exist, creating it." % DEFAULT_LAYOUT)
filehandle = open(DEFAULT_LAYOUT, 'w')
filehandle.write(DEFAULT_LAYOUT_HTML)
filehandle.close()
# clean RESULT_DIR # clean RESULT_DIR
shutil.rmtree(os.path.sep.join([os.curdir, RESULT_DIR]), True) shutil.rmtree(os.path.sep.join([os.curdir, RESULT_DIR]), True)
def generate_site(): def generate_site():
"""Generate the dynamic part of the site.""" """Generate the dynamic part of the site."""
logging.info("Generating Site.")
for root, dirs, files in os.walk(os.curdir): for root, dirs, files in os.walk(os.curdir):
# ignore directories starting with _ # ignore directories starting with _
if root.startswith(os.path.sep.join([os.curdir, '_'])): if root.startswith(os.path.sep.join([os.curdir, '_'])):
@@ -53,6 +74,7 @@ def generate_site():
def copy_static_content(): def copy_static_content():
"""Copy the static content to RESULT_DIR.""" """Copy the static content to RESULT_DIR."""
logging.info("Copying static content.")
shutil.copytree(os.path.sep.join([os.curdir, STATIC_DIR]), shutil.copytree(os.path.sep.join([os.curdir, STATIC_DIR]),
os.path.sep.join([os.curdir, RESULT_DIR])) os.path.sep.join([os.curdir, RESULT_DIR]))
@@ -125,7 +147,7 @@ def check_unused_variables(txt):
template = '\\$[_a-z][_a-z0=9]*' template = '\\$[_a-z][_a-z0=9]*'
f = re.findall(template, txt) f = re.findall(template, txt)
if len(f) > 0: if len(f) > 0:
print "Unconsumed variables in template found:", f logging.warning("Unconsumed variables in template found: %s" % f)
def render_page(path): def render_page(path):
@@ -134,12 +156,12 @@ def render_page(path):
It starts with the file under path, and processes it by pushing it through It starts with the file under path, and processes it by pushing it through
the processing pipeline. It returns a string. the processing pipeline. It returns a string.
""" """
print "Rendering", path logging.debug("Rendering %s" % path)
fh = codecs.open(path, 'r', 'utf-8') fh = codecs.open(path, 'r', 'utf-8')
txt = "".join(fh.readlines()) txt = "".join(fh.readlines())
fh.close() fh.close()
fh = codecs.open(os.path.sep.join([LAYOUTS_DIR, 'default.html']), 'r', 'utf-8') fh = codecs.open(DEFAULT_LAYOUT, 'r', 'utf-8')
template = ''.join(fh.readlines()) template = ''.join(fh.readlines())
fh.close() fh.close()
@@ -155,6 +177,7 @@ def render_page(path):
if __name__ == "__main__": if __name__ == "__main__":
logging.basicConfig(level=logging.DEBUG, format="%(levelname)s\t%(message)s")
prepare_site() prepare_site()
copy_static_content() copy_static_content()
generate_site() generate_site()