aboutsummaryrefslogtreecommitdiff
path: root/pyblog
diff options
context:
space:
mode:
authorZhiming Wang <zmwangx@gmail.com>2015-05-04 18:45:17 -0700
committerZhiming Wang <zmwangx@gmail.com>2015-05-04 18:45:17 -0700
commit07bf43a314fe65ccd9c7cb663c3c6134a47cc269 (patch)
tree93104d3f07143b1e62d773d95dd973888cdd7f27 /pyblog
parentee374553f2ba385157eec9a816cf9b023fbfb18a (diff)
downloadmy_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-xpyblog145
1 files changed, 145 insertions, 0 deletions
diff --git a/pyblog b/pyblog
new file mode 100755
index 00000000..df701ba1
--- /dev/null
+++ b/pyblog
@@ -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()