Merge pull request #16 from aitormagan/develop

Develop

src/message_handler.py

@@ -1,10 +1,10 @@
-import os
-from datetime import datetime
+import requests
 from aws_lambda_powertools import Logger
 from src import telegram_helpers
-from src import db
 
 
+URL_AUTOCITA_CONFIG = "https://autocitavacuna.sanidadmadrid.org/ohcitacovid/assets/config/app-config.json"
+YEAR_BOOSTER = 1961
 logger = Logger(service="vacunacovidmadridbot")
 
 
@@ -15,19 +15,24 @@ def handle_update(update):
     name = user_info.get("first_name", "")
 
     if user_id:
-        answer = "¡Ahora puedes vacunarte sin cita previa 🎉! Tienes más info ➡️ [aquí]" + \
-                 "(https://www.comunidad.madrid/servicios/salud/vacunacion-frente-coronavirus-comunidad-madrid)" + \
-                 "\n\n¡No esperes más, vacúnate 💉 ya!"
+        answers = ["🤔 ¿No te has vacunado aún? Ahora puedes vacunarte sin cita. Tienes más info ➡️ [aquí]" +
+                   "(https://www.comunidad.madrid/servicios/salud/vacunacion-frente-coronavirus-comunidad-madrid)",
+                   "Si por lo contrario quieres te notifique cuando puedas ponerte una dosis de recuerdo, "
+                   "aún no estoy preparado para eso 😔.\n👉 A día de hoy, sólo pueden conseguir su tercera dosis "
+                   f"los nacidos en {YEAR_BOOSTER} o antes."]
+
         if message in ["/start", "/help"]:
-            answer = f"¡Hola {name}! Bienvenidx al sistema de notificación de vacunación.\n\n{answer}"
+            answers = [f"¡Hola {name}! Bienvenidx al sistema de notificación de vacunación."] + answers
         elif message in ["/cancel"]:
-            answer = "Toda tu información personal ya ha sido eliminada del sistema. ¡Gracias por tu confianza!"
+            answers = ["Toda tu información personal ya ha sido eliminada del sistema. ¡Gracias por tu confianza!"]
         elif message == "/currentage":
-            answer = handle_current_age(update)
+            answers = handle_current_age(update)
 
-        update["answer"] = answer
+        update["answer"] = "\n".join(answers)
         logger.info(update)
-        telegram_helpers.send_text(user_id, answer)
+        for answer in answers:
+            telegram_helpers.send_text(user_id, answer)
+
         telegram_helpers.send_text(user_id, "🤔 ¿Vives en Madrid? ¿Usas mucho el metro? ¡Ahora tienes un nuevo "
                                             "🤖 bot disponible! Con "
                                             "[Metro Madrid - Tiempos de Espera](t.me/MetroMadridTiempoEsperaBot) "
@@ -42,11 +47,12 @@ def handle_update(update):
 
 
 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 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
+    max_year_of_birth = requests.get(URL_AUTOCITA_CONFIG).json()["dFin_Birthday"].split("/")[-1]
+    messages = [f"👉 Para 1️⃣ primeras citas, el sistema permite pedir cita a personas nacidas en {max_year_of_birth} o "
+                f"antes.",
+                f"👉 Para 3️⃣ terceras dosis, el sistema de autocita permite pedir cita a personas nacidas en "
+                f"{YEAR_BOOSTER} o antes",
+                f"¡Si cumples con algunos de estos criterios, no esperes más vacúnate! Puedes pedir cita 🕘 en " +
+                f"🔗 https://autocitavacuna.sanidadmadrid.org/ohcitacovid."]
+
+    return messages