covidErlangenAPI/covidAPI.py at master · Lyonnard/covidErlangenAPI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
import requests
from requests_cache import install_cache
import json
import datetime as dt
from matplotlib import pyplot as plt
plt.close('all')

fig, ax1 = plt.subplots()
# ax2 = plt.twinx(ax1)
#Call the APIs
install_cache('cv19erl', expire_after = 3600) #to avoid too many requests
API_hystory = 'https://geodaten.erlangen.de/opendata/api/v1/corona?api_key=db065a8e-68b2-4dae-b94b-d7c3376f6185'
landkreisObjectId = '282' #Erlangen
API_fresh = 'https://services7.arcgis.com/mOBPykOjAyBO2ZKk/arcgis/rest/services/RKI_Landkreisdaten/FeatureServer/0/query?where=OBJECTID%20%3E%3D%20'+landkreisObjectId+'%20AND%20OBJECTID%20%3C%3D%20'+landkreisObjectId+'&outFields=OBJECTID,death_rate,cases,deaths,cases_per_100k,cases_per_population,BL,BL_ID,county,last_update,cases7_per_100k,recovered,cases7_bl_per_100k,cases7_per_100k_txt,cases7_lk&outSR=4326&f=json'
response_hystory = requests.get(API_hystory)
data_hystory = response_hystory.json()
response_fresh = requests.get(API_fresh)
data_fresh = response_fresh.json()

# Take and format the hystorical data
#FIXME the date is managed in a sporty way, it combines dates from the APIs with local date.
deaths = [int(i['deaths']) for i in data_hystory]
dates = [i['stand'] for i in data_hystory]
cases = [int(i['cases']) for i in data_hystory]
cases100k7 = [i['cases7_per_100k'] for i in data_hystory]
dates = [dt.datetime.strptime(i, '%Y-%m-%dT%H:%M:%S.%fZ').date() for i in dates]

# format and appends the fresh data
data_fresh = data_fresh['features'][0]['attributes']
dates.append(dt.date.today() - dt.timedelta(days=1))
cases.append(data_fresh['cases'])
cases100k7.append(data_fresh['cases7_per_100k'])
deaths.append(data_fresh['deaths'])

# print(json.dumps(data2, sort_keys=True, indent=4))

#plot the data
ax1.plot(dates,deaths,label='deaths',color='k')
ax1.plot(dates,cases100k7,label='cases/100k/7',color='b')
# ax2.plot(dates,cases,label='total cases',color='saddlebrown')
ax1.plot(dates[-1],deaths[-1],marker='o',color='k')
ax1.plot(dates[-1],cases100k7[-1],marker='o',color='b')

#formatting the graph
today=dt.date.today().strftime("%d.%m.%Y")
ax1.set_title('Covid Erlangen {} (inzidenz = {:.0f})'.format(today,cases100k7[-1]))
ax1.grid(which='both')
ax1.legend()
# ax2.set_ylabel('total cases',color='saddlebrown')
ax1.set_ylim(bottom=0)
ax1.set_xlim(left=dates[0])
plt.gcf().autofmt_xdate()

#adding the color bands
a,b=ax1.get_xlim()
c,d=ax1.get_ylim()
ax1.fill_between((a,b),35,50,color='green',alpha=0.2)
ax1.fill_between((a,b),50,100,color='orange',alpha=0.2)
ax1.fill_between((a,b),100,d,color='red',alpha=0.2)
ax1.set_xlim(a,b)
ax1.set_ylim(c,d)

#save the figure
fig.savefig('covidplot{}.png'.format(today), bbox_inches=0, dpi=600)
plt.show()