#!//usr/bin/env python3
-import sys
-if len(sys.argv) != 2:
- print('Expecting infections table file path as only argument.')
- exit(1)
-infections_table = sys.argv[1]
-
# District population numbers as per Wikipedia.
district_pops = {
'CW': 342332,
'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:
+ 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()
+# Basic input validation.
+import datetime
+header_elements = lines[0].split()
+if set(header_elements) != district_pops.keys() or \
+ len(header_elements) != len(district_pops.keys()):
+ raise Exception('infections table: invalid header')
+line_count = 0
+for line in lines[1:]:
+ line_count += 1
+ fields = line.split()
+ if len(header_elements) != len(fields) - 1:
+ raise Exception('infections table: too many elements on line %s',
+ line_count)
+ try:
+ datetime.date.fromisoformat(fields[0])
+ except ValueError:
+ raise Exception('infections table: bad ISO date on line %s',
+ line_count)
+ for field in fields[1:]:
+ try:
+ int(field)
+ except ValueError:
+ raise Exception('infections table: bad value on line %s',
+ line_count)
+
# Parse first table file line for the names and order of districts.
db = {}
sorted_districts = []
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>
+<style>
+table, tr, th, td { border: 1px solid black; }
+.day_row:nth-child(7n+2) { background-color: yellow; }
+.district_name { writing-mode: vertical-rl; transform: rotate(180deg); }
+</style>
+<h1>Table of Berlin's Corona infection number development by districts.
+</h1>
+<p>Updated daily at 9pm. <a href="https://plomlompom.com/repos/?p=berlin-corona-table">Source code</a>. <a href="berlin_corona.txt">Text view optimized for terminal curl</a>.</p>
+<table>
+<tr>
+<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
-
-Source code: https://plomlompom.com/repos/?p=berlin-corona-table
"""
-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:
+ for k in translate:
+ intro += "%s: %s\n" % (k, translate[k])
+ intro += """
+Source code: https://plomlompom.com/repos/?p=berlin-corona-table
- # Week table header.
- if weekday_count == 0:
- print(' '*11, ' '.join(sorted_districts[:-1]),
- sorted_districts[-1], 'wsum', ' wavg', 'winc')
- week_start_date = date
+HTML view: https://plomlompom.com/berlin_corona.html
+"""
+ print(intro)
- # 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()