diff --git a/AKModel/models.py b/AKModel/models.py index c2383c33b74495eb5976e630b109d729a197841b..a612f49e9ac2f1f24876fc56f1260c3fa2c2de7b 100644 --- a/AKModel/models.py +++ b/AKModel/models.py @@ -1075,6 +1075,7 @@ class AKSlot(models.Model): "reso": self.ak.reso, "duration_in_hours": float(self.duration), "django_ak_id": self.ak.pk, + "types": list(self.ak.types.values_list("name", flat=True).order_by()), }, } diff --git a/AKModel/tests/test_json_export.py b/AKModel/tests/test_json_export.py index 8bfe554dddc56598bc34d576f8476cc570e49d8d..cbd3701333de5128ca54c2e5764e2b7722079b83 100644 --- a/AKModel/tests/test_json_export.py +++ b/AKModel/tests/test_json_export.py @@ -143,6 +143,7 @@ class JSONExportTest(TestCase): "reso", "duration_in_hours", "django_ak_id", + "types", }, f"{item} info keys not as expected", ) @@ -445,6 +446,10 @@ class JSONExportTest(TestCase): ) self.assertEqual(ak["info"]["description"], slot.ak.description) self.assertEqual(ak["info"]["django_ak_id"], slot.ak.pk) + self.assertEqual( + ak["info"]["types"], + list(slot.ak.types.values_list("name", flat=True).order_by()), + ) def test_ak_room_constraints(self): """Test if AK room constraints are exported as expected."""