chore: add types

.github/workflows/ci.yml

@@ -32,6 +32,8 @@ jobs:
         run: python -m pip install tox
       - name: Run linting
         run: python -m tox -e pep8
+      - name: Run mypy
+        run: python -m tox -e mypy
   packaging:
     name: Packaging
     runs-on: ubuntu-latest

MANIFEST.in

@@ -1,5 +1,6 @@
 include LICENSE README.rst CHANGES.rst
 include tox.ini .coveragerc pytest.ini
+include readme_renderer/py.typed
 
 recursive-include tests *.html
 recursive-include tests *.py

pyproject.toml

@@ -1,3 +1,12 @@
 [build-system]
 requires = ["setuptools>=40.8.0", "wheel", "bleach>=2.1.0", "docutils>=0.13.1", "Pygments>=2.5.1"]
 build-backend = "setuptools.build_meta:__legacy__"
+
+[tool.mypy]
+warn_unused_configs = true
+[[tool.mypy.overrides]]
+# These modules do not yet have types available.
+module = [
+    "cmarkgfm.*"
+]
+ignore_missing_imports = true

readme_renderer/clean.py

@@ -14,6 +14,7 @@
 from __future__ import absolute_import, division, print_function
 
 import functools
+import typing
 
 import bleach
 import bleach.callbacks
@@ -59,15 +60,12 @@
     "input": ["type", "checked", "disabled"],
 }
 
-ALLOWED_STYLES = [
-]
-
 
 class DisabledCheckboxInputsFilter:
-    def __init__(self, source):
+    def __init__(self, source: typing.Any) -> None:
         self.source = source
 
-    def __iter__(self):
+    def __iter__(self) -> typing.Iterator[typing.Dict[str, typing.Optional[str]]]:
         for token in self.source:
             if token.get("name") == "input":
                 # only allow disabled checkbox inputs
@@ -85,23 +83,24 @@ def __iter__(self):
             else:
                 yield token
 
-    def __getattr__(self, name):
+    def __getattr__(self, name: str) -> typing.Any:
         return getattr(self.source, name)
 
 
-def clean(html, tags=None, attributes=None, styles=None):
+def clean(
+    html: str,
+    tags: typing.Optional[typing.List[str]] = None,
+    attributes: typing.Optional[typing.Dict[str, typing.List[str]]] = None
+) -> typing.Optional[str]:
     if tags is None:
         tags = ALLOWED_TAGS
     if attributes is None:
         attributes = ALLOWED_ATTRIBUTES
-    if styles is None:
-        styles = ALLOWED_STYLES
 
     # Clean the output using Bleach
     cleaner = bleach.sanitizer.Cleaner(
         tags=tags,
         attributes=attributes,
-        styles=styles,
         filters=[
             # Bleach Linkify makes it easy to modify links, however, we will
             # not be using it to create additional links.

readme_renderer/integration/distutils.py

@@ -11,6 +11,13 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+# The `distutils` integration is going to need to get updated to `setuptools``
+# soon enough, as `distutils` is deprecated and will be removed in Python 3.12.
+# There's currently some pass-through imports that allow this to work, but is
+# challenging for `mypy` in `strict` mode, so let's skip this file for now.
+# See https://peps.python.org/pep-0632/
+# mypy: ignore-errors
 from __future__ import absolute_import, division, print_function
 
 import cgi

readme_renderer/markdown.py

@@ -15,8 +15,9 @@
 
 import re
 import warnings
+from typing import Any, Match, Optional
 
-from html.parser import unescape
+from html import unescape
 
 import pygments
 import pygments.lexers
@@ -51,7 +52,7 @@
 }
 
 
-def render(raw, variant="GFM", **kwargs):
+def render(raw: str, variant: str = "GFM", **kwargs: Any) -> Optional[str]:
     if not variants:
         warnings.warn(_EXTRA_WARNING)
         return None
@@ -61,7 +62,8 @@ def render(raw, variant="GFM", **kwargs):
     if not renderer:
         return None
 
-    rendered = renderer(raw)
+    # The renderer is a lambda function, and mypy fails lambdas right now.
+    rendered = renderer(raw)  # type: ignore
 
     if not rendered:
         return None
@@ -71,7 +73,7 @@ def render(raw, variant="GFM", **kwargs):
     return cleaned
 
 
-def _highlight(html):
+def _highlight(html: str) -> str:
     """Syntax-highlights HTML-rendered Markdown.
 
     Plucks sections to highlight that conform the the GitHub fenced code info
@@ -94,7 +96,7 @@ def _highlight(html):
         '(?(in_code)|<code>)(?P<code>.+?)'
         r'</code></pre>', re.DOTALL)
 
-    def replacer(match):
+    def replacer(match: Match[Any]) -> str:
         try:
             lang = match.group('lang')
             lang = _LANG_ALIASES.get(lang, lang)

readme_renderer/rst.py

@@ -14,20 +14,28 @@
 from __future__ import absolute_import, division, print_function
 
 import io
+import typing
 
 from docutils.core import publish_parts
+from docutils.nodes import colspec, image
 from docutils.writers.html4css1 import HTMLTranslator, Writer
 from docutils.utils import SystemMessage
 
 from .clean import clean
 
 
-class ReadMeHTMLTranslator(HTMLTranslator):
+class ReadMeHTMLTranslator(HTMLTranslator):  # type: ignore
 
     # Overrides base class not to output `<object>` tag for SVG images.
-    object_image_types = {}
-
-    def emptytag(self, node, tagname, suffix="\n", **attributes):
+    object_image_types = {}  # type: ignore
+
+    def emptytag(
+        self,
+        node: typing.Union[colspec, image],
+        tagname: str,
+        suffix: str = "\n",
+        **attributes: typing.Any
+    ) -> typing.Any:
         """Override this to add back the width/height attributes."""
         if tagname == "img":
             if "width" in node:
@@ -95,7 +103,11 @@ def emptytag(self, node, tagname, suffix="\n", **attributes):
 }
 
 
-def render(raw, stream=None, **kwargs):
+def render(
+    raw: str,
+    stream: typing.Optional[typing.Any] = None,
+    **kwargs: typing.Any
+) -> typing.Optional[str]:
     if stream is None:
         # Use a io.StringIO as the warning stream to prevent warnings from
         # being printed to sys.stderr.

readme_renderer/txt.py

@@ -14,6 +14,7 @@
 from __future__ import absolute_import, division, print_function
 
 import sys
+import typing
 
 from .clean import clean
 
@@ -26,6 +27,6 @@ def html_escape(s):
         return escape(s, quote=True).replace("'", ''')
 
 
-def render(raw, **kwargs):
+def render(raw: str, **kwargs: typing.Any) -> typing.Optional[str]:
     rendered = html_escape(raw).replace("\n", "<br>")
     return clean(rendered, tags=["br"])

tox.ini

@@ -1,13 +1,22 @@
 [tox]
-envlist = py36,py37,py38,py39,py310,pep8,packaging,noextra
+envlist = py36,py37,py38,py39,py310,pep8,packaging,noextra,mypy
 
 [testenv]
 deps =
     pytest
 commands =
-    pytest --strict {posargs}
+    pytest --strict-markers {posargs}
 extras = md
 
+[testenv:mypy]
+basepython = python3
+deps =
+    mypy
+    types-bleach
+    types-docutils
+    types-Pygments
+commands = mypy --strict readme_renderer
+
 [testenv:pep8]
 basepython = python3
 deps =