strike: merge as subdir

1 files changed, 70 insertions, 0 deletions

diff --git a/strike/strike.py b/strike/strike.py
new file mode 100644
index 0000000..03d991e
--- /dev/null
+++ b/strike/strike.py
@@ -0,0 +1,70 @@
+#!/usr/bin/env python3
+import os
+from time import gmtime, strftime
+from argparse import ArgumentParser
+from configparser import ConfigParser
+from pathlib import Path
+
+def handle_args():
+	parser = ArgumentParser()
+	parser.add_argument('config', help='location of strike.ini file (or a directory containing it)')
+	return parser.parse_args()
+
+def read_config(location):
+	location = Path(location)
+	config = ConfigParser()
+	if location.is_dir():
+		location = location/'strike.ini'
+	if not location.exists():
+		raise FileNotFoundError(f'Config file not found at {location}.')
+	config.read(location)
+	return location.resolve().parent, config
+
+def handle_file(file):
+	meta = dict(
+		title = file.stem,
+		date = strftime('%a, %d %b %Y %H:%M:%S GMT', gmtime(file.stat().st_mtime))
+	)
+	contents = file.read_text()
+	delim = '---\n'
+	if contents.startswith(delim):
+		parts = contents.split(delim, maxsplit=2)
+		meta.update(dict((key.strip(), value.strip())
+			for key, value in (line.split('=') for line in parts[1].splitlines())
+		))
+		contents = parts[2]
+	return contents, meta
+
+def apply_template(content, meta, template="{content}"):
+	return template.format(
+		content = content,
+		meta = meta
+	)
+
+def main():
+	args = handle_args()
+	basedir, config = read_config(args.config)
+
+	input_dir = basedir/config['Input']['directory']
+
+	try: exclusions = config['Input']['excludes'].splitlines()
+	except KeyError: exclusions = {}
+
+	default_template = basedir/config['Templates']['default']
+	template = default_template.read_text()
+
+	output_dir = basedir/config['Output']['directory']
+
+	for path, _, files in os.walk(input_dir):
+		loc = os.path.relpath(path, input_dir)
+		(output_dir/loc).mkdir(parents=True, exist_ok=True)
+		for file in files:
+			if file in exclusions:
+				continue
+			contents, meta = handle_file(input_dir/loc/file)
+			output = apply_template(contents, meta, template)
+			(output_dir/loc/file).write_text(output)
+
+if __name__ == '__main__':
+	import sys
+	sys.exit(main())