From 89d5f3ae5fdce2ea059812e97650e9780d465d6d Mon Sep 17 00:00:00 2001 From: Jan Tuomi Date: Wed, 25 Jan 2017 17:59:02 +0200 Subject: [PATCH] Fix issue where HSL data would fill up the DB --- infoscreen/hsl_fetcher.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/infoscreen/hsl_fetcher.py b/infoscreen/hsl_fetcher.py index c85e239..9acd5c4 100644 --- a/infoscreen/hsl_fetcher.py +++ b/infoscreen/hsl_fetcher.py @@ -35,7 +35,14 @@ class HSLFetcher: arr.append({"name": parsed['name_fi'], "lines": parsed['lines'], "dist": element['dist'], "departures": parsed['departures']}) - HSLDataModel.objects.create(data=json.dumps(arr)) + model_arr = HSLDataModel.objects.all() + count = len(model_arr) + json_dump = json.dumps(arr) + + if count == 0: + HSLDataModel.objects.create(data=json_dump) + else: + model_arr[count - 1].data = json_dump now = datetime.now() HSLFetcher.last_fetched = now