Code highlighting in docs

1 files changed, 2 insertions, 0 deletions

diff --git a/mkdocs.py b/mkdocs.py
index 63e15d8f..0acb14d1 100755
--- a/mkdocs.py
+++ b/mkdocs.py
@@ -66,4 +66,6 @@ for (dirpath, dirnames, filenames) in os.walk(docs_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 = re.sub(r'a href="([^"]*)\.md"', r'a href="\1%s"' % suffix, output)
+        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)
         open(build_file, 'w').write(output.encode('utf-8'))