[docs] Fix docs language switcher for the version (backport #5679) (#…

…5684) [docs] Fix docs language switcher for the version (#5679) * [docs] Fix docs language switcher for the version * Update conf.py (cherry picked from commit f8e1ef6) Co-authored-by: Hailin Wang <[email protected]>
haiiliin · Jul 14, 2024 · 07cb9f1 · 07cb9f1
1 parent f0413f8
commit 07cb9f1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -24,6 +24,7 @@
 import sys
 import typing
 
+from packaging.version import Version
 from sphinx.builders.html import StandaloneHTMLBuilder
 
 # Environment variables
@@ -36,8 +37,9 @@
 author = "WANG Hailin"
 
 release = abqpy.__version__
-major, minor, patch, *_ = release.split(".")
-release = version = major
+rel = Version(release)
+version, major = rel.base_version, rel.major
+branch = major if not rel.is_prerelease else "latest"
 
 sys.path.insert(0, os.path.abspath("../../src"))
 sys.path.insert(0, os.path.abspath("./_ext"))
@@ -278,7 +280,7 @@
     ],
     # END: version_dropdown
     "languages": [
-        {"name": alias, "link": f"/{lang}/{version}" if READTHEDOCS else f"/abqpy/{lang}/{version}", "lang": lang}
+        {"name": alias, "link": f"/{lang}/{branch}" if READTHEDOCS else f"/abqpy/{lang}/{branch}", "lang": lang}
         for lang, alias in zip(("en", "zh_CN"), ("English", "简体中文"))
     ],
     "toc_title_is_page_title": True,

diff --git a/pyproject.toml b/pyproject.toml
@@ -58,6 +58,7 @@ docs = [
     "gitpython",
     "myst-parser",
     "pandas",
+    "packaging",
     "pillow",
     "sphinx",
     "sphinx-autoapi",