marcogll/telegram_new_socias
1
0
Fork 0
mirror of https://github.com/marcogll/telegram_new_socias.git synced 2026-01-13 13:15:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Implementar finalización del flujo /horario

Browse Source 
Este commit introduce la lógica para procesar y guardar los datos recopilados por el flujo de conversación .

Cambios:

1.  **Nueva tabla de base de datos**: Se ha añadido una nueva tabla  a la base de datos  para almacenar los horarios de trabajo de los usuarios.
2.  **Nuevo modelo SQLAlchemy**: Se ha creado el modelo  en .
3.  **Nuevo **: Se ha creado un nuevo módulo para centralizar la lógica de finalización de los flujos de conversación.
    *   La función  determina qué acción tomar en función del flujo que ha finalizado.
    *   La función  se encarga de:
        *   Enviar los datos del horario al .
        *   Guardar (o actualizar si ya existe) el horario en la nueva tabla .
4.  **Actualización de **: El constructor de flujos ahora llama a  cuando una conversación llega a su fin, conectando la lógica de conversación con la de procesamiento de datos.
This commit is contained in:
Marco Gallegos
2025-12-20 10:31:17 -06:00
parent 60dddc147b
commit 338108d7b7
4 changed files with 212 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
modules/flow_builder.py
View File

@@ -11,72 +11,62 @@ from telegram.ext import (
filters,
)
# Assuming finalization logic will be handled elsewhere for now
# from .onboarding import finalizar, cancelar
from .finalizer import finalize_flow
# A simple end state for now
async def end(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
await update.message.reply_text("Flow ended.")
async def end_cancel(update: Update, context: ContextTypes.DEFAULT_TYPE) -> int:
await update.message.reply_text("Flujo cancelado.")
return ConversationHandler.END
async def generic_callback(update: Update, context: ContextTypes.DEFAULT_TYPE, flow: dict):
current_state_key = context.user_data.get("current_state", 0)
# Find the current step in the flow
current_step = next((step for step in flow["steps"] if step["state"] == current_state_key), None)
if not current_step:
await update.message.reply_text("Hubo un error en el flujo. Por favor, inicia de nuevo.")
return ConversationHandler.END
# Save the answer
user_answer = update.message.text
variable_name = current_step.get("variable")
if variable_name:
context.user_data[variable_name] = user_answer
# Determine the next state
next_state_key = None
if "next_steps" in current_step:
for condition in current_step["next_steps"]:
if condition["value"] == user_answer:
if condition.get("value") == user_answer:
next_state_key = condition["go_to"]
break
elif condition["value"] == "default":
elif condition.get("value") == "default":
next_state_key = condition["go_to"]
elif "next_step" in current_step:
next_state_key = current_step["next_step"]
if next_state_key is None:
# If no next step is defined, end the conversation
return await end(update, context)
return await end_cancel(update, context)
if next_state_key == -1:
await finalize_flow(update, context)
return ConversationHandler.END
# Find the next step
next_step = next((step for step in flow["steps"] if step["state"] == next_state_key), None)
if not next_step:
# If the next step is the end of the conversation
if next_state_key == -1:
# Here we would call the generic "finalizar" function
# For now, just end it
await update.message.reply_text("Has completado el flujo. ¡Gracias!")
# return await finalizar(update, context)
return ConversationHandler.END
else:
await update.message.reply_text("Error: No se encontró el siguiente paso del flujo.")
return ConversationHandler.END
await update.message.reply_text("Error: No se encontró el siguiente paso del flujo.")
return ConversationHandler.END
# Ask the next question
reply_markup = ReplyKeyboardRemove()
if next_step.get("type") == "keyboard" and "options" in next_step:
# Create a 2D array for the keyboard
options = next_step["options"]
keyboard = [options[i:i+2] for i in range(0, len(options), 2)]
reply_markup = ReplyKeyboardMarkup(
[next_step["options"][i:i+3] for i in range(0, len(next_step["options"]), 3)],
keyboard,
one_time_keyboard=True, resize_keyboard=True
)
await update.message.reply_text(next_step["question"], reply_markup=reply_markup)
# Update the current state
context.user_data["current_state"] = next_state_key
return next_state_key
@@ -85,14 +75,15 @@ async def start_flow(update: Update, context: ContextTypes.DEFAULT_TYPE, flow: d
context.user_data.clear()
context.user_data["flow_name"] = flow["flow_name"]
# Start with the first step
first_step = flow["steps"][0]
context.user_data["current_state"] = first_step["state"]
reply_markup = ReplyKeyboardRemove()
if first_step.get("type") == "keyboard" and "options" in first_step:
options = first_step["options"]
keyboard = [options[i:i+2] for i in range(0, len(options), 2)]
reply_markup = ReplyKeyboardMarkup(
[first_step["options"][i:i+3] for i in range(0, len(first_step["options"]), 3)],
keyboard,
one_time_keyboard=True, resize_keyboard=True
)
@@ -102,25 +93,35 @@ async def start_flow(update: Update, context: ContextTypes.DEFAULT_TYPE, flow: d
def create_handler(flow: dict):
states = {}
for step in flow["steps"]:
all_states = sorted(list(set([step["state"] for step in flow["steps"]])))
for state_key in all_states:
# Skip the end state
if state_key == -1:
continue
callback = partial(generic_callback, flow=flow)
states[step["state"]] = [MessageHandler(filters.TEXT & ~filters.COMMAND, callback)]
states[state_key] = [MessageHandler(filters.TEXT & ~filters.COMMAND, callback)]
# The entry point should be a command with the same name as the flow
entry_point = CommandHandler(flow["flow_name"], partial(start_flow, flow=flow))
return ConversationHandler(
entry_points=[entry_point],
states=states,
fallbacks=[CommandHandler("cancelar", end)], # Replace with generic cancel
fallbacks=[CommandHandler("cancelar", end_cancel)],
allow_reentry=True,
)
def load_flows():
flow_handlers = []
for filename in os.listdir("conv-flows"):
flow_dir = "conv-flows"
if not os.path.isdir(flow_dir):
logging.warning(f"Directory not found: {flow_dir}")
return flow_handlers
for filename in os.listdir(flow_dir):
if filename.endswith(".json"):
filepath = os.path.join("conv-flows", filename)
filepath = os.path.join(flow_dir, filename)
with open(filepath, "r", encoding="utf-8") as f:
try:
flow_definition = json.load(f)