summaryrefslogtreecommitdiff
path: root/ftt-docgen
diff options
context:
space:
mode:
Diffstat (limited to 'ftt-docgen')
-rwxr-xr-xftt-docgen14
1 files changed, 12 insertions, 2 deletions
diff --git a/ftt-docgen b/ftt-docgen
index f1f23a7..f73bb27 100755
--- a/ftt-docgen
+++ b/ftt-docgen
@@ -1,6 +1,7 @@
#!/usr/bin/env python3
import copy
+import os
import re
import sys
@@ -228,14 +229,23 @@ def walk(o, func):
for c in o.children:
walk(c, func)
+def infer_bindings_filename(markdowns):
+ root, ext = os.path.splitext(markdowns[0])
+ if ext not in ['.md', '.mdwn']:
+ sys.exit('Input filenames must end in .md or .mdwn')
+ return root + '.yaml'
+
def debug(msg):
if False:
sys.stderr.write('DEBUG: {}\n'.format(msg))
sys.stderr.flush()
-bindings = yaml.safe_load(open(sys.argv[1]))
+markdowns = sys.argv[1:]
+text = ''.join(open(filename).read() for filename in markdowns)
+
+with open(infer_bindings_filename(markdowns)) as f:
+ bindings = yaml.safe_load(f)
-text = ''.join(open(filename).read() for filename in sys.argv[2:])
start = '---\n'
end = '\n...\n'