Fix issue where HSL data would fill up the DB

2017-01-25 17:59:02 +02:00
parent a063337f0a
commit 89d5f3ae5f
1 changed files with 8 additions and 1 deletions
@@ -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