Skip to content

Commit

Permalink
Merge branch 'develop' into main
Browse files Browse the repository at this point in the history
  • Loading branch information
Aitor Magán committed Aug 28, 2021
2 parents 6f530b2 + 11e4b47 commit 48ac87c
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 1,189 deletions.
Empty file added notify_young_people.py
View file
Empty file.
249 changes: 19 additions & 230 deletions src/message_handler.py
View file
Original file line number Diff line number Diff line change
@@ -1,13 +1,8 @@
import re
import os
from datetime import datetime, timedelta
from collections import defaultdict
from datetime import datetime
from aws_lambda_powertools import Logger
import requests
from src import telegram_helpers
from src import db
from func_timeout import func_set_timeout
from func_timeout.exceptions import FunctionTimedOut


UPDATE_CENTRES_TIME = int(os.environ.get("UPDATE_CENTRES_TIME", 300))
Expand All @@ -18,28 +13,22 @@ def handle_update(update):
message = update.get("message", {}).get("text", "")
user_info = update.get("message", {}).get("from", {})
user_id = user_info.get("id")
name = user_info.get("first_name", "")

if user_id:
try:
if message in ["/start", "/help"]:
answer = handle_start(update)
elif message == "/cancel":
answer = handle_cancel(update)
elif message == "/status":
answer = handle_status(update)
elif message == "/currentage":
answer = handle_current_age(update)
elif message == "/subscribe":
answer = handle_subscribe(update)
elif message == "/mindate":
# telegram_helpers.send_text(user_id, "⌛ Esto me puede llevar unos segunditos...")
answer = "Perdoname 🙏, pero la Comunidad de Madrid ha cambiado el sistema para impedirme darte esta información 😓."
else:
answer = handle_generic_message(update)
except Exception:
logger.exception("Unexpected error")
answer = "Perdoname 🙏, pero ha ocurrido un error inesperado 🤷 que me impide responder a tu solicitud. " \
"¿Podrías volver a intentarlo pasados unos minutos? Gracias 😃"
answer = "¡Ahora puedes vacunarte sin cita previa 🎉! Aquí tienes la lista de centros donde puedes " \
"hacerlo:\n\n➡️ *Wizink Center*: 24h\n➡️ *Wanda Metropolitano*: de 9.30 a 14:30 y de 15:30 " \
"a 20:30 (salvo días de partido, el anterior y el posterior)\n➡️ *Hospital Enfermera Isabel " \
"Zendal*: 24h\n➡️ [Puntos Centralizados de Vacunación](https://shorturl.at/itBER): de " \
"9.30 a 18.00\n\n¡No esperes más, vacúnate 💉 ya!"

if message in ["/start", "/help"]:
answer = f"¡Hola {name}! Bienvenidx al sistema de notificación de vacunación.\n\n{answer}"
elif message in ["/cancel"]:
answer = "Si quieres borrar ❌ tu suscripción sólo tienes que detener el bot. Para ello, accede al perfil " \
"y haz click en *Detener bot*."
elif message == "/currentage":
answer = handle_current_age(update)

update["answer"] = answer
logger.info(update)
Expand All @@ -52,212 +41,12 @@ def handle_update(update):
db.delete_notification(user_id)


def handle_start(update):
user_info = update.get("message", {}).get("from", {})
name = user_info.get("first_name", "")
return f"¡Hola {name}! Bienvenidx al sistema de notificación de vacunación.\n\n👉 ¡Si quieres que te avise 🔔 " \
f"cuando puedas pedir cita para vacunarte 💉 en la Comunidad de Madrid, simplemente indicame " \
f"tu año de nacimiento!\n\nOtros comandos útiles:\n- /subscribe: 🔔 Crea una suscripción para " \
f"cuando puedas pedir cita para vacunarte\n- /help: 🙋 Muestra esta ayuda\n- /status: " \
f"ℹ️ Muestra si ya estás suscrito\n- /cancel: 🔕 Cancela la notificación registrada\n - /currentage: " \
f"📆 Muestra la edad mínima con la que puedes pedir cita"


