X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/9e3175428470403d01cdee9c710d5d38e8610860..5c6a0dd03f7f6d8694d1b9a62f9770be44f0b809:/docs/conf.py

diff --git a/docs/conf.py b/docs/conf.py
index 4a3c67c..5d0e9e3 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -12,13 +12,33 @@
 # add these directories to sys.path here. If the directory is relative to the
 # documentation root, use os.path.abspath to make it absolute, like shown here.
 #
-import os
-import sys
+import ast
+from pathlib import Path
+import re
+import string
 
 from recommonmark.parser import CommonMarkParser
 
 
-sys.path.insert(0, os.path.abspath('.' ))
+CURRENT_DIR = Path(__file__).parent
+
+
+def get_version():
+    black_py = CURRENT_DIR / '..' / 'black.py'
+    _version_re = re.compile(r'__version__\s+=\s+(?P<version>.*)')
+    with open(str(black_py), 'r', encoding='utf8') as f:
+        version = _version_re.search(f.read()).group('version')
+    return str(ast.literal_eval(version))
+
+
+def make_pypi_svg(version):
+    template = CURRENT_DIR / '_static' / 'pypi_template.svg'
+    target = CURRENT_DIR / '_static' / 'pypi.svg'
+    with open(str(template), 'r', encoding='utf8') as f:
+        svg = string.Template(f.read()).substitute(version=version)
+    with open(str(target), 'w', encoding='utf8') as f:
+        f.write(svg)
+
 
 # -- Project information -----------------------------------------------------
 
@@ -27,12 +47,14 @@ copyright = '2018, Łukasz Langa and contributors to Black'
 author = 'Łukasz Langa and contributors to Black'
 
 # Autopopulate version
-import black
-# The short X.Y version.
-# TODO: fix for 2 digit month
-version = f'{black.__version__[:4]}'
 # The full version, including alpha/beta/rc tags.
-release = black.__version__
+release = get_version()
+# The short X.Y version.
+version = release
+for sp in 'abcfr':
+    version = version.split(sp)[0]
+make_pypi_svg(release)
+
 
 # -- General configuration ---------------------------------------------------
 
@@ -96,7 +118,7 @@ html_sidebars = {
 }
 
 html_theme_options = {
-    'show_related': True,
+    'show_related': False,
     'description': '“Any color you like.”',
     'github_button': True,
     'github_user': 'ambv',