Skip to content
This repository has been archived by the owner on Dec 22, 2022. It is now read-only.

Commit

Permalink
Script that gets Costa Rcias's cases from the API in coronaviruscr.com
Browse files Browse the repository at this point in the history
  • Loading branch information
Diego committed Apr 5, 2020
1 parent 8a8f3ed commit e202802
Showing 1 changed file with 53 additions and 0 deletions.
53 changes: 53 additions & 0 deletions utils/scripts/costarica_data.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
import pandas as pd
import requests
import json
import sys
import os

api_url = "https://coronaviruscr.com/api/reports"

json_data = requests.get(api_url).content.decode()

parsed = json.loads(json_data)["data"]

for entry in parsed:
date = entry["date"]
locations = entry["byLocation"]
print(date)
confirmed = pd.DataFrame(
zip(locations.keys(), locations.values()),
columns=["Subdivision", "Confirmed"]
)

confirmed.Subdivision = confirmed.Subdivision.str.title()
confirmed.Subdivision = confirmed.Subdivision.str.replace("Sanjose", "San Jose")
confirmed = confirmed[confirmed.Subdivision!="Unknown"]
confirmed = confirmed.sort_values("Subdivision")

print(confirmed)

daily_report_file = f"../../latam_covid_19_data/latam_covid_19_daily_reports/{date}.csv"
print(f"Opening {daily_report_file}")
daily_report = pd.read_csv(daily_report_file)
cr = daily_report[daily_report.Country=="Costa Rica"]
cr_index = cr.index
del cr["Confirmed"]
cr["Last Update"] = date
ncr = pd.merge(cr, confirmed, how="left", on="Subdivision").set_index(cr_index)

daily_report.update(ncr)


daily_report.Deaths = daily_report.Deaths.fillna(0)
daily_report.Confirmed = daily_report.Confirmed.fillna(0)
daily_report.Recovered = daily_report.Recovered.fillna(0)

daily_report.Deaths = daily_report.Deaths.astype("int64")
daily_report.Confirmed = daily_report.Confirmed.astype("int64")
daily_report.Recovered = daily_report.Recovered.astype("int64")

print(daily_report[daily_report.Country=="Costa Rica"])
os.system("git pull")
daily_report.to_csv(daily_report_file, index=False)
os.system(f"git add {daily_report_file}")
os.system("git commit -m 'Update Costa rica'")

0 comments on commit e202802

Please sign in to comment.