mirror of https://github.com/stijndcl/didier
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
from aiohttp import ClientSession
|
|
|
|
from didier.data.embeds.disease_sh import CovidData
|
|
from didier.utils.http.requests import ensure_get
|
|
|
|
__all__ = ["get_country_info", "get_global_info"]
|
|
|
|
|
|
async def get_country_info(http_session: ClientSession, country: str) -> CovidData:
|
|
"""Fetch the info for a given country for today and yesterday"""
|
|
endpoint = f"https://disease.sh/v3/covid-19/countries/{country}"
|
|
|
|
params = {"yesterday": 0, "strict": 1, "allowNull": 0}
|
|
async with ensure_get(http_session, endpoint, params=params) as response:
|
|
today = response
|
|
|
|
params["yesterday"] = 1
|
|
async with ensure_get(http_session, endpoint, params=params) as response:
|
|
yesterday = response
|
|
|
|
data = {"today": today, "yesterday": yesterday}
|
|
return CovidData.model_validate(data)
|
|
|
|
|
|
async def get_global_info(http_session: ClientSession) -> CovidData:
|
|
"""Fetch the global info for today and yesterday"""
|
|
endpoint = "https://disease.sh/v3/covid-19/all"
|
|
|
|
params = {"yesterday": 0, "allowNull": 0}
|
|
async with ensure_get(http_session, endpoint, params=params) as response:
|
|
today = response
|
|
|
|
params["yesterday"] = 1
|
|
async with ensure_get(http_session, endpoint, params=params) as response:
|
|
yesterday = response
|
|
|
|
data = {"today": today, "yesterday": yesterday}
|
|
return CovidData.model_validate(data)
|