def handle_cancel(update):
user_info = update.get("message", {}).get("from", {})
user_id = user_info.get("id")
db.delete_notification(user_id)
return f"¡Vale {user_info.get('first_name')}! He borrado ❌ tus datos y ya no te notificaré. Si quieres volver " \
f"a activar la suscripción, simplemente di /subscribe"


def handle_status(update):
user_info = update.get("message", {}).get("from", {})
user_id = user_info.get("id")
user_notification = db.get_user_notification(user_id)
if user_notification:
age = user_notification["age"]
if not user_notification["notified"]:
year_of_birth = datetime.now().year - age
message = "¡Todo listo 👏! Ya tienes activas las notificaciones 🔔 para cuando el sistema de autocita " \
f"permita pedir cita a personas nacidas en {year_of_birth} o antes. Si quieres cancelarla, " \
f"simplemente escribe /cancel."
else:
message = "¡Vaya 🤔! Parece que ya te he notificado de que puedes pedir cita para vacunarte. " \
"Si quieres puedes crear otra suscripción: simplemente, dime la edad que tienes " \
"o tu año de nacimiento. ¡Estaré encantado de volver a notificarte! 📳"
else:
message = "Actualmente no tienes ninguna notificación registrada 😓. Si quieres que te notifique 🔔 cuando " \
"puedas pedir cita para vacunarte simplemente dime tu año de nacimiento o tu edad."

return message


def handle_current_age(update):
user_info = update.get("message", {}).get("from", {})
user_id = user_info.get("id")
def handle_current_age(_):
min_years = db.get_min_years()
max_year_of_birth = datetime.now().year - min_years
message = f"El sistema de autocita permite pedir cita a personas nacidas en {max_year_of_birth} o antes. " \
f"Si cumples con este criterio, no esperes más y ve a " \
f"🔗 https://autocitavacuna.sanidadmadrid.org/ohcitacovid para pedir tu cita."

user_notification = db.get_user_notification(user_id)
if not user_notification:
message += "\n\n⚠️ Puedo notificarte 🔔 cuando el sistema de autocitación permita vacunar a gente con tu " \
"edad. Simplemente dime tu año de nacimiento."

return message


def handle_subscribe(update):
user_info = update.get("message", {}).get("from", {})
user_name = user_info.get('first_name')
message = f"¡👌 Vale {user_name}! ¿Me dices tu año de nacimiento?"
f"¡Si cumples con este criterio, no esperes más vacúnate! Ahora puedes hacerlo sin cita 🏃: di " \
f"/mindate para obtener más información. También puedes pedir cita 🕘 en " \
f"🔗 https://autocitavacuna.sanidadmadrid.org/ohcitacovid"

return message


def handle_generic_message(update):
user_info = update.get("message", {}).get("from", {})
received_message = update.get("message", {}).get("text", "")
user_id = user_info.get("id")
user_name = user_info.get('first_name')
age = get_age(received_message)
today_year = datetime.now().year

if age is not None:
min_years = db.get_min_years()

if age >= min_years:
message = "‼️ ¡Ey! Parece que el sistema ya te permite pedir cita. Hazlo ya en 🔗 " \
"https://autocitavacuna.sanidadmadrid.org/ohcitacovid/"
elif age < 12:
message = "Lo siento 🥺, pero no hay vacunas 💉 aprobadas para menores de 12 años y por tanto no parece " \
"que el sistema te vaya a permitir en algún momento solicitar cita. Aún así, siempre puedes " \
"consultar /currentage para obtener la edad mínima con la que el sistema de autocita permite " \
"pedir citas."
else:
user_notification = db.get_user_notification(user_id)
db.save_notification(user_id, user_name, age)
year_of_birth = today_year - age
message = f"¡Genial {user_name} ✅! Te notificaré 🔔 en cuando el sistema de autocitación " \
f"de la Comunidad de Madrid permita pedir cita a gente nacida en {year_of_birth}. Si quieres " \
f"cancelar la suscripción, simplemente escribe /cancel."

if user_notification and user_notification["age"] != age:
message += f"\n\n⚠️ Ya tenías una suscripción activa. La he reemplazado por ésta."
else:
message = "¡Vaya 🥺! Parece que no te he entendido. Para que te 🔔 notifique cuando puedas pedir cita en el " \
"sistema de autocita de la Comunidad de Madrid, simplemente dime tu año de nacimiento (ejemplo: 1991)"

