ldooks/dragon-bot
ldooks/dragon-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Now accepts a state. defaults to california

Browse Source
This commit is contained in:
luke 2020-03-26 09:22:54 -07:00
parent 2e02cb90d1
commit db94e7b73c
4 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -1,4 +1,4 @@
FROM python:3.6-buster AS builder FROM python:3.6-buster
RUN apt-get update && apt-get install -y gcc python-pip RUN apt-get update && apt-get install -y gcc python-pip
ADD app/requirements.txt /requirements.txt ADD app/requirements.txt /requirements.txt
RUN pip install -U pip && pip install -r requirements.txt RUN pip install -U pip && pip install -r requirements.txt

2
Dockerfile-test-env
View File

@ -1,4 +1,4 @@
FROM python:3.6-buster AS builder FROM python:3.6-buster
RUN apt-get update && apt-get install -y gcc python-pip RUN apt-get update && apt-get install -y gcc python-pip
ADD app/requirements.txt /requirements.txt ADD app/requirements.txt /requirements.txt
RUN pip install -U pip pylint && pip install -r requirements.txt RUN pip install -U pip pylint && pip install -r requirements.txt

19
app/corona.py
View File

@ -20,17 +20,24 @@ def get_csv():
return local_csv return local_csv
def sum_numbers(): def sum_numbers(state):
series = read_csv(get_csv(), header=0, parse_dates=[0], index_col=0, squeeze=True) series = read_csv(get_csv(), header=0, parse_dates=[0], index_col=0, squeeze=True)
california = series.loc[series['Province_State'] == 'California'].sum() sums = series.loc[series['Province_State'] == state].sum()
confirmed = california['Confirmed'] confirmed = sums['Confirmed']
deaths = california['Deaths'] deaths = sums['Deaths']
recovered = california['Recovered'] recovered = sums['Recovered']
embed = discord.Embed(description='Most recent Corona stats for California', color=0x428bca, type="rich") embed = discord.Embed(description="Most recent Corona stats for %s" % state, color=0x428bca, type="rich")
embed.set_author(name="CSSE at Johns Hopkins University", icon_url='https://avatars2.githubusercontent.com/u/60674295') embed.set_author(name="CSSE at Johns Hopkins University", icon_url='https://avatars2.githubusercontent.com/u/60674295')
embed.add_field(name='Confirmed Cases', value=confirmed) embed.add_field(name='Confirmed Cases', value=confirmed)
embed.add_field(name='Recovered Cases', value=recovered) embed.add_field(name='Recovered Cases', value=recovered)
embed.add_field(name='Deaths', value=deaths) embed.add_field(name='Deaths', value=deaths)
embed.add_field(name='Source', value='https://github.com/CSSEGISandData/COVID-19') embed.add_field(name='Source', value='https://github.com/CSSEGISandData/COVID-19')
return embed return embed
def parse_message(message):
try:
state = message.split()[1].capitalize()
except IndexError:
state = 'California'
return sum_numbers(state)

2
app/dragon-bot.py
View File

@ -162,7 +162,7 @@ async def on_message(message):
if message.content.startswith('!corona'): if message.content.startswith('!corona'):
await client.send_message( await client.send_message(
message.channel, message.channel,
embed=corona.sum_numbers() embed=corona.parse_message(message.content)
) )
if message.content.startswith('!decide'): if message.content.startswith('!decide'):