añadida logica de botones
This commit is contained in:
eloraculodiario
parent
e2114d053c
commit
c03105a9d0
1 changed files with 389 additions and 7 deletions
396
bot.py
396
bot.py
|
|
@ -7,8 +7,21 @@ from dotenv import load_dotenv
|
||||||
from mastodon import Mastodon
|
from mastodon import Mastodon
|
||||||
import requests
|
import requests
|
||||||
|
|
||||||
from telegram import Update, Message, PhotoSize
|
from telegram import (
|
||||||
from telegram.ext import ApplicationBuilder, ContextTypes, MessageHandler, filters
|
Update,
|
||||||
|
Message,
|
||||||
|
PhotoSize,
|
||||||
|
InlineKeyboardButton,
|
||||||
|
InlineKeyboardMarkup,
|
||||||
|
)
|
||||||
|
from telegram.ext import (
|
||||||
|
ApplicationBuilder,
|
||||||
|
ContextTypes,
|
||||||
|
MessageHandler,
|
||||||
|
CommandHandler,
|
||||||
|
CallbackQueryHandler,
|
||||||
|
filters,
|
||||||
|
)
|
||||||
import logging
|
import logging
|
||||||
import sys
|
import sys
|
||||||
import re
|
import re
|
||||||
|
|
@ -54,6 +67,9 @@ class PostPayload:
|
||||||
image_paths: List[str] # rutas locales de imágenes descargadas
|
image_paths: List[str] # rutas locales de imágenes descargadas
|
||||||
|
|
||||||
|
|
||||||
|
# ----------------------------
|
||||||
|
# Ayuda /1212 (comportamiento existente)
|
||||||
|
# ----------------------------
|
||||||
def extract_command_payload(msg: Message) -> Optional[str]:
|
def extract_command_payload(msg: Message) -> Optional[str]:
|
||||||
"""
|
"""
|
||||||
Si el texto o el caption comienza con /1212, devuelve el resto (payload).
|
Si el texto o el caption comienza con /1212, devuelve el resto (payload).
|
||||||
|
|
@ -151,7 +167,361 @@ def post_to_instagram(payload: PostPayload):
|
||||||
|
|
||||||
|
|
||||||
# ----------------------------
|
# ----------------------------
|
||||||
# Handlers
|
# Borradores para /publicar
|
||||||
|
# ----------------------------
|
||||||
|
DRAFT_KEY = "current_draft"
|
||||||
|
DRAFT_STATE_KEY = "draft_state" # 'idle', 'waiting_title', 'waiting_description', 'waiting_hashtags', 'waiting_time', 'waiting_date'
|
||||||
|
|
||||||
|
|
||||||
|
def init_draft(
|
||||||
|
context: ContextTypes.DEFAULT_TYPE,
|
||||||
|
chat_id: int,
|
||||||
|
title: str = "",
|
||||||
|
description: str = "",
|
||||||
|
image_paths=None
|
||||||
|
):
|
||||||
|
if image_paths is None:
|
||||||
|
image_paths = []
|
||||||
|
context.user_data[DRAFT_KEY] = {
|
||||||
|
"chat_id": chat_id,
|
||||||
|
"title": title,
|
||||||
|
"description": description,
|
||||||
|
"hashtags": "",
|
||||||
|
"time_start": "",
|
||||||
|
"time_end": "",
|
||||||
|
"date": "",
|
||||||
|
"image_paths": image_paths,
|
||||||
|
"message_id": None, # id del mensaje de preview
|
||||||
|
}
|
||||||
|
context.user_data[DRAFT_STATE_KEY] = "idle"
|
||||||
|
|
||||||
|
|
||||||
|
def get_draft(context: ContextTypes.DEFAULT_TYPE) -> Optional[dict]:
|
||||||
|
return context.user_data.get(DRAFT_KEY)
|
||||||
|
|
||||||
|
|
||||||
|
def set_state(context: ContextTypes.DEFAULT_TYPE, state: str):
|
||||||
|
context.user_data[DRAFT_STATE_KEY] = state
|
||||||
|
|
||||||
|
|
||||||
|
def get_state(context: ContextTypes.DEFAULT_TYPE) -> str:
|
||||||
|
return context.user_data.get(DRAFT_STATE_KEY, "idle")
|
||||||
|
|
||||||
|
|
||||||
|
def build_draft_preview_text(draft: dict) -> str:
|
||||||
|
title = draft.get("title") or "—"
|
||||||
|
description = draft.get("description") or "—"
|
||||||
|
hashtags = draft.get("hashtags") or "—"
|
||||||
|
date = draft.get("date") or "—"
|
||||||
|
|
||||||
|
ts = draft.get("time_start") or ""
|
||||||
|
te = draft.get("time_end") or ""
|
||||||
|
if ts and te:
|
||||||
|
time_str = f"{ts} / {te}"
|
||||||
|
elif ts:
|
||||||
|
time_str = ts
|
||||||
|
else:
|
||||||
|
time_str = "—"
|
||||||
|
|
||||||
|
text = (
|
||||||
|
"✏️ *Borrador de publicación*\n\n"
|
||||||
|
f"📌 *Título*: {title}\n\n"
|
||||||
|
f"📝 *Descripción:*\n{description}\n\n"
|
||||||
|
f"🏷 Hashtags: {hashtags}\n"
|
||||||
|
f"⏰ Horario: {time_str}\n"
|
||||||
|
f"📅 Fecha: {date}"
|
||||||
|
)
|
||||||
|
return text
|
||||||
|
|
||||||
|
|
||||||
|
def build_draft_keyboard() -> InlineKeyboardMarkup:
|
||||||
|
keyboard = [
|
||||||
|
[
|
||||||
|
InlineKeyboardButton("Título", callback_data="draft:title"),
|
||||||
|
InlineKeyboardButton("Descripción", callback_data="draft:description"),
|
||||||
|
],
|
||||||
|
[
|
||||||
|
InlineKeyboardButton("Hashtag", callback_data="draft:hashtags"),
|
||||||
|
InlineKeyboardButton("Horario", callback_data="draft:time"),
|
||||||
|
InlineKeyboardButton("Fecha", callback_data="draft:date"),
|
||||||
|
],
|
||||||
|
[
|
||||||
|
InlineKeyboardButton("✅ Publicar", callback_data="draft:publish"),
|
||||||
|
InlineKeyboardButton("❌ Cancelar", callback_data="draft:cancel"),
|
||||||
|
]
|
||||||
|
]
|
||||||
|
return InlineKeyboardMarkup(keyboard)
|
||||||
|
|
||||||
|
|
||||||
|
async def refresh_draft_message(context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
"""
|
||||||
|
Borra el mensaje de preview anterior (si existe) y envía uno nuevo
|
||||||
|
con la preview + botones, para que siempre quede al final del chat.
|
||||||
|
"""
|
||||||
|
draft = get_draft(context)
|
||||||
|
if not draft:
|
||||||
|
return
|
||||||
|
|
||||||
|
chat_id = draft["chat_id"]
|
||||||
|
old_message_id = draft.get("message_id")
|
||||||
|
|
||||||
|
# Borramos el mensaje anterior del borrador si existe
|
||||||
|
if old_message_id:
|
||||||
|
try:
|
||||||
|
await context.bot.delete_message(chat_id=chat_id, message_id=old_message_id)
|
||||||
|
except Exception as e:
|
||||||
|
log.warning(f"No se pudo borrar el mensaje anterior del borrador: {e}")
|
||||||
|
|
||||||
|
# Enviamos un nuevo mensaje de preview con la botonera
|
||||||
|
sent = await context.bot.send_message(
|
||||||
|
chat_id=chat_id,
|
||||||
|
text=build_draft_preview_text(draft),
|
||||||
|
reply_markup=build_draft_keyboard(),
|
||||||
|
parse_mode="Markdown"
|
||||||
|
)
|
||||||
|
draft["message_id"] = sent.message_id
|
||||||
|
|
||||||
|
|
||||||
|
# ----------------------------
|
||||||
|
# Comando /publicar
|
||||||
|
# ----------------------------
|
||||||
|
async def start_publish(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
msg = update.message
|
||||||
|
if msg is None:
|
||||||
|
return
|
||||||
|
|
||||||
|
full_text = msg.text or ""
|
||||||
|
parts = full_text.split(" ", 1)
|
||||||
|
|
||||||
|
# Si el usuario pone texto tras /publicar, lo usamos como descripción inicial
|
||||||
|
base_description = ""
|
||||||
|
if len(parts) >= 2 and parts[1].strip():
|
||||||
|
base_description = parts[1].strip()
|
||||||
|
|
||||||
|
# Descargamos foto si la hay
|
||||||
|
image_paths: List[str] = []
|
||||||
|
if msg.photo:
|
||||||
|
image_paths = await download_photos(context, msg.photo)
|
||||||
|
|
||||||
|
# Iniciamos borrador con título vacío y descripción opcional
|
||||||
|
init_draft(context, msg.chat.id, title="", description=base_description, image_paths=image_paths)
|
||||||
|
|
||||||
|
# Mostramos el borrador con la botonera al final del chat
|
||||||
|
await refresh_draft_message(context)
|
||||||
|
|
||||||
|
|
||||||
|
# ----------------------------
|
||||||
|
# Callback de botones (Título / Descripción / Hashtag / Horario / Fecha / Publicar / Cancelar)
|
||||||
|
# ----------------------------
|
||||||
|
async def draft_keyboard_callback(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
query = update.callback_query
|
||||||
|
await query.answer()
|
||||||
|
|
||||||
|
data = query.data or ""
|
||||||
|
if not data.startswith("draft:"):
|
||||||
|
return
|
||||||
|
|
||||||
|
draft = get_draft(context)
|
||||||
|
if not draft:
|
||||||
|
try:
|
||||||
|
await query.message.edit_text("No hay borrador activo.")
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
return
|
||||||
|
|
||||||
|
action = data.split(":", 1)[1]
|
||||||
|
|
||||||
|
if action == "title":
|
||||||
|
set_state(context, "waiting_title")
|
||||||
|
await query.message.reply_text(
|
||||||
|
"Escribe ahora el *título* de la publicación.",
|
||||||
|
parse_mode="Markdown"
|
||||||
|
)
|
||||||
|
|
||||||
|
elif action == "description":
|
||||||
|
set_state(context, "waiting_description")
|
||||||
|
await query.message.reply_text(
|
||||||
|
"Escribe ahora la *descripción* de la publicación.",
|
||||||
|
parse_mode="Markdown"
|
||||||
|
)
|
||||||
|
|
||||||
|
elif action == "hashtags":
|
||||||
|
set_state(context, "waiting_hashtags")
|
||||||
|
await query.message.reply_text(
|
||||||
|
"Escribe ahora los *hashtags* en un mensaje (por ejemplo: `#feminismo #sostenibilidad`).",
|
||||||
|
parse_mode="Markdown"
|
||||||
|
)
|
||||||
|
|
||||||
|
elif action == "time":
|
||||||
|
set_state(context, "waiting_time")
|
||||||
|
await query.message.reply_text(
|
||||||
|
"Escribe ahora el *horario* en formato `HH:MM / HH:MM` (por ejemplo: `19:30 / 21:30`).",
|
||||||
|
parse_mode="Markdown"
|
||||||
|
)
|
||||||
|
|
||||||
|
elif action == "date":
|
||||||
|
set_state(context, "waiting_date")
|
||||||
|
await query.message.reply_text(
|
||||||
|
"Escribe ahora la *fecha* en formato `DD/MM/AAAA` (por ejemplo: `13/04/2025`).",
|
||||||
|
parse_mode="Markdown"
|
||||||
|
)
|
||||||
|
|
||||||
|
elif action == "publish":
|
||||||
|
await finalize_and_post_draft(update, context)
|
||||||
|
|
||||||
|
elif action == "cancel":
|
||||||
|
context.user_data.pop(DRAFT_KEY, None)
|
||||||
|
set_state(context, "idle")
|
||||||
|
try:
|
||||||
|
await query.message.edit_text("Publicación cancelada.")
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
# ----------------------------
|
||||||
|
# Manejo de los textos de título / descripción / hashtag / hora / fecha
|
||||||
|
# ----------------------------
|
||||||
|
async def handle_draft_text(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
msg = update.message
|
||||||
|
if msg is None or not msg.text:
|
||||||
|
return
|
||||||
|
|
||||||
|
state = get_state(context)
|
||||||
|
draft = get_draft(context)
|
||||||
|
if not draft:
|
||||||
|
return # no hay borrador activo, ignoramos
|
||||||
|
|
||||||
|
text = msg.text.strip()
|
||||||
|
|
||||||
|
# Título
|
||||||
|
if state == "waiting_title":
|
||||||
|
draft["title"] = text
|
||||||
|
set_state(context, "idle")
|
||||||
|
await refresh_draft_message(context)
|
||||||
|
return
|
||||||
|
|
||||||
|
# Descripción
|
||||||
|
if state == "waiting_description":
|
||||||
|
draft["description"] = text
|
||||||
|
set_state(context, "idle")
|
||||||
|
await refresh_draft_message(context)
|
||||||
|
return
|
||||||
|
|
||||||
|
# Hashtags
|
||||||
|
if state == "waiting_hashtags":
|
||||||
|
draft["hashtags"] = text
|
||||||
|
set_state(context, "idle")
|
||||||
|
await refresh_draft_message(context)
|
||||||
|
return
|
||||||
|
|
||||||
|
# Horario rango HH:MM / HH:MM
|
||||||
|
if state == "waiting_time":
|
||||||
|
# Validar formato tipo "19:30 / 21:30"
|
||||||
|
m = re.match(r"^\s*(\d{1,2}:\d{2})\s*/\s*(\d{1,2}:\d{2})\s*$", text)
|
||||||
|
if not m:
|
||||||
|
await msg.reply_text(
|
||||||
|
"Formato de horario no válido. Usa `HH:MM / HH:MM`, por ejemplo `19:30 / 21:30`.",
|
||||||
|
parse_mode="Markdown"
|
||||||
|
)
|
||||||
|
return
|
||||||
|
start, end = m.group(1), m.group(2)
|
||||||
|
draft["time_start"] = start
|
||||||
|
draft["time_end"] = end
|
||||||
|
set_state(context, "idle")
|
||||||
|
await refresh_draft_message(context)
|
||||||
|
return
|
||||||
|
|
||||||
|
# Fecha DD/MM/AAAA
|
||||||
|
if state == "waiting_date":
|
||||||
|
if not re.match(r"^\d{1,2}/\d{1,2}/\d{4}$", text):
|
||||||
|
await msg.reply_text(
|
||||||
|
"Formato de fecha no válido. Usa `DD/MM/AAAA`, por ejemplo `13/04/2025`.",
|
||||||
|
parse_mode="Markdown"
|
||||||
|
)
|
||||||
|
return
|
||||||
|
draft["date"] = text
|
||||||
|
set_state(context, "idle")
|
||||||
|
await refresh_draft_message(context)
|
||||||
|
return
|
||||||
|
|
||||||
|
# Si no estamos esperando nada especial, no hacemos nada
|
||||||
|
return
|
||||||
|
|
||||||
|
|
||||||
|
# ----------------------------
|
||||||
|
# Publicar borrador (/publicar) realmente
|
||||||
|
# ----------------------------
|
||||||
|
async def finalize_and_post_draft(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
|
query = update.callback_query
|
||||||
|
draft = get_draft(context)
|
||||||
|
if not draft:
|
||||||
|
try:
|
||||||
|
await query.message.edit_text("No hay borrador para publicar.")
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
return
|
||||||
|
|
||||||
|
title = draft.get("title") or ""
|
||||||
|
description = draft.get("description") or ""
|
||||||
|
base_text_parts = []
|
||||||
|
|
||||||
|
if title:
|
||||||
|
base_text_parts.append(title)
|
||||||
|
if description:
|
||||||
|
if title:
|
||||||
|
base_text_parts.append("") # línea en blanco entre título y descripción
|
||||||
|
base_text_parts.append(description)
|
||||||
|
|
||||||
|
base_text = "\n".join(base_text_parts) if base_text_parts else ""
|
||||||
|
|
||||||
|
extra_lines = []
|
||||||
|
|
||||||
|
if draft.get("date"):
|
||||||
|
extra_lines.append(f"📅 {draft['date']}")
|
||||||
|
|
||||||
|
ts = draft.get("time_start") or ""
|
||||||
|
te = draft.get("time_end") or ""
|
||||||
|
if ts and te:
|
||||||
|
extra_lines.append(f"⏰ {ts} / {te}")
|
||||||
|
elif ts:
|
||||||
|
extra_lines.append(f"⏰ {ts}")
|
||||||
|
|
||||||
|
if draft.get("hashtags"):
|
||||||
|
extra_lines.append(draft["hashtags"])
|
||||||
|
|
||||||
|
final_text = base_text
|
||||||
|
if extra_lines:
|
||||||
|
if final_text:
|
||||||
|
final_text += "\n\n" + "\n".join(extra_lines)
|
||||||
|
else:
|
||||||
|
final_text = "\n".join(extra_lines)
|
||||||
|
|
||||||
|
payload = PostPayload(
|
||||||
|
text=final_text,
|
||||||
|
image_paths=draft.get("image_paths", []),
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
post_to_mastodon(payload)
|
||||||
|
except Exception as e:
|
||||||
|
log.exception(f"[Mastodon] Error: {e}")
|
||||||
|
|
||||||
|
try:
|
||||||
|
post_to_instagram(payload)
|
||||||
|
except Exception as e:
|
||||||
|
log.exception(f"[Instagram] Error: {e}")
|
||||||
|
|
||||||
|
context.user_data.pop(DRAFT_KEY, None)
|
||||||
|
set_state(context, "idle")
|
||||||
|
|
||||||
|
try:
|
||||||
|
await query.message.edit_text("Publicado en redes ✅")
|
||||||
|
except Exception:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
# ----------------------------
|
||||||
|
# Handlers existentes /1212
|
||||||
# ----------------------------
|
# ----------------------------
|
||||||
async def handle_channel_post(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
async def handle_channel_post(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
||||||
"""Para posts en CANAL."""
|
"""Para posts en CANAL."""
|
||||||
|
|
@ -235,20 +605,32 @@ async def handle_group_message(update: Update, context: ContextTypes.DEFAULT_TYP
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
# ----------------------------
|
||||||
|
# main()
|
||||||
|
# ----------------------------
|
||||||
def main():
|
def main():
|
||||||
app = ApplicationBuilder().token(TELEGRAM_TOKEN).build()
|
app = ApplicationBuilder().token(TELEGRAM_TOKEN).build()
|
||||||
|
|
||||||
# CANALES
|
# NUEVO: comando /publicar + flujo de borrador
|
||||||
|
app.add_handler(CommandHandler("publicar", start_publish))
|
||||||
|
app.add_handler(CallbackQueryHandler(draft_keyboard_callback, pattern=r"^draft:"))
|
||||||
|
# Textos normales para título / descripción / hashtags / hora / fecha
|
||||||
|
app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_draft_text))
|
||||||
|
|
||||||
|
# EXISTENTE: CANALES con /1212
|
||||||
app.add_handler(MessageHandler(filters.ChatType.CHANNEL, handle_channel_post))
|
app.add_handler(MessageHandler(filters.ChatType.CHANNEL, handle_channel_post))
|
||||||
|
|
||||||
# GRUPOS/SUPERGRUPOS: sólo cuando el texto o el caption empieza por /1212
|
# EXISTENTE: GRUPOS/SUPERGRUPOS con /1212
|
||||||
group_filter = filters.ChatType.GROUPS & (filters.Regex(TRIGGER_RE) | filters.CaptionRegex(TRIGGER_RE))
|
group_filter = filters.ChatType.GROUPS & (filters.Regex(TRIGGER_RE) | filters.CaptionRegex(TRIGGER_RE))
|
||||||
app.add_handler(MessageHandler(group_filter, handle_group_message))
|
app.add_handler(MessageHandler(group_filter, handle_group_message))
|
||||||
|
|
||||||
log.info("Bot escuchando canal/grupos…")
|
log.info("Bot escuchando canal/grupos…")
|
||||||
# Escuchamos ambos tipos
|
app.run_polling(
|
||||||
app.run_polling(allowed_updates=["message", "channel_post"], drop_pending_updates=False)
|
allowed_updates=["message", "channel_post", "callback_query"],
|
||||||
|
drop_pending_updates=False
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue