From 5741e080f3d82de7c4ff9385b9b310c659c8a255 Mon Sep 17 00:00:00 2001 From: luke Date: Thu, 26 Mar 2020 09:38:42 -0700 Subject: [PATCH] Now acepts states with spaces in the name --- app/corona.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/app/corona.py b/app/corona.py index 92ff5da4..6e47acc1 100644 --- a/app/corona.py +++ b/app/corona.py @@ -3,6 +3,7 @@ import pandas as pd import requests import os import discord +import string def get_csv(): git_url = 'https://api.github.com/repos/CSSEGISandData/COVID-19/contents/csse_covid_19_data/csse_covid_19_daily_reports' @@ -37,7 +38,7 @@ def sum_numbers(state): def parse_message(message): try: - state = message.split()[1].capitalize() + state = string.capwords(' '.join(message.split()[1:])) except IndexError: state = 'California' - return sum_numbers(state) \ No newline at end of file + return sum_numbers(state)