Fix pydantic deserialization issues with root obj (#854)

Jason Munro · web-flow · commit 47fc6087da82 · 2023-10-10T18:08:50.000-07:00
* Fix pydantic deserialization issues with root obj

* Linting
diff --git a/mp_api/client/core/client.py b/mp_api/client/core/client.py
@@ -842,7 +842,7 @@ def _submit_request_and_process(
                         data_model(
                             **{
                                 field: value
-                                for field, value in raw_doc.model_dump().items()
+                                for field, value in dict(raw_doc).items()
                                 if field in set_fields
                             }
                         )
@@ -876,9 +876,8 @@ def _submit_request_and_process(
             )
 
     def _generate_returned_model(self, doc):
-        set_fields = [
-            field for field, _ in doc if field in doc.model_dump(exclude_unset=True)
-        ]
+        set_fields = doc.model_fields_set
+
         unset_fields = [field for field in doc.model_fields if field not in set_fields]
 
         data_model = create_model(

Original file line number	Diff line number	Diff line change
`@@ -842,7 +842,7 @@ def _submit_request_and_process(`
`842`	`842`	`data_model(`
`843`	`843`	`**{`
`844`	`844`	`field: value`
`845`		`- for field, value in raw_doc.model_dump().items()`
	`845`	`+ for field, value in dict(raw_doc).items()`
`846`	`846`	`if field in set_fields`
`847`	`847`	`}`
`848`	`848`	`)`
`@@ -876,9 +876,8 @@ def _submit_request_and_process(`
`876`	`876`	`)`
`877`	`877`
`878`	`878`	`def _generate_returned_model(self, doc):`
`879`		`- set_fields = [`
`880`		`- field for field, _ in doc if field in doc.model_dump(exclude_unset=True)`
`881`		`- ]`
	`879`	`+ set_fields = doc.model_fields_set`
	`880`	`+`
`882`	`881`	`unset_fields = [field for field in doc.model_fields if field not in set_fields]`
`883`	`882`
`884`	`883`	`data_model = create_model(`