From ce735ed991a0e00d4a5933e04b2c9a20003c21be Mon Sep 17 00:00:00 2001 From: ein Date: Fri, 26 Feb 2021 06:59:08 -0800 Subject: [PATCH] Error handling for corona function if people type in nonsensical locations --- app/corona.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) 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']))