Added Type Hints

.pre-commit-config.yaml

@@ -13,3 +13,5 @@ repos:
       - id: flake8
         additional_dependencies:
         - flake8-comprehensions
+        - flake8-typing-imports
+        - flake8-type-checking

MANIFEST.in

@@ -5,3 +5,4 @@ include README.rst
 recursive-include crispy_forms/static *
 recursive-include crispy_forms/templates *
 recursive-include crispy_forms/tests *.py *.html
+include crispy_forms/py.typed

crispy_forms/base.py

@@ -1,3 +1,13 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Collection
+
+if TYPE_CHECKING:
+    from types import TracebackType
+
+    from django.template import Context, Node
+
+
 class KeepContext:
     """
     Context manager that receives a `django.template.Context` instance and a list of keys
@@ -9,14 +19,19 @@ class KeepContext:
     touch context object themselves, that could introduce side effects.
     """
 
-    def __init__(self, context, keys):
+    def __init__(self, context: Context, keys: Collection[int | str | Node]) -> None:
         self.context = context
         self.keys = keys
 
-    def __enter__(self):
+    def __enter__(self) -> None:
         pass
 
-    def __exit__(self, type, value, traceback):
+    def __exit__(
+        self,
+        type: type[BaseException] | None,
+        value: BaseException | None,
+        traceback: TracebackType | None,
+    ) -> None:
         for key in list(self.keys):
             if key in self.context:
                 del self.context[key]