mirror of
https://github.com/marcogll/talia_bot.git
synced 2026-01-13 13:25:19 +00:00
eb680edc54
- Implement `llm.py` with OpenAI API integration for smart responses. - Implement `scheduler.py` to send a daily summary to the bot owner using `python-telegram-bot`'s `JobQueue`. - Integrate the scheduler into the main application. - Add `pytz` as a new dependency. - Update `tasks.md` to mark all tasks as complete.
117 lines
3.9 KiB
Python
117 lines
3.9 KiB
Python
# app/main.py
|
|
import logging
|
|
from telegram import Update
|
|
from telegram.ext import (
|
|
Application,
|
|
CommandHandler,
|
|
CallbackQueryHandler,
|
|
ConversationHandler,
|
|
MessageHandler,
|
|
ContextTypes,
|
|
filters,
|
|
)
|
|
|
|
from config import TELEGRAM_BOT_TOKEN
|
|
from permissions import get_user_role
|
|
from modules.onboarding import handle_start as onboarding_handle_start
|
|
from modules.agenda import get_agenda
|
|
from modules.citas import request_appointment
|
|
from modules.equipo import (
|
|
propose_activity_start,
|
|
get_description,
|
|
get_duration,
|
|
cancel_proposal,
|
|
view_requests_status,
|
|
DESCRIPTION,
|
|
DURATION,
|
|
)
|
|
from modules.aprobaciones import view_pending, handle_approval_action
|
|
from modules.servicios import get_service_info
|
|
from modules.admin import get_system_status
|
|
from modules.print import print_handler
|
|
from app.scheduler import schedule_daily_summary
|
|
|
|
# Enable logging
|
|
logging.basicConfig(
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
"""Sends a welcome message and menu when the /start command is issued."""
|
|
chat_id = update.effective_chat.id
|
|
user_role = get_user_role(chat_id)
|
|
|
|
logger.info(f"User {chat_id} started conversation with role: {user_role}")
|
|
|
|
response_text, reply_markup = onboarding_handle_start(user_role)
|
|
|
|
await update.message.reply_text(response_text, reply_markup=reply_markup)
|
|
|
|
async def button_dispatcher(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
"""Parses the CallbackQuery and routes it to the appropriate handler."""
|
|
query = update.callback_query
|
|
await query.answer()
|
|
logger.info(f"Dispatcher received callback query: {query.data}")
|
|
|
|
# Default response if no handler is found
|
|
response_text = "Acción no reconocida."
|
|
reply_markup = None
|
|
|
|
# Simple callbacks that return a string
|
|
simple_handlers = {
|
|
'view_agenda': get_agenda,
|
|
'view_requests_status': view_requests_status,
|
|
'schedule_appointment': request_appointment,
|
|
'get_service_info': get_service_info,
|
|
'view_system_status': get_system_status,
|
|
'manage_users': lambda: "Función de gestión de usuarios no implementada.",
|
|
}
|
|
|
|
# Callbacks that return a tuple (text, reply_markup)
|
|
complex_handlers = {
|
|
'view_pending': view_pending,
|
|
}
|
|
|
|
if query.data in simple_handlers:
|
|
response_text = simple_handlers[query.data]()
|
|
elif query.data in complex_handlers:
|
|
response_text, reply_markup = complex_handlers[query.data]()
|
|
elif query.data.startswith(('approve:', 'reject:')):
|
|
response_text = handle_approval_action(query.data)
|
|
|
|
await query.edit_message_text(text=response_text, reply_markup=reply_markup, parse_mode='Markdown')
|
|
|
|
def main() -> None:
|
|
"""Start the bot."""
|
|
if not TELEGRAM_BOT_TOKEN:
|
|
logger.error("TELEGRAM_BOT_TOKEN is not set in the environment variables.")
|
|
return
|
|
|
|
application = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
|
|
|
|
# Schedule daily summary
|
|
schedule_daily_summary(application)
|
|
|
|
# Conversation handler for proposing activities
|
|
conv_handler = ConversationHandler(
|
|
entry_points=[CallbackQueryHandler(propose_activity_start, pattern='^propose_activity$')],
|
|
states={
|
|
DESCRIPTION: [MessageHandler(filters.TEXT & ~filters.COMMAND, get_description)],
|
|
DURATION: [MessageHandler(filters.TEXT & ~filters.COMMAND, get_duration)],
|
|
},
|
|
fallbacks=[CommandHandler('cancel', cancel_proposal)],
|
|
per_message=False
|
|
)
|
|
|
|
application.add_handler(conv_handler)
|
|
application.add_handler(CommandHandler("start", start))
|
|
application.add_handler(CommandHandler("print", print_handler))
|
|
application.add_handler(CallbackQueryHandler(button_dispatcher))
|
|
|
|
logger.info("Starting Talía Bot...")
|
|
application.run_polling()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|