return message


def get_age(user_input):
age = None

try:
age = int(user_input)
except ValueError:
regular_expressions = [r'\d+/\d+/(\d\d\d\d)', r'\d+-\d+-(\d\d\d\d)',
r'(\d\d\d\d)-\d+-\d+', r'(\d\d\d\d)/\d+/\d+',
r'(\d+)']

i = 0
while age is None and i < len(regular_expressions):
groups = re.findall(regular_expressions[i], user_input)
if groups:
age = int(groups[len(groups) - 1])

i += 1

if age is not None and age >= 1900:
age = datetime.now().year - age

return age if age is not None and 0 <= age <= 120 else None


def handle_min_date(_):

centres_by_date, last_update = db.get_min_date_info()

if last_update is None or (datetime.now() - last_update).seconds >= UPDATE_CENTRES_TIME:
try:
centres_by_date, last_update = update_centres()
except (FunctionTimedOut, requests.exceptions.RequestException):
pass

if centres_by_date:
updated_ago_seconds = (datetime.now() - last_update).seconds
message = "¡Estupendo 😊! Aquí tienes las primeras fechas disponibles en el sistema de autocita:" \
if updated_ago_seconds < UPDATE_CENTRES_TIME else "Perdona 😔, pero me está costando un poquito contactar " \
"con el servicio de autocita de la Comunidad de " \
"Madrid. Aquí tienes la última información que puede " \
"extraer:"
message += "\n\n"
for date in sorted(centres_by_date.keys()):
date_str = date.strftime("%d/%m/%Y")
centres = "\n".join(map(lambda x: f"- {x}", centres_by_date[date]))
message += f"*{date_str}*:\n{centres}\n\n"

updated_ago = int(round(updated_ago_seconds / 60))
updated_at_msg = f"Actualizado hace {updated_ago} minutos"
updated_at_msg = updated_at_msg[:-1] if updated_ago == 1 else updated_at_msg
message += updated_at_msg
else:
message = "No he sido capaz de encontrar citas disponibles. Pruébalo de nuevo más tarde."

return message


@func_set_timeout(22)
def update_centres():
centres_by_date = defaultdict(lambda: list())
centres = requests.post("https://autocitavacuna.sanidadmadrid.org/ohcitacovid/autocita/obtenerCentros",
json={"edad_paciente": 45}, verify=False).json()

for centre in centres:
centre_min_date = get_centre_min_date(centre)
if centre_min_date:
centres_by_date[centre_min_date].append(centre['descripcion'].replace("_", "-"))

last_update = datetime.now()
db.save_min_date_info(centres_by_date, last_update)

return centres_by_date, last_update


def get_centre_min_date(centre):
data = requests.post(
"https://autocitavacuna.sanidadmadrid.org/ohcitacovid/autocita/obtenerHuecosMes",
json=get_spots_body(centre["idCentro"], centre["idPrestacion"], centre["agendas"]),
verify=False).json()

if type(data) != list or len(data) == 0:
data = requests.post(
"https://autocitavacuna.sanidadmadrid.org/ohcitacovid/autocita/obtenerHuecosMes",
json=get_spots_body(centre["idCentro"], centre["idPrestacion"], centre["agendas"],
month_modifier=1),
verify=False).json()

data = data if type(data) == list else []
dates = [x.get("fecha") for x in data]
dates = [datetime.strptime(x, "%d-%m-%Y") for x in dates]
return min(dates) if dates else None


def get_spots_body(id_centre, id_prestacion, agendas, month_modifier=0):
today = datetime.now()
check_date = datetime(year=today.year, month=today.month, day=1) + timedelta(days=31 * month_modifier)

return {
"idPaciente": "1",
"idPrestacion": id_prestacion,
"agendas": agendas,
"idCentro": id_centre,
"mes": check_date.month,
"anyo": check_date.year,
"horaInicio": "00:00",
"horaFin": "23:59"
}
Loading

0 comments on commit 48ac87c

Please sign in to comment.