django-rest-framework/mkdocs.py

76 lines
2.9 KiB
Python
Raw Normal View History

2012-09-01 23:26:27 +04:00
#!/usr/bin/env python
import markdown
import os
import re
2012-09-02 00:23:50 +04:00
import shutil
import sys
2012-09-01 23:26:27 +04:00
root_dir = os.path.abspath(os.path.dirname(__file__))
2012-09-02 00:23:50 +04:00
docs_dir = os.path.join(root_dir, 'docs')
html_dir = os.path.join(root_dir, 'html')
local = not '--deploy' in sys.argv
2012-09-01 23:26:27 +04:00
if local:
2012-09-02 00:23:50 +04:00
base_url = 'file://%s/' % os.path.normpath(os.path.join(os.getcwd(), html_dir))
2012-09-01 23:26:27 +04:00
suffix = '.html'
index = 'index.html'
else:
2012-09-02 00:26:03 +04:00
base_url = 'http://tomchristie.github.com/django-rest-framework'
2012-09-01 23:26:27 +04:00
suffix = ''
index = ''
main_header = '<li class="main"><a href="#{{ anchor }}">{{ title }}</a></li>'
sub_header = '<li><a href="#{{ anchor }}">{{ title }}</a></li>'
code_label = r'<a class="github" href="https://github.com/tomchristie/django-rest-framework/blob/restframework2/rest_framework/\1"><span class="label label-info">\1</span></a>'
2012-09-01 23:26:27 +04:00
2012-09-02 00:23:50 +04:00
page = open(os.path.join(docs_dir, 'template.html'), 'r').read()
# Copy static files
for static in ['css', 'js', 'img']:
2012-09-02 00:23:50 +04:00
source = os.path.join(docs_dir, 'static', static)
target = os.path.join(html_dir, static)
if os.path.exists(target):
shutil.rmtree(target)
shutil.copytree(source, target)
2012-09-01 23:26:27 +04:00
2012-09-02 00:23:50 +04:00
for (dirpath, dirnames, filenames) in os.walk(docs_dir):
2012-09-01 23:26:27 +04:00
for filename in filenames:
if not filename.endswith('.md'):
continue
toc = ''
text = open(os.path.join(dirpath, filename), 'r').read().decode('utf-8')
for line in text.splitlines():
if line.startswith('# '):
title = line[2:].strip()
template = main_header
elif line.startswith('## '):
title = line[3:].strip()
template = sub_header
else:
continue
anchor = title.lower().replace(' ', '-').replace(':-', '-').replace("'", '').replace('?', '').replace('.', '')
template = template.replace('{{ title }}', title)
template = template.replace('{{ anchor }}', anchor)
toc += template + '\n'
content = markdown.markdown(text, ['headerid'])
2012-09-27 16:27:40 +04:00
relative_dir = dirpath.replace(docs_dir, '').lstrip(os.path.sep)
build_dir = os.path.join(html_dir, relative_dir)
2012-09-01 23:26:27 +04:00
build_file = os.path.join(build_dir, filename[:-3] + '.html')
if not os.path.exists(build_dir):
os.makedirs(build_dir)
output = page.replace('{{ content }}', content).replace('{{ toc }}', toc).replace('{{ base_url }}', base_url).replace('{{ suffix }}', suffix).replace('{{ index }}', index)
output = output.replace('{{ page_id }}', filename[:-3])
2012-09-05 21:25:30 +04:00
output = re.sub(r'a href="([^"]*)\.md"', r'a href="\1%s"' % suffix, output)
2012-09-08 23:23:32 +04:00
output = re.sub(r'<pre><code>:::bash', r'<pre class="prettyprint lang-bsh">', output)
output = re.sub(r'<pre>', r'<pre class="prettyprint lang-py">', output)
2012-09-09 01:06:13 +04:00
output = re.sub(r'<a class="github" href="([^"]*)"></a>', code_label, output)
2012-09-01 23:26:27 +04:00
open(build_file, 'w').write(output.encode('utf-8'))