Don't fail serializing querysets of non-models

Author: akx

inertia/tests/test_encoder.py

@@ -7,17 +7,21 @@
 from inertia.utils import InertiaJsonEncoder
 
 
+def make_brandon() -> User:
+    return User(
+        name="Brandon",
+        password="something-top-secret",
+        birthdate=date(1987, 2, 15),
+        registered_at=datetime(2022, 10, 31, 10, 13, 1),
+    )
+
+
 class InertiaJsonEncoderTestCase(TestCase):
     def setUp(self):
         self.encode = lambda obj: dumps(obj, cls=InertiaJsonEncoder)
 
     def test_it_handles_models_with_dates_and_removes_passwords(self):
-        user = User(
-            name="Brandon",
-            password="something-top-secret",
-            birthdate=date(1987, 2, 15),
-            registered_at=datetime(2022, 10, 31, 10, 13, 1),
-        )
+        user = make_brandon()
 
         self.assertEqual(
             dumps(
@@ -51,12 +55,7 @@ def test_it_handles_inertia_meta_fields(self):
         )
 
     def test_it_handles_querysets(self):
-        User(
-            name="Brandon",
-            password="something-top-secret",
-            birthdate=date(1987, 2, 15),
-            registered_at=datetime(2022, 10, 31, 10, 13, 1),
-        ).save()
+        make_brandon().save()
 
         self.assertEqual(
             dumps(
@@ -71,3 +70,12 @@ def test_it_handles_querysets(self):
             ),
             self.encode(User.objects.all()),
         )
+
+
+    def test_it_handles_non_model_querysets(self):
+        user = make_brandon()
+        user.save()
+        self.assertEqual(
+            self.encode(User.objects.values_list("name", flat=True)),
+            self.encode([user.name]),
+        )

inertia/utils.py

@@ -24,7 +24,10 @@ def default(self, value):
             return model_to_dict(value)
 
         if isinstance(value, QuerySet):
-            return [model_to_dict(model) for model in value]
+            return [
+                (model_to_dict(obj) if isinstance(value.model, models.Model) else obj)
+                for obj in value
+            ]
 
         return super().default(value)