Now acepts states with spaces in the name

This commit is contained in:
luke 2020-03-26 09:38:42 -07:00
parent db94e7b73c
commit e41001538f

View File

@ -3,6 +3,7 @@ import pandas as pd
import requests import requests
import os import os
import discord import discord
import string
def get_csv(): def get_csv():
git_url = 'https://api.github.com/repos/CSSEGISandData/COVID-19/contents/csse_covid_19_data/csse_covid_19_daily_reports' 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): def parse_message(message):
try: try:
state = message.split()[1].capitalize() state = string.capwords(' '.join(message.split()[1:]))
except IndexError: except IndexError:
state = 'California' state = 'California'
return sum_numbers(state) return sum_numbers(state)