README preprocessor

Author: dahlia

setup.py

@@ -3,6 +3,7 @@
 import distutils.cmd
 import os
 import os.path
+import re
 import shutil
 import sys
 import tempfile
@@ -71,9 +72,23 @@ def spawn(self, cmd):
 def readme():
     try:
         with open(os.path.join(os.path.dirname(__file__), 'README.rst')) as f:
-            return f.read()
+            readme = f.read()
     except IOError:
         pass
+    return re.sub(
+        r'''
+        (?P<colon> : \n{2,})?
+        \.\. [ ] code-block:: \s+ [^\n]+ \n
+        [^ \t]* \n
+        (?P<block>
+            (?: (?: (?: \t | [ ]{3}) [^\n]* | [ \t]* ) \n)+
+        )
+        ''',
+        lambda m: (':' + m.group('colon') if m.group('colon') else '') +
+                  '\n'.join(' ' + l for l in m.group('block').splitlines()) +
+                  '\n\n',
+        readme, 0, re.VERBOSE
+    )
 
 
 class upload_doc(distutils.cmd.Command):