From da0e1a2885d4deaa07e3f69435210912634a23d4 Mon Sep 17 00:00:00 2001
From: Felix Blanke <info@fblanke.de>
Date: Thu, 27 Feb 2025 20:28:15 +0100
Subject: [PATCH] Add AK types to json export
---
AKModel/models.py | 1 +
AKModel/tests/test_json_export.py | 5 +++++
2 files changed, 6 insertions(+)
diff --git a/AKModel/models.py b/AKModel/models.py
index c2383c33..a612f49e 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 8bfe554d..cbd37013 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."""
--
GitLab