Now acepts states with spaces in the name
This commit is contained in:
parent
a146c38da5
commit
c48f0c87cf
@ -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)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user