From: Christian Heller <c.heller@plomlompom.de>
Date: Sat, 4 Jul 2020 00:02:48 +0000 (+0200)
Subject: Move sum_district to the left, as it's of more interest than others.
X-Git-Url: https://plomlompom.com/repos/?a=commitdiff_plain;h=f91a74c7aa31a742e50629e4e57736cf5192320d;p=berlin-corona-table

Move sum_district to the left, as it's of more interest than others.
---

diff --git a/enhance_table.py b/enhance_table.py
index 2410099..4ce7792 100755
--- a/enhance_table.py
+++ b/enhance_table.py
@@ -94,6 +94,10 @@ for line in lines[1:]:
         db[district][date] = {'new_infections': int(district_data)}
 sorted_dates.sort()
 
+# Define and move sum_district from end to start.
+sum_district = sorted_districts.pop()
+sorted_districts.insert(0, sum_district)
+
 # In LaGeSo's data, the last "district" is actually the sum of all districts /
 # the whole of Berlin.
 #
@@ -103,9 +107,8 @@ sorted_dates.sort()
 # the table file, since we should have a human look at what mistake was
 # probably made.
 for date in sorted_dates:
-    sum_district = sorted_districts[-1]
     day_sum = 0
-    for district in sorted_districts[:-1]:
+    for district in [d for d in sorted_districts if not d==sum_district]:
         day_sum += db[district][date]['new_infections']
     if day_sum != db[sum_district][date]['new_infections']:
         raise Exception('Questionable district infection sum in %s' % date)