|
|
#!/usr/bin/env python3
import glob import markdown import os import re import shutil
input_directory = 'content' static_directory = 'static' output_directory = 'output'
md_extensions = ['fenced_code', 'codehilite', 'nl2br', 'toc', 'smarty', 'tables', 'linkify']
def print_file(in_file, out_file): print('%-62s -> %s' % (in_file, out_file))
template = open('template.html').read()
os.makedirs(output_directory, exist_ok=True)
for (dirpath, _, filenames) in os.walk(static_directory): for filename in filenames: source = os.path.join(dirpath, filename) out_path = dirpath.replace(static_directory, '', 1) out_path = out_path.lstrip('/') dest_dir = os.path.join(output_directory, out_path) os.makedirs(dest_dir, exist_ok=True) dest = os.path.join(dest_dir, filename) print_file(source, dest) shutil.copy2(source, dest)
out_filenames = []
for (dirpath, _, filenames) in os.walk(input_directory): for filename in filenames: markdown_filename = os.path.join(dirpath, filename) if not markdown_filename.endswith('.md'): continue
markdown_file = open(markdown_filename) text = markdown_file.read() markdown_file.close()
if not text.startswith('# '): text = '# ' + text
match = re.match(r'^(.*?)\n', text) if match: title = match.group(1).lstrip('# ') else: title = text title += ' | Colin McMillen' if markdown_filename == os.path.join(input_directory, 'index.md'): title = 'Colin McMillen'
out_filename = os.path.basename(markdown_filename).replace('.md', '.html')
html = markdown.markdown(text, extensions=md_extensions, output_format='html5') output = template.replace('__TITLE_GOES_HERE__', title) output = output.replace('__CONTENT_GOES_HERE__', html)
page_url = out_filename if page_url.endswith('index.html'): # strip off index.html page_url = page_url[:-len('index.html')] output = output.replace('__PAGE_URL_GOES_HERE__', page_url)
out_filenames.append(out_filename) out_dirpath = os.path.join(output_directory, dirpath) out_dirpath = out_dirpath.replace('/content', '', 1) os.makedirs(out_dirpath, exist_ok=True) out_fullpath = os.path.join(out_dirpath, out_filename) print_file(markdown_filename, out_fullpath) out_file = open(out_fullpath, 'w') out_file.write(output) out_file.close()
# TODO: make a sitemap / RSS? #index_filename = os.path.join(output_directory, 'index.html') #print_file('', index_filename) #index = open(index_filename, 'w') #for f in out_filenames: # index.write('<a href="%s">%s</a><br>' % (f, f)) #index.close()
|