From 810d3c8e5c4b8e0493a092accb452ad83d4e3cb8 Mon Sep 17 00:00:00 2001 From: puckoprutt Date: Tue, 25 Feb 2025 16:21:48 +0100 Subject: [PATCH] fixed bug if json is list and solved empty heat response. --- spiders/dagens_lunch/views.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/spiders/dagens_lunch/views.py b/spiders/dagens_lunch/views.py index 3c7706c..4f7601e 100644 --- a/spiders/dagens_lunch/views.py +++ b/spiders/dagens_lunch/views.py @@ -19,6 +19,8 @@ class Lunch_Valsaren_View(PuckoView): file = DAGENS_LUNCH_PATH / "valsaren.json" with open(file, "r") as f: data = json.load(f) + if isinstance(data, list): + data = data[0] serializer = self.serializer_class(data) return Response(serializer.data, status=200) @@ -30,6 +32,8 @@ class Lunch_Malmens_View(PuckoView): file = DAGENS_LUNCH_PATH / "malmens.json" with open(file, "r") as f: data = json.load(f) + if isinstance(data, list): + data = data[0] serializer = self.serializer_class(data) return Response(serializer.data, status=200) @@ -41,6 +45,8 @@ class Lunch_Varda_View(PuckoView): file = DAGENS_LUNCH_PATH / "varda.json" with open(file, "r") as f: data = json.load(f) + if isinstance(data, list): + data = data[0] serializer = self.serializer_class({ "place": data["place"], "about": data["about"], @@ -55,7 +61,9 @@ class Lunch_Varda_View(PuckoView): "pasta": data["pasta"], "fisken": data["fish"], "kaninen": data["green"], - "sallad": data["salad"] + "sallad": data["salad"], + "scraped_by": data["scraped_by"], + "scraped_at": data["scraped_at"] }) return Response(serializer.data, status=200) @@ -67,7 +75,9 @@ class Lunch_Heat_View(PuckoView): file = DAGENS_LUNCH_PATH / "heat.json" with open(file, "r") as f: data = json.load(f) - serializer = self.serializer_class() + if isinstance(data, list): + data = data[0] + serializer = self.serializer_class(data) return Response(serializer.data, status=200) class Lunch_Max_View(PuckoView): @@ -78,5 +88,7 @@ class Lunch_Max_View(PuckoView): file = DAGENS_LUNCH_PATH / "max.json" with open(file, "r") as f: data = json.load(f) + if isinstance(data, list): + data = data[0] serializer = self.serializer_class(data) return Response(serializer.data, status=200) \ No newline at end of file