First stab at the Vim plugin!

[etc/vim.git] / docs / conf.py

diff --git a/docs/conf.py b/docs/conf.py
index 5d0e9e3c6d183c10f416e4085a4ea775ac82f8a9..6edddba729f753803168911347ee22b68c5cedb4 100644 (file)
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -15,6 +15,7 @@
 import ast
 from pathlib import Path
 import re
+import shutil
 import string
 
 from recommonmark.parser import CommonMarkParser
@@ -40,6 +41,48 @@ def make_pypi_svg(version):
         f.write(svg)
 
 
+def make_filename(line):
+    non_letters = re.compile(r'[^a-z]+')
+    filename = line[3:].rstrip().lower()
+    filename = non_letters.sub('_', filename)
+    if filename.startswith('_'):
+        filename = filename[1:]
+    if filename.endswith('_'):
+        filename = filename[:-1]
+    return filename + '.md'
+
+
+def generate_sections_from_readme():
+    target_dir = CURRENT_DIR / '_build' / 'generated'
+    readme = CURRENT_DIR / '..' / 'README.md'
+    shutil.rmtree(str(target_dir), ignore_errors=True)
+    target_dir.mkdir(parents=True)
+
+    output = None
+    with open(str(readme), 'r', encoding='utf8') as f:
+        for line in f:
+            if line.startswith('## '):
+                if output is not None:
+                    output.close()
+                filename = make_filename(line)
+                output_path = CURRENT_DIR / filename
+                if output_path.is_symlink() or output_path.is_file():
+                    output_path.unlink()
+                output_path.symlink_to(target_dir / filename)
+                output = open(str(output_path), 'w', encoding='utf8')
+                output.write(
+                    '[//]: # (NOTE: THIS FILE IS AUTOGENERATED FROM README.md)\n\n'
+                )
+
+            if output is None:
+                continue
+
+            if line.startswith('##'):
+                line = line[1:]
+
+            output.write(line)
+
+
 # -- Project information -----------------------------------------------------
 
 project = 'Black'
@@ -54,6 +97,7 @@ version = release
 for sp in 'abcfr':
     version = version.split(sp)[0]
 make_pypi_svg(release)
+generate_sections_from_readme()
 
 
 # -- General configuration ---------------------------------------------------