Add type hints

Author: StanFromIreland

python_docs_theme/__init__.py

@@ -6,6 +6,8 @@
 
 TYPE_CHECKING = False
 if TYPE_CHECKING:
+    from typing import Any
+
     from sphinx.application import Sphinx
     from sphinx.util.typing import ExtensionMetadata
 
@@ -16,15 +18,21 @@
 MESSAGE_CATALOG_NAME = "python-docs-theme"
 
 
-def add_translation_to_context(app, pagename, templatename, context, doctree):
+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(msg):
-        trans = theme_gettext(msg)
-        if trans == msg:
-            return sphinx_gettext(msg)
-        return trans
+    def combined(message: str) -> str:
+        translation = theme_gettext(message)
+        if translation == message:
+            return sphinx_gettext(message)
+        return translation
 
     context["_"] = context["gettext"] = context["ngettext"] = combined