Merge pull request #76 from yaal-coop/issue-75-pydantic-discriminator

pydantic discriminator issue introduced with pydantic 2.9.0

Author: azmeuk

doc/changelog.rst

@@ -1,6 +1,13 @@
 Changelog
 =========
 
+[0.2.2] - Unreleased
+--------------------
+
+Fixed
+^^^^^
+- :class:`~scim2_models.ListResponse` pydantic discriminator issue introduced with pydantic 2.9.0. #75
+
 [0.2.1] - 2024-09-06
 --------------------
 

scim2_models/rfc7644/list_response.py

@@ -39,14 +39,18 @@ def get_schema_from_payload(payload: Any) -> Optional[str]:
             if not payload:
                 return None
 
+            payload_schemas = (
+                payload.get("schemas", [])
+                if isinstance(payload, dict)
+                else getattr(payload, "schemas")
+            )
+
             resource_types_schemas = [
                 resource_type.model_fields["schemas"].default[0]
                 for resource_type in resource_types
             ]
             common_schemas = [
-                schema
-                for schema in payload.get("schemas", [])
-                if schema in resource_types_schemas
+                schema for schema in payload_schemas if schema in resource_types_schemas
             ]
             return common_schemas[0] if common_schemas else None