Horizontale "all of Berlin" header in HTML table.

[berlin-corona-table] / enhance_table.py
diff --git a/enhance_table.py b/enhance_table.py

index 02a7ac9b30b8cd0375a79a49468e2dc5c19e3d77..66614c36433a9e64422fe1768ee11ef217624fde 100755 (executable)
--- a/enhance_table.py
+++ b/enhance_table.py
@@ -17,6 +17,26 @@ district_pops = {
    'sum': 3754418,
  }
  
+# 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) != 3:
@@ -114,17 +134,28 @@ 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:
-        print('<th>%s</th>' % district)
+        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>')
-    sorted_dates.reverse()
      for date in sorted_dates:
-        print('<tr>')
+        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 = ''
@@ -136,12 +167,15 @@ if output_type == 'html':
              if 'week_incidence' in district_data:
                  week_inc = '%.1f' % district_data['week_incidence']
              print('<td>')
-            print('<table>')
-            print('<tr><th>new</th><td>%s</td></tr>' % new_infections)
-            print('<tr><th>wsum</th><td>%s</td></tr>' % week_sum)
-            print('<tr><th>wavg</th><td>%s</td></tr>' % week_avg)
-            print('<tr><th>winc</th><td>%s</td></tr>' % week_inc)
-            print('</table>')
+            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>')
@@ -156,24 +190,10 @@ elif output_type == 'txt':
  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)