From: Christian Heller <c.heller@plomlompom.de>
Date: Sat, 4 Jul 2020 01:08:29 +0000 (+0200)
Subject: Fix.
X-Git-Url: https://plomlompom.com/repos/?p=berlin-corona-table;a=commitdiff_plain;h=ef830881ea19fe5686220b66f07591307a4948a8

Fix.
---

diff --git a/enhance_table.py b/enhance_table.py
index f8add4e..975abd8 100755
--- a/enhance_table.py
+++ b/enhance_table.py
@@ -239,10 +239,14 @@ HTML view: https://plomlompom.com/berlin_corona.html"""
         weekly_avgs = []
         weekly_incs = []
         for district in sorted_districts:
-            new_infections += [db[district][date]['new_infections']]
-            weekly_sums += [db[district][date]['week_sum']]
-            weekly_avgs += [db[district][date]['week_average']]
-            weekly_incs += [db[district][date]['week_incidence']]
+            district_day_data = db[district][date]
+            new_infections += [district_day_data['new_infections']]
+            if 'week_sum' in district_day_data:
+                weekly_sums += [district_day_data['week_sum']]
+            if 'week_average' in district_day_data:
+                weekly_avgs += [district_day_data['week_average']]
+            if 'week_incidence' in district_day_data:
+                weekly_incs += [district_day_data['week_incidence']]
         print('+', ' '*11, '  '.join(['%3s' % i for i in new_infections]))
         print('Σ', ' '*10, ' '.join(['%4s' % wsum for wsum in weekly_sums]))
         print('Ø', ' '*9, ''.join(['%5.1f' % wavg for wavg in weekly_avgs]))