diff options
author | Zhiming Wang <zmwangx@gmail.com> | 2015-05-04 18:45:17 -0700 |
---|---|---|
committer | Zhiming Wang <zmwangx@gmail.com> | 2015-05-04 18:45:17 -0700 |
commit | 07bf43a314fe65ccd9c7cb663c3c6134a47cc269 (patch) | |
tree | 93104d3f07143b1e62d773d95dd973888cdd7f27 /pyblog | |
parent | ee374553f2ba385157eec9a816cf9b023fbfb18a (diff) | |
download | my_new_personal_website-07bf43a314fe65ccd9c7cb663c3c6134a47cc269.tar.xz my_new_personal_website-07bf43a314fe65ccd9c7cb663c3c6134a47cc269.zip |
edit posts and (mostly) figured out the theme
Also wrote pyblog that currently can generate parts most of the blog.
Diffstat (limited to '')
-rwxr-xr-x | pyblog | 145 |
1 files changed, 145 insertions, 0 deletions
@@ -0,0 +1,145 @@ +#!/usr/bin/env python3 + +"""A simple blog generator with Pandoc as backend.""" + +import argparse +import os +import re +import shutil +import subprocess +import sys + + +ROOTDIR = os.path.dirname(os.path.realpath(__file__)) +SOURCEDIR = os.path.join(ROOTDIR, "source") +INDEX = os.path.join(SOURCEDIR, "index.md") +TEMPLATEDIR = os.path.join(ROOTDIR, "templates") +HTMLTEMPLATE = os.path.join(TEMPLATEDIR, "template.html") +BUILDDIR = os.path.join(ROOTDIR, "build") + + +# TODO: +def new_post(): + pass + + +# TODO: +def generate_index(): + pass + + +def generate(fresh=False): + """Generate the blog in BUILDDIR. + + Parameters + ---------- + fresh : bool + If True, remove all existing build artifects and start afresh; + otherwise, only copy or build new or modified files. Default is + False. + + Returns + ------- + failed_builds : int + Number of build failures. + + """ + + # pylint: disable=too-many-branches + + if not os.path.isdir(SOURCEDIR): + raise OSError("source directory %s does not exist" % SOURCEDIR) + if not os.path.exists(HTMLTEMPLATE): + raise OSError("HTML template %s not found" % HTMLTEMPLATE) + + if not os.path.isdir(BUILDDIR): + if os.path.exists(BUILDDIR): + os.remove(BUILDDIR) + os.mkdir(BUILDDIR, mode=0o755) + if fresh: + for name in os.listdir(BUILDDIR): + if name == ".git": + continue + obj = os.path.join(BUILDDIR, name) + if os.path.isdir(obj): + shutil.rmtree(obj) + else: + os.remove(obj) + + failed_builds = 0 + + for root, _, files in os.walk(SOURCEDIR): + relroot = os.path.relpath(root, start=SOURCEDIR) + dstroot = os.path.join(BUILDDIR, relroot) + if not os.path.isdir(dstroot): + if os.path.exists(dstroot): + os.remove(dstroot) + os.mkdir(dstroot, mode=0o755) + + for name in files: + extension = name.split(".")[-1] + if extension not in ["css", "md"]: + continue + + relpath = os.path.join(relroot, name) + srcpath = os.path.join(root, name) + if extension == "md": + dstpath = os.path.join(dstroot, re.sub(r'\.md$', '.html', name)) + else: + dstpath = os.path.join(dstroot, name) + if ((not os.path.exists(dstpath) or + os.path.getmtime(dstpath) <= os.path.getmtime(srcpath))): + if extension == "css": + sys.stderr.write("copying %s\n" % relpath) + shutil.copy(srcpath, dstpath) + elif extension == "md": + sys.stderr.write("generating %s\n" % relpath) + pandoc_args = [ + "pandoc", srcpath, + "--template", HTMLTEMPLATE, + "--highlight-style=pygments", + "-o", dstpath, + ] + try: + subprocess.check_call(pandoc_args) + except subprocess.CalledProcessError: + failed_builds += 1 + sys.stderr.write("error: failed to generate %s" % + relpath) + + sys.stderr.write("build finished with %d errors\n" % failed_builds) + return failed_builds + + +# TODO: +def deploy(): + pass + + +# TODO: regenerate and deploy +def gen_deploy(): + pass + + +# TODO: start HTTP server in another process and watch for changes +def preview(): + pass + + +def main(): + """CLI interface.""" + description = "Simple blog generator in Python with Pandoc as backend." + parser = argparse.ArgumentParser(description=description) + parser.add_argument('action', choices=[ + 'generate', 'regenerate', + ]) + args = parser.parse_args() + + if args.action == 'generate': + exit(generate(fresh=False)) + elif args.action == 'regenerate': + exit(generate(fresh=True)) + + +if __name__ == '__main__': + main() |