def fetch_and_prepare_lk_time_series(lk_id: str) -> list:
"""
calles fetch_landkreis_time_series
convert and add fields of time series list
returns list
writes json and tsv to filesystem
"""
l_time_series_fetched = fetch_landkreis_time_series(
lk_id=lk_id, readFromCache=True)
l_time_series = []
# entry = one data point
for entry in l_time_series_fetched:
d = {}
# covert to int
d['Cases'] = int(entry['SummeFall'])
d['Deaths'] = int(entry['SummeTodesfall'])
# these are calculated below
# d['Cases_New'] = int(entry['AnzahlFall'])
# d['Deaths_New'] = int(entry['AnzahlTodesfall'])
# Rename 'Meldedatum' (ms) -> Timestamp (s)
d['Timestamp'] = int(entry['Meldedatum'] / 1000)
# add Date
d['Date'] = helper.convert_timestamp_to_date_str(
d['Timestamp'])
l_time_series.append(d)
l_time_series = helper.prepare_time_series(l_time_series)
for i in range(len(l_time_series)):
d = l_time_series[i]
# _Per_Million
d = helper.add_per_million_via_lookup(d, d_ref_landkreise, lk_id)
l_time_series[i] = d
# data_t.append(d['Days_Past'])
# data_cases.append(d['Cases'])
# data_deaths.append(d['Deaths'])
# data_cases_new.append((d['Days_Past'], d['Cases_New']))
# data_deaths_new.append((d['Days_Past'], d['Deaths_New']))
# # perform fit for last 7 days to obtain doubling time
# data = list(zip(data_t, data_cases))
# fit_series_res = helper.series_of_fits(
# data, fit_range=7, max_days_past=14)
# for i in range(len(l_time_series)):
# entry = l_time_series[i]
# this_doubling_time = ""
# this_days_past = entry['Days_Past']
# if this_days_past in fit_series_res:
# this_doubling_time = fit_series_res[this_days_past]
# entry['Cases_Doubling_Time'] = this_doubling_time
# l_time_series[i] = entry
return l_time_series
def read_csv_to_dict() -> dict:
"""
read and convert the source csv file, containing: federalstate,infections,deaths,date,newinfections,newdeaths
re-calc _New via helper.prepare_time_series
add _Per_Million via helper.add_per_million_via_lookup
"""
global d_ref_states
# Preparations
d_states_data = {'BW': [], 'BY': [], 'BE': [], 'BB': [], 'HB': [], 'HH': [], 'HE': [], 'MV': [
], 'NI': [], 'NW': [], 'RP': [], 'SL': [], 'SN': [], 'ST': [], 'SH': [], 'TH': []}
# add German sum
d_states_data['DE-total'] = []
d_german_sums = {} # date -> 'infections', 'deaths', 'new infections', 'new deaths'
# data body
with open(download_file, mode='r', encoding='utf-8') as f:
csv_reader = csv.DictReader(f, delimiter=",")
for row in csv_reader:
d = {}
s = row['date']
l = s.split("-")
d['Date'] = helper.date_format(
int(l[0]), int(l[1]), int(l[2]))
d['Cases'] = int(row["infections"])
d['Deaths'] = int(row["deaths"])
if row["federalstate"] == 'Baden-Württemberg':
d_states_data['BW'].append(d)
elif row["federalstate"] == 'Bavaria':
d_states_data['BY'].append(d)
elif row["federalstate"] == 'Berlin':
d_states_data['BE'].append(d)
elif row["federalstate"] == 'Brandenburg':
d_states_data['BB'].append(d)
elif row["federalstate"] == 'Bremen':
d_states_data['HB'].append(d)
elif row["federalstate"] == 'Hamburg':
d_states_data['HH'].append(d)
elif row["federalstate"] == 'Hesse':
d_states_data['HE'].append(d)
elif row["federalstate"] == 'Lower Saxony':
d_states_data['NI'].append(d)
elif row["federalstate"] == 'North Rhine-Westphalia':
d_states_data['NW'].append(d)
elif row["federalstate"] == 'Mecklenburg-Western Pomerania':
d_states_data['MV'].append(d)
elif row["federalstate"] == 'Rhineland-Palatinate':
d_states_data['RP'].append(d)
elif row["federalstate"] == 'Saarland':
d_states_data['SL'].append(d)
elif row["federalstate"] == 'Saxony':
d_states_data['SN'].append(d)
elif row["federalstate"] == 'Saxony-Anhalt':
d_states_data['ST'].append(d)
elif row["federalstate"] == 'Schleswig-Holstein':
d_states_data['SH'].append(d)
elif row["federalstate"] == 'Thuringia':
d_states_data['TH'].append(d)
else:
assert 1 == 2, f"ERROR: unknown state: {row['federalstate']}"
# add to German sum
if d['Date'] not in d_german_sums:
d2 = {}
d2['Cases'] = d['Cases']
d2['Deaths'] = d['Deaths']
else:
d2 = d_german_sums[d['Date']]
d2['Cases'] += d['Cases']
d2['Deaths'] += d['Deaths']
d_german_sums[d['Date']] = d2
del d2
# German sum -> same dict
for datum in d_german_sums.keys():
d = d_german_sums[datum]
d['Date'] = datum # add date field
d_states_data['DE-total'].append(d)
del d_german_sums, d
# check if DE-total of today and yesterday are equal, if so: remove last date
if d_states_data['DE-total'][-1]['Cases'] == d_states_data['DE-total'][-2]['Cases']:
print("WARNING: DE cases sum is unchanged")
for code in d_states_data:
d_states_data[code].pop()
print(f"DE-States Last Date: {d_states_data['DE-total'][-1]['Date']}")
for code in d_states_data.keys():
l_time_series = d_states_data[code]
# add days past, _New, _Last_Week, etc
l_time_series = helper.prepare_time_series(l_time_series)
for i in range(len(l_time_series)):
d = l_time_series[i]
# add per Million rows
d = helper.add_per_million_via_lookup(d, d_ref_states, code)
# # fit cases data
# dataCases = []
# dataDeaths = []
# for i in range(1, len(l_time_series)):
# # x= day , y = cases
# dataCases.append(
# (
# l_time_series[i]['Days_Past'],
# l_time_series[i]['Cases']
# )
# )
# dataDeaths.append(
# (
# l_time_series[i]['Days_Past'],
# l_time_series[i]['Deaths']
# )
# )
# fit_series_res = helper.series_of_fits(
# dataCases, fit_range=7, max_days_past=60)
# for i in range(0, len(l_time_series)):
# this_Doubling_Time = ""
# this_days_past = l_time_series[i]['Days_Past']
# if this_days_past in fit_series_res:
# this_Doubling_Time = fit_series_res[this_days_past]
# l_time_series[i]['Cases_Doubling_Time'] = this_Doubling_Time
# fit_series_res = helper.series_of_fits(
# dataDeaths, fit_range=7, max_days_past=60)
# for i in range(0, len(l_time_series)):
# this_Doubling_Time = ""
# this_days_past = l_time_series[i]['Days_Past']
# if this_days_past in fit_series_res:
# this_Doubling_Time = fit_series_res[this_days_past]
# l_time_series[i]['Deaths_Doubling_Time'] = this_Doubling_Time
d_states_data[code] = l_time_series
return d_states_data