ensure_type: supports complex types

like TypeDict

Signed-off-by: Gaëtan Lehmann <[email protected]>

Author: glehmann

lib/common.py

@@ -1,3 +1,4 @@
+from contextlib import suppress
 import getpass
 import inspect
 import itertools
@@ -7,6 +8,7 @@
 import time
 import traceback
 from enum import Enum
+from pydantic import TypeAdapter, ValidationError
 from typing import Any, Dict, Literal, Optional, Type, TypeVar, overload, TYPE_CHECKING, Union
 from uuid import UUID
 
@@ -71,8 +73,21 @@ def callable_marker(value, request):
 T = TypeVar("T")
 
 def ensure_type(typ: Type[T], value: Any) -> T:
-    """Converts a value to the specified type. Also performs a runtime check."""
-    if not isinstance(value, typ):
+    """
+    Converts a value to the specified type.
+
+    Unlike typing.cast, it also performs a runtime check.
+    Unlike isinstance, it also supports complex types.
+    """
+    ok = False
+    try:
+        ok = isinstance(value, typ)
+    except TypeError:
+        # not just a simple type, lets try with pydantic
+        with suppress(ValidationError):
+            TypeAdapter(typ).validate_python(value)
+            ok = True
+    if not ok:
         raise TypeError(f"'{type(value).__name__}' object is not of the expected type '{typ.__name__}'")
     return value
 

requirements/base.txt

@@ -6,3 +6,4 @@ pluggy>=1.1.0
 requests
 legacycrypt
 pytest-dependency
+pydantic