X-Git-Url: https://plomlompom.com/repos/?p=berlin-corona-table;a=blobdiff_plain;f=enhance_table.py;h=66614c36433a9e64422fe1768ee11ef217624fde;hp=02a7ac9b30b8cd0375a79a49468e2dc5c19e3d77;hb=e8850998bd3f831bf1291b0d5ea02d53cc7f2802;hpb=a3d2a6533cb6aee29c48c8e1e08e4981c0c8829b

diff --git a/enhance_table.py b/enhance_table.py
index 02a7ac9..66614c3 100755
--- 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)
