from __future__ import annotations

from pathlib import Path

from sphinx.locale import get_translation

TYPE_CHECKING = False

if TYPE_CHECKING:

from typing import Any

from sphinx.application import Sphinx

from sphinx.util.typing import ExtensionMetadata

__version__ = "2026.4"

THEME_PATH = Path(__file__).resolve().parent

LOCALE_DIR = THEME_PATH / "locale"

MESSAGE_CATALOG_NAME = "python-docs-theme"

def add_translation_to_context(

app: Sphinx,

pagename: str,

templatename: str,

context: dict[str, Any],

doctree: None,

) -> None:

theme_gettext = get_translation(MESSAGE_CATALOG_NAME)

sphinx_gettext = get_translation("sphinx")

def combined(message: str) -> str:

translation = theme_gettext(message)

if translation == message:

return sphinx_gettext(message)

return translation

context["_"] = context["gettext"] = context["ngettext"] = combined

def setup(app: Sphinx) -> ExtensionMetadata:

app.require_sphinx("7.3")

app.add_html_theme("python_docs_theme", str(THEME_PATH))

app.add_message_catalog(MESSAGE_CATALOG_NAME, LOCALE_DIR)

app.connect("html-page-context", add_translation_to_context)

return {

"version": __version__,

"parallel_read_safe": True,

"parallel_write_safe": True,

}