X-Git-Url: https://plomlompom.com/repos/?p=berlin-corona-table;a=blobdiff_plain;f=enhance_table.py;h=cee78f82bb674fb98a879c1814bc6a2cca078ea8;hp=7da699dfdee9cc1feab083c05113942085aa1d08;hb=56de61949ed35ecb8ab728dd1912e161892bc521;hpb=707cfa7c1b642936ba94feb1be2d7a071bc0cdcf

diff --git a/enhance_table.py b/enhance_table.py
index 7da699d..cee78f8 100755
--- a/enhance_table.py
+++ b/enhance_table.py
@@ -32,9 +32,10 @@ translate = {
   'TS': 'Tempelhof-Schöneberg',
   'TK': 'Treptow-Köpenick',
   'sum': 'all of Berlin',
-  'wsum': 'sum of new infections for last 7 days',
-  'wavg': 'per-day average of new infections for last 7 days',
-  'winc': 'incidence (x per 100k inhabitants) of new infections for last 7 days',
+  '+': 'new infections counted that day',
+  'Σ': 'sum of new infections for last 7 days',
+  'Ø': 'per-day average of new infections for last 7 days',
+  'i': 'incidence (x per 100k inhabitants) of new infections for last 7 days',
 }
 
 # Read infections table path and output type.
@@ -94,6 +95,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 +108,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)
@@ -150,26 +154,25 @@ th { text-align: left; vertical-align: bottom; }
 <tr>
 <th colspan=2></th>""")
     sorted_dates.reverse()
-    sum_district = sorted_districts[-1]
     for district in sorted_districts:
-        long_form = translate[district]
         # Wrap in div because the vertical orientation otherwise fails
         # in Chromium.
-        print('<th><div class="vertical_header">%s</div></th>' % long_form)
+        print('<th><div class="vertical_header">%s</div></th>' %
+              translate[district])
     print('</tr>')
     print('<tr class="fixed_head">')
     # In Chromium, the th only stay fixed if also given this class.
     print('<th class="fixed_head">date</th>')
-    print('<th class="fixed_head"></th>')
+    print('<th class="fixed_head"><a href="#symbols">?</a></th>')
     for district in sorted_districts:
-        print('<th class="fixed_head">%s</th>' % district)
+        print('<th class="fixed_head"><abbr title="%s">%s</abbr></th>' %
+              (translate[district], district))
     print('</tr>')
     for date in sorted_dates:
         print('<tr class="day_row">')
         print('<td>%s</td>' % date)
         print('<td><table>')
-        print('<tr><th><abbr title="new infections counted">new</abbr></th></tr>')
-        for abbr in ['wsum', 'wavg', 'winc']:
+        for abbr in ['+', 'Σ', 'Ø', 'i']:
             print('<tr><th><abbr title="%s">%s</abbr></th></tr>' %
                   (translate[abbr], abbr))
         print('</table></td>')
@@ -193,6 +196,11 @@ th { text-align: left; vertical-align: bottom; }
             print('</td>')
         print('</tr>')
     print('</table>')
+    print('<h3 id="symbols">Symbols</h3>')
+    print('<dl>')
+    for abbr in ['+', 'Σ', 'Ø', 'i']:
+        print('<dt>%s</dt><dd>%s</dd>' % (abbr, translate[abbr]))
+    print('</dl>')
     print('</html>')
 
 # Optimized for in-terminal curl.
@@ -206,6 +214,8 @@ Updated daily at 9pm based on data from the "Senatsverwaltung für Gesundheit, P
 Abbrevations/explanations:
 """
     for k in translate:
+        if k == '+':  # not used in terminal version
+            continue
         intro += "%s: %s\n" % (k, translate[k])
     intro += """
 Source code: https://plomlompom.com/repos/?p=berlin-corona-table
@@ -217,13 +227,12 @@ HTML view: https://plomlompom.com/berlin_corona.html
     # separated into 7-day units.
     sorted_dates.reverse()
     weekday_count = 0
-    sum_district = sorted_districts[-1]
     for date in sorted_dates:
 
         # Week table header.
         if weekday_count == 0:
-            print(' '*11, '  '.join(sorted_districts[:-1]),
-                  sorted_districts[-1], 'wsum', ' wavg', 'winc')
+            print(' '*10, '  '.join(sorted_districts),
+                  '   Σ', '    Ø', '   i')
             week_start_date = date
 
         # Day data line.
@@ -252,14 +261,14 @@ HTML view: https://plomlompom.com/berlin_corona.html
         weekly_sums = []
         weekly_avgs = []
         weekly_incs = []
-        for district in sorted_districts[:-1]:
+        for district in sorted_districts[1:]:
             weekly_sums += [db[district][week_start_date]['week_sum']]
             weekly_avgs += [db[district][week_start_date]['week_average']]
             weekly_incs += [db[district][week_start_date]['week_incidence']]
         print()
         print('district stats for week from %s to %s:' % (date, week_start_date))
-        print(' '*7, '    '.join(sorted_districts[:-1]))
-        print('wsum', ' '.join(['%5.1f' % wsum for wsum in weekly_sums]))
-        print('wavg', ' '.join(['%5.1f' % wavg for wavg in weekly_avgs]))
-        print('winc', ' '.join(['%5.1f' % winc for winc in weekly_incs]))
+        print(' '*4, '    '.join(sorted_districts[1:]))
+        print('Σ', ' '.join(['%5.1f' % wsum for wsum in weekly_sums]))
+        print('Ø', ' '.join(['%5.1f' % wavg for wavg in weekly_avgs]))
+        print('i', ' '.join(['%5.1f' % winc for winc in weekly_incs]))
         print()