home · contact · privacy
Horizontale "all of Berlin" header in HTML table.
[berlin-corona-table] / enhance_table.py
index 774b333a77cf96951609c4363f219f96d158244a..66614c36433a9e64422fe1768ee11ef217624fde 100755 (executable)
@@ -17,12 +17,35 @@ district_pops = {
   'sum': 3754418,
 }
 
-# Read infections table file lines.
+# Map abbreviations to full names.
+translate = {
+  'CW': 'Charlottenburg-Wilmersdorf',
+  'FK': 'Friedrichshain-Kreuzberg',
+  'Li': 'Lichtenberg',
+  'MH': 'Marzahn-Hellersdorf',
+  'Mi': 'Mitte',
+  'Ne': 'Neukölln',
+  'Pa': 'Pankow',
+  'Re': 'Reinickendorf',
+  'Sp': 'Spandau',
+  'SZ': 'Steglitz-Zehlendorf',
+  'TS': 'Tempelhof-Schöneberg',
+  'TK': 'Treptow-Köpenick',
+  'sum': 'all of Berlin',
+  'wsum': 'sum 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',
+}
+
+# Read infections table path and output type.
 import sys
-if len(sys.argv) != 2:
-    print('Expecting infections table file path as only argument.')
+if len(sys.argv) != 3:
+    print('Expecting infections table file path and output type as only arguments.')
     exit(1)
 infections_table = sys.argv[1]
+output_type = sys.argv[2]
+
+# Read infections table file lines.
 f = open(infections_table, 'r')
 lines = f.readlines()
 f.close()
@@ -106,79 +129,122 @@ for i in range(len(sorted_dates)):
         db[district][date]['week_average'] = week_sum / 7
         db[district][date]['week_incidence'] = (week_sum / district_pop) * 100000
 
-# Explain what this is.
-intro = """Table of Berlin's Corona infection number development by districts.
+# Optimized for web browser viewing.
+if output_type == 'html':
+    print('<html>')
+    print('<style>')
+    print('table, tr, th, td { border: 1px solid black; }')
+    print('.day_row:nth-child(7n+2) { background-color: yellow; }')
+    print('.district_name { writing-mode: vertical-rl; '
+          'transform: rotate(180deg); }')
+    print('</style>')
+    print('<table>')
+    print('<tr>')
+    print('<th>date</th>')
+    sorted_dates.reverse()
+    sum_district = sorted_districts[-1]
+    for district in sorted_districts:
+        long_form = translate[district]
+        if sum_district == district:
+            print('<th>%s</th>' % long_form)
+        else:
+            print('<th class="district_name">%s</th>' % long_form)
+    print('</tr>')
+    for date in sorted_dates:
+        print('<tr class="day_row">')
+        print('<td>%s</td>' % date)
+        long_wsum = translate['wsum']
+        long_wavg = translate['wavg']
+        long_winc = translate['winc']
+        for district in sorted_districts:
+            district_data = db[district][date]
+            week_sum = week_avg = week_inc = ''
+            new_infections = district_data['new_infections']
+            if 'week_sum' in district_data:
+                week_sum = '%s' % district_data['week_sum']
+            if 'week_average' in district_data:
+                week_avg = '%.1f' % district_data['week_average']
+            if 'week_incidence' in district_data:
+                week_inc = '%.1f' % district_data['week_incidence']
+            print('<td>')
+            print(new_infections)
+            if district == sum_district:
+                print('<details open><summary></summary><table>')
+            else:
+                print('<details><summary></summary><table>')
+            print('<tr><th>%s</th><td>%s</td></tr>' % (long_wsum, week_sum))
+            print('<tr><th>%s</th><td>%s</td></tr>' % (long_wavg, week_avg))
+            print('<tr><th>%s</th><td>%s</td></tr>' % (long_winc, week_inc))
+            print('</table></details>')
+            print('</td>')
+        print('</tr>')
+    print('</table>')
+    print('</html>')
+
+# Optimized for in-terminal curl.
+elif output_type == 'txt':
+
+    # Explain what this is.
+    intro = \
+"""Table of Berlin's Corona infection number development by districts.
 Updated daily around 9pm.
 
 Abbrevations/explanations:
-
-CW: Charlottenburg-Wilmersdorf
-FK: Friedrichshain-Kreuzberg
-Li: Lichtenberg
-MH: Marzahn-Hellersdorf
-Mi: Mitte
-Ne: Neukölln
-Pa: Pankow
-Re: Reinickendorf
-Sp: Spandau
-SZ: Steglitz-Zehlendorf
-TS: Tempelhof-Schöneberg
-TK: Treptow-Köpenick
-sum: sum for all the districts
-wsum: sum 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
-
+"""
+    for k in translate:
+        intro += "%s: %s\n" % (k, translate[k])
+    intro += """
 Source code: https://plomlompom.com/repos/?p=berlin-corona-table
 """
-print(intro)
+    print(intro)
 
-# Output table of enhanced daily infection data, newest on top, separated into
-# 7-day units.
-sorted_dates.reverse()
-weekday_count = 0
-for date in sorted_dates:
-
-    # Week table header.
-    if weekday_count == 0:
-        print(' '*11, '  '.join(sorted_districts[:-1]),
-              sorted_districts[-1], 'wsum', ' wavg', 'winc')
-        week_start_date = date
-
-    # Day data line.
-    new_infections = []
-    for district in sorted_districts:
-        new_infections += [db[district][date]['new_infections']]
-    week_sum = week_avg = week_inc = ''
-    sum_district = sorted_districts[-1]
-    sum_district_data = db[sum_district][date]
-    if 'week_sum' in sum_district_data:
-        week_sum = '%4s' % sum_district_data['week_sum']
-    if 'week_average' in sum_district_data:
-        week_avg = '%5.1f' % sum_district_data['week_average']
-    if 'week_incidence' in sum_district_data:
-        week_inc = '%4.1f' % sum_district_data['week_incidence']
-    print(date, ' '.join(['%3s' % infections for infections in new_infections]),
-          week_sum, week_avg, week_inc)
-
-    # Maintain 7-day cycle.
-    weekday_count += 1
-    if weekday_count != 7:
-        continue
+    # Output table of enhanced daily infection data, newest on top,
+    # separated into 7-day units.
+    sorted_dates.reverse()
     weekday_count = 0
-
-    # After each 7 days, print summary for individual districts.
-    weekly_sums = []
-    weekly_avgs = []
-    weekly_incs = []
-    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()
+    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')
+            week_start_date = date
+
+        # Day data line.
+        new_infections = []
+        for district in sorted_districts:
+            new_infections += [db[district][date]['new_infections']]
+        week_sum = week_avg = week_inc = ''
+        sum_district_data = db[sum_district][date]
+        if 'week_sum' in sum_district_data:
+            week_sum = '%4s' % sum_district_data['week_sum']
+        if 'week_average' in sum_district_data:
+            week_avg = '%5.1f' % sum_district_data['week_average']
+        if 'week_incidence' in sum_district_data:
+            week_inc = '%4.1f' % sum_district_data['week_incidence']
+        print(date, ' '.join(['%3s' % infections
+                              for infections in new_infections]),
+              week_sum, week_avg, week_inc)
+
+        # Maintain 7-day cycle.
+        weekday_count += 1
+        if weekday_count != 7:
+            continue
+        weekday_count = 0
+
+        # After each 7 days, print summary for individual districts.
+        weekly_sums = []
+        weekly_avgs = []
+        weekly_incs = []
+        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()