Compare commits
2 Commits
c954fff012
...
91aab39b5f
Author | SHA1 | Date |
---|---|---|
Gabriele De Rosa | 91aab39b5f | |
Gabriele De Rosa | 6863ba9dc2 |
16
bot.py
16
bot.py
|
@ -65,17 +65,29 @@ db = client['bot']
|
||||||
# Function to get data
|
# Function to get data
|
||||||
def download():
|
def download():
|
||||||
|
|
||||||
|
# Download from open data
|
||||||
r = requests.get(DATA_URL)
|
r = requests.get(DATA_URL)
|
||||||
|
# Create dataframe from data
|
||||||
df = pd.read_csv(
|
df = pd.read_csv(
|
||||||
io.StringIO(r.text),
|
io.StringIO(r.text),
|
||||||
index_col="data_somministrazione",
|
index_col="data_somministrazione",
|
||||||
)
|
)
|
||||||
|
# Set date as index
|
||||||
df.index = pd.to_datetime(
|
df.index = pd.to_datetime(
|
||||||
df.index,
|
df.index,
|
||||||
format="%Y-%m-%d",
|
format="%Y-%m-%d",
|
||||||
)
|
)
|
||||||
df = df.loc[df["area"] == "ITA"]
|
# Sort value based on date
|
||||||
|
df.sort_values('data_somministrazione')
|
||||||
|
# Delete sum data if already exists
|
||||||
|
df = df[df["area"] != "ITA"]
|
||||||
|
# Set target counter to numeric
|
||||||
df["totale"] = pd.to_numeric(df["totale"])
|
df["totale"] = pd.to_numeric(df["totale"])
|
||||||
|
# Group by day and sum counters
|
||||||
|
df = df.groupby(['data_somministrazione'])['totale'].sum().reset_index()
|
||||||
|
# Re-set date as ID in new dataframe
|
||||||
|
df = df.set_index('data_somministrazione')
|
||||||
|
# If there are current day data...
|
||||||
if dt.now() - df.index[-1] < td(days=1):
|
if dt.now() - df.index[-1] < td(days=1):
|
||||||
df = df[:-1] # Ignore the current day because it's often incomplete
|
df = df[:-1] # Ignore the current day because it's often incomplete
|
||||||
|
|
||||||
|
@ -148,7 +160,7 @@ def help(update, context):
|
||||||
help_msg = "_Lista dei comandi_ \n\n"
|
help_msg = "_Lista dei comandi_ \n\n"
|
||||||
help_msg += "/start - Avvia il bot \n"
|
help_msg += "/start - Avvia il bot \n"
|
||||||
help_msg += "/get - Ricevi i dati aggiornati \n"
|
help_msg += "/get - Ricevi i dati aggiornati \n"
|
||||||
help_msg += "/news `ON`/`OFF` - Attiva o Disattiva le notifiche giornaliere \n"
|
help_msg += "/news ON/OFF - Attiva o Disattiva le notifiche giornaliere \n"
|
||||||
help_msg += "/help - Messaggio di aiuto \n"
|
help_msg += "/help - Messaggio di aiuto \n"
|
||||||
help_msg += "/info - Informazioni su questo bot \n"
|
help_msg += "/info - Informazioni su questo bot \n"
|
||||||
help_msg += "/stop - Disattiva il bot \n"
|
help_msg += "/stop - Disattiva il bot \n"
|
||||||
|
|
Loading…
Reference in New Issue