X-Git-Url: https://plomlompom.com/repos/do_day?a=blobdiff_plain;f=enhance_table.py;h=5287c03a3ea30796251eb1fda25455daa25b8632;hb=f0145a1f179e5fd3e61ab123fcc8ad967586fa90;hp=0482bdfbffbf8fe0e89bb174454b04f561318a92;hpb=5252dee46a33cce9616ba4e4103b59dba2ec44c1;p=berlin-corona-table
diff --git a/enhance_table.py b/enhance_table.py
index 0482bdf..5287c03 100755
--- a/enhance_table.py
+++ b/enhance_table.py
@@ -32,6 +32,9 @@ translate = {
'TS': 'Tempelhof-Schöneberg',
'TK': 'Treptow-Köpenick',
'sum': 'all of Berlin',
+ 'wsum': 'sum of new infections 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.
@@ -91,6 +94,10 @@ for line in lines[1:]:
db[district][date] = {'new_infections': int(district_data)}
sorted_dates.sort()
+# Define and move sum_district from end to start.
+sum_district = sorted_districts.pop()
+sorted_districts.insert(0, sum_district)
+
# In LaGeSo's data, the last "district" is actually the sum of all districts /
# the whole of Berlin.
#
@@ -100,9 +107,8 @@ sorted_dates.sort()
# 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)
@@ -128,23 +134,50 @@ for i in range(len(sorted_dates)):
# Optimized for web browser viewing.
if output_type == 'html':
- print('')
- print('')
- print('
')
- print('')
- print('date | ')
+ print("""
+
+
+
+Berlin's Corona infection numbers, development by districts
+
+home · contact · privacy
+Berlin's Corona infection numbers, development by districts
+Updated daily at 9pm based on data from the "Senatsverwaltung für Gesundheit, Pflege und Gleichstellung". Source code. Text view optimized for terminal curl.
+
+
+ | """)
+ sorted_dates.reverse()
for district in sorted_districts:
- print('%s | ' % district)
+ long_form = translate[district]
+ # Wrap in div because the vertical orientation otherwise fails
+ # in Chromium.
+ print(' | ' % long_form)
+ print('
')
+ print('')
+ # In Chromium, the th only stay fixed if also given this class.
+ print('date | ')
+ print(' | ')
+ for district in sorted_districts:
+ print('%s | ' % district)
print('
')
- sorted_dates.reverse()
for date in sorted_dates:
- print('')
+ print('
')
print('%s | ' % date)
+ print('')
+ print('new | ')
+ for abbr in ['wsum', 'wavg', 'winc']:
+ print('%s | ' %
+ (translate[abbr], abbr))
+ print(' | ')
for district in sorted_districts:
district_data = db[district][date]
- week_sum = week_avg = week_inc = ''
+ 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']
@@ -154,10 +187,10 @@ if output_type == 'html':
week_inc = '%.1f' % district_data['week_incidence']
print('')
print('')
- print('new | %s | ' % new_infections)
- print('wsum | %s | ' % week_sum)
- print('wavg | %s | ' % week_avg)
- print('winc | %s | ' % week_inc)
+ print('%s | ' % new_infections)
+ print('%s | ' % week_sum)
+ print('%s | ' % week_avg)
+ print('%s | ' % week_inc)
print(' ')
print(' | ')
print('
')
@@ -170,17 +203,15 @@ elif output_type == 'txt':
# Explain what this is.
intro = \
"""Table of Berlin's Corona infection number development by districts.
-Updated daily around 9pm.
+Updated daily at 9pm based on data from the "Senatsverwaltung für Gesundheit, Pflege und Gleichstellung".
Abbrevations/explanations:
"""
for k in translate:
intro += "%s: %s\n" % (k, translate[k])
- intro += """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
-
+ intro += """
Source code: https://plomlompom.com/repos/?p=berlin-corona-table
+HTML view: https://plomlompom.com/berlin_corona.html
"""
print(intro)
@@ -188,13 +219,12 @@ Source code: https://plomlompom.com/repos/?p=berlin-corona-table
# separated into 7-day units.
sorted_dates.reverse()
weekday_count = 0
- 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')
+ print(' '*10, ' '.join(sorted_districts),
+ 'wsum', ' wavg', 'winc')
week_start_date = date
# Day data line.
@@ -223,13 +253,13 @@ Source code: https://plomlompom.com/repos/?p=berlin-corona-table
weekly_sums = []
weekly_avgs = []
weekly_incs = []
- for district in sorted_districts[:-1]:
+ 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(' '*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]))