81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
|
import json
|
||
|
|
from flask import Flask, render_template, request
|
||
|
|
from datetime import datetime, date
|
||
|
|
|
||
|
|
import requests
|
||
|
|
|
||
|
|
url = 'https://covidtrackerapi.bsg.ox.ac.uk/api/v2/stringency/actions/'
|
||
|
|
'''
|
||
|
|
01. Belarus - BLR
|
||
|
|
02. China - CHN
|
||
|
|
03. France - FRA
|
||
|
|
04. Germany - DEU
|
||
|
|
05. India - IND
|
||
|
|
06. Israel - ISR
|
||
|
|
07. Russia - RUS
|
||
|
|
08. Serbia - SRB
|
||
|
|
09. Spain - ESP
|
||
|
|
10. Turkey - TUR
|
||
|
|
'''
|
||
|
|
countries = ['blr', 'chn', 'fra', 'deu', 'ind', 'isr', 'rus', 'srb', 'esp', 'tur']
|
||
|
|
tmp_output = {}
|
||
|
|
|
||
|
|
|
||
|
|
def check_date(input_date):
|
||
|
|
""" Check date format from input string.
|
||
|
|
|
||
|
|
Args: input_date - (str).
|
||
|
|
Return: Bool - True if date correct. """
|
||
|
|
try:
|
||
|
|
# valid_date = datetime.strptime(input_date, '%Y-%m-%d').date()
|
||
|
|
datetime.strptime(input_date, '%Y-%m-%d').date()
|
||
|
|
return True
|
||
|
|
except ValueError:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def request_one_country(api, cntr, day):
|
||
|
|
""" Request from api-url.
|
||
|
|
|
||
|
|
Args: api - (str) - api url,
|
||
|
|
cntr - (str) country alpha-3 code,
|
||
|
|
day - (int) YYYY-MM-DD.
|
||
|
|
Return: Response - (dict). """
|
||
|
|
r_out = requests.get(f'{api}{cntr}/{day}')
|
||
|
|
return r_out.json()
|
||
|
|
|
||
|
|
######################################################
|
||
|
|
|
||
|
|
|
||
|
|
app = Flask(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
@app.route('/', methods=['post', 'get'])
|
||
|
|
def index():
|
||
|
|
if request.method == 'POST':
|
||
|
|
input_date = request.form.get('input_date') # запрос к данным формы
|
||
|
|
if check_date(input_date):
|
||
|
|
correct_date = datetime.strptime(input_date, '%Y-%m-%d').date()
|
||
|
|
if correct_date >= date.today():
|
||
|
|
return render_template('index.html', message='Input date has not arrived yet')
|
||
|
|
|
||
|
|
for country in countries:
|
||
|
|
out = request_one_country(url, country, correct_date)
|
||
|
|
tmp_output.setdefault(country, out.get('stringencyData'))
|
||
|
|
|
||
|
|
output = {input_date: tmp_output}
|
||
|
|
with open('../temp_out.json', 'w') as f:
|
||
|
|
json.dump(output, f, sort_keys=True, indent=2)
|
||
|
|
|
||
|
|
return render_template('index.html')
|
||
|
|
elif not input_date:
|
||
|
|
return render_template('index.html', message='Please input date')
|
||
|
|
else:
|
||
|
|
return render_template('index.html', message='Incorrect input')
|
||
|
|
else:
|
||
|
|
return render_template('index.html')
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
app.run(host='127.0.0.1', port=8181, debug=True)
|