Authorize to have a union for request parameters

mardiros · Jan 9, 2024 · c56c3b7 · c56c3b7
1 parent 2ece9f8
commit c56c3b7
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/blacksmith/domain/registry.py b/src/blacksmith/domain/registry.py
@@ -3,7 +3,7 @@
 
 from collections import defaultdict
 from dataclasses import dataclass
-from typing import Mapping, MutableMapping, Optional, Tuple, Type
+from typing import Any, Mapping, MutableMapping, Optional, Tuple, Type, TypeVar
 
 from blacksmith.typing import (
     ClientName,
@@ -18,8 +18,11 @@
 from .exceptions import ConfigurationError, UnregisteredClientException
 from .model import AbstractCollectionParser, Request, Response
 
-Schemas = Tuple[Type[Request], Optional[Type[Response]]]
-Contract = Mapping[HTTPMethod, Schemas]
+
+TRequest = TypeVar("TRequest", bound=Request)
+
+Schemas = Tuple[TRequest, Optional[Type[Response]]]
+Contract = Mapping[HTTPMethod, Schemas[Any]]
 
 
 @dataclass(frozen=True)