Better approach

assertpy · Mar 19, 2021 · fe7a870 · fe7a870
1 parent 30853b4
commit fe7a870
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/assertpy/extracting.py b/assertpy/extracting.py
@@ -39,6 +39,13 @@
 __tracebackhide__ = True
 
 
+try:
+    from pydantic import BaseModel as PydanticBase
+except ImportError:
+    class PydanticBase:
+        pass  # implement necessary parts here
+
+
 class ExtractingMixin(object):
     """Collection flattening mixin.
 
@@ -177,7 +184,7 @@ def _extract(x, name):
                     return x[name]
                 else:
                     raise ValueError('item keys %s did not contain key <%s>' % (list(x.keys()), name))
-            elif isinstance(x, Iterable):
+            elif isinstance(x, Iterable) and not isinstance(x, PydanticBase):
                 self._check_iterable(x, name='item')
                 return x[name]
             elif hasattr(x, name):