X-Git-Url: https://git.madduck.net/etc/vim.git/blobdiff_plain/dbe26161fa68632d608a440666a0960a32630902..7b11f04d54d6da117ed5e640fd9420727f07ef81:/docs/conf.py

diff --git a/docs/conf.py b/docs/conf.py
index aeb57f9..ce7536d 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -12,26 +12,18 @@
 # 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 ast
 from pathlib import Path
-import re
+import regex as re
 import shutil
 import string
 
+from setuptools_scm import get_version
 from recommonmark.parser import CommonMarkParser
 
 
 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"
@@ -92,7 +84,7 @@ author = "Łukasz Langa and contributors to Black"
 
 # Autopopulate version
 # The full version, including alpha/beta/rc tags.
-release = get_version()
+release = get_version(root=CURRENT_DIR.parent)
 # The short X.Y version.
 version = release
 for sp in "abcfr":
@@ -161,12 +153,13 @@ html_theme_options = {
     "show_related": False,
     "description": "“Any color you like.”",
     "github_button": True,
-    "github_user": "ambv",
+    "github_user": "psf",
     "github_repo": "black",
     "github_type": "star",
     "show_powered_by": True,
     "fixed_sidebar": True,
     "logo": "logo2.png",
+    "travis_button": True,
 }