diff --git a/app/corona.py b/app/corona.py index b37a67e3..ac4f0304 100644 --- a/app/corona.py +++ b/app/corona.py @@ -48,9 +48,12 @@ def query_csv(csv_path, location): def sum_numbers(location): today, yesterday = get_csv() + try: + last_updated, yesterday_sum = query_csv(yesterday, location) + last_updated, today_sum = query_csv(today, location) + except TypeError: + return discord.Embed(description="No data for %s. Did you make a typo?" % location, color=0x428bca, type="rich") - last_updated, yesterday_sum = query_csv(yesterday, location) - last_updated, today_sum = query_csv(today, location) confirmed = "%s (+%s from yesterday)" % (today_sum['Confirmed'], int(today_sum['Confirmed']) - int(yesterday_sum['Confirmed'])) deaths = "%s (+%s from yesterday)" % (today_sum['Deaths'], int(today_sum['Deaths']) - int(yesterday_sum['Deaths']))