X-Git-Url: https://plomlompom.com/repos/new_day?a=blobdiff_plain;f=enhance_table.py;h=28c472eeeda95650488700c8b4384c3cae7e4d44;hb=a2f73f130d45247e0097278f0c1388583bf18986;hp=774b333a77cf96951609c4363f219f96d158244a;hpb=ea8a53f23b20bebeeabb6716a5587731f60b28d5;p=berlin-corona-table
diff --git a/enhance_table.py b/enhance_table.py
index 774b333..28c472e 100755
--- a/enhance_table.py
+++ b/enhance_table.py
@@ -17,12 +17,36 @@ 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',
+ '+': '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.
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()
@@ -72,17 +96,20 @@ for line in lines[1:]:
sorted_dates.sort()
# In LaGeSo's data, the last "district" is actually the sum of all districts /
-# the whole of Berlin.
-#
+# the whole of Berlin. For our district order, move it in front of the other
+# districts, as its numbers are the most interesting, so in the table views
+# we want to see it first.
+sum_district = sorted_districts.pop()
+sorted_districts.insert(0, sum_district)
+
# Fail on any day where the "sum" district's new infections are not the proper
# sum of the individual districts new infections. Yes, sometimes Lageso sends
# data that is troubled in this way. It will then have to be fixed manually in
# 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)
@@ -106,79 +133,138 @@ 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.
-Updated daily around 9pm.
+# Optimized for web browser viewing.
+import calendar
+if output_type == 'html':
+ print("""
+
+
+
+Berlin's Corona infection numbers, development by districts
+
+home · contact · privacy
+Berlin's Corona infection numbers, development by districts
+Updated daily at 7pm based on data from the "Senatsverwaltung für Gesundheit, Pflege und Gleichstellung". Source code. Plain text view (optimized for terminal curl).
+
+
+ | """)
+ sorted_dates.reverse()
+ for district in sorted_districts:
+ # Wrap in div because the vertical orientation otherwise fails
+ # in Chromium.
+ print(' | ' %
+ translate[district])
+ print('
')
+ weekday_count = 0
+ for date in sorted_dates:
+ if weekday_count == 0:
+ print('')
+ print('date | ')
+ print('? | ')
+ for district in sorted_districts:
+ print('%s | ' %
+ (translate[district], district))
+ print('
')
+ print('')
+ weekday = calendar.day_name[datetime.date.fromisoformat(date).weekday()]
+ print('%s %s | ' % (date, weekday))
+ print('')
+ for abbr in ['+', 'Σ', 'Ã', 'i']:
+ print('%s | ' %
+ (translate[abbr], abbr))
+ print(' | ')
+ 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('')
+ print('')
+ print('%s | ' % new_infections)
+ print('%s | ' % week_sum)
+ print('%s | ' % week_avg)
+ print('%s | ' % week_inc)
+ print(' ')
+ print(' | ')
+ print('
')
+ weekday_count += 1
+ if weekday_count != 7:
+ continue
+ weekday_count = 0
+ print('
')
+ print('Symbols
')
+ print('')
+ for abbr in ['+', 'Σ', 'Ã', 'i']:
+ print('- %s
- %s
' % (abbr, translate[abbr]))
+ print('
')
+ print('')
-Abbrevations/explanations:
+# Optimized for in-terminal curl.
+elif output_type == 'txt':
-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
+ # Explain what this is.
+ intro = \
+"""Table of Berlin's Corona infection number development by districts.
+Updated daily at 7pm based on data from the "Senatsverwaltung für Gesundheit, Pflege und Gleichstellung".
-Source code: https://plomlompom.com/repos/?p=berlin-corona-table
+Abbrevations/explanations:
"""
-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
+ for k in translate:
+ intro += "%s: %s\n" % (k, translate[k])
+ intro += """
+Source code: https://plomlompom.com/repos/?p=berlin-corona-table
+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
+ for date in sorted_dates:
+
+ # Week table header.
+ if weekday_count == 0:
+ print()
+ print(' '*13, ' '.join(sorted_districts))
+ print('-'*77)
- # 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()
+ # Day table.
+ weekday = calendar.day_name[datetime.date.fromisoformat(date).weekday()]
+ print('%s (%s)' % (date, weekday))
+ new_infections = []
+ weekly_sum_strings = []
+ weekly_avg_strings = []
+ weekly_inc_strings = []
+ for district in sorted_districts:
+ district_day_data = db[district][date]
+ new_infections += [district_day_data['new_infections']]
+ wsum_string = ' '*3 + '?'
+ wavg_string = winc_string = ' '*4 + '?'
+ if 'week_sum' in district_day_data:
+ wsum_string = '%4s' % district_day_data['week_sum']
+ weekly_sum_strings += [wsum_string]
+ if 'week_average' in district_day_data:
+ wavg_string = '%5.1f' % district_day_data['week_average']
+ weekly_avg_strings += [wavg_string]
+ if 'week_incidence' in district_day_data:
+ winc_string = '%5.1f' % district_day_data['week_incidence']
+ weekly_inc_strings += [winc_string]
+ print('+', ' '*11, ' '.join(['%3s' % i for i in new_infections]))
+ print('Σ', ' '*10, ' '.join(weekly_sum_strings))
+ print('Ã', ' '*9, ''.join(weekly_avg_strings))
+ print('i', ' '*9, ''.join(weekly_inc_strings))
+ weekday_count += 1
+ if weekday_count != 7:
+ continue
+ weekday_count = 0