feat: add types for PEP 735 (#203)

Signed-off-by: Henry Schreiner <[email protected]>
pypa · Oct 11, 2024 · 70c447e · 70c447e
1 parent 5f49b28
commit 70c447e
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/pyproject_metadata/project_table.py b/pyproject_metadata/project_table.py
@@ -11,6 +11,7 @@
 from __future__ import annotations
 
 import sys
+import typing
 from typing import Any, Dict, List, Union
 
 if sys.version_info < (3, 11):
@@ -28,6 +29,7 @@
     "BuildSystemTable",
     "ContactTable",
     "Dynamic",
+    "IncludeGroupTable",
     "LicenseTable",
     "ProjectTable",
     "PyProjectTable",
@@ -107,12 +109,44 @@ class LicenseTable(TypedDict, total=False):
     total=False,
 )
 
+# total=False here because this could be
+# extended in the future
+IncludeGroupTable = TypedDict(
+    "IncludeGroupTable",
+    {"include-group": str},
+    total=False,
+)
+
 PyProjectTable = TypedDict(
     "PyProjectTable",
     {
         "build-system": BuildSystemTable,
         "project": ProjectTable,
         "tool": Dict[str, Any],
+        "dependency-groups": Dict[str, List[Union[str, IncludeGroupTable]]],
     },
     total=False,
 )
+
+# Tests for type checking
+if typing.TYPE_CHECKING:
+    PyProjectTable(
+        {
+            "build-system": BuildSystemTable(
+                {"build-backend": "one", "requires": ["two"]}
+            ),
+            "project": ProjectTable(
+                {
+                    "name": "one",
+                    "version": "0.1.0",
+                }
+            ),
+            "tool": {"thing": object()},
+            "dependency-groups": {
+                "one": [
+                    "one",
+                    IncludeGroupTable({"include-group": "two"}),
+                ]
+            },
+        }
+    )