This commit is contained in:
2025-03-12 22:00:40 +01:00
parent 9156262a5e
commit f0e8573348

View File

@ -1,10 +1,17 @@
from telegram import Bot
import asyncio
from sqlalchemy.orm import Session
from database import get_db
from models import NewsItem
from datetime import datetime, timedelta
from datetime import datetime
import asyncio
import logging
LOG_FILE = "bot.log"
logging.basicConfig(
filename=LOG_FILE, # Archivo de logs
level=logging.INFO, # Nivel de logging (DEBUG, INFO, WARNING, ERROR, CRITICAL)
format="%(asctime)s - %(levelname)s - %(message)s", # Formato de los logs
)
# Configura tu bot con el token de BotFather
TOKEN = "7626026035:AAHEMp_iIN3y8AwywL0R6OTQvNi7EcJZ0iY"
@ -13,18 +20,20 @@ CHAT_ID = "-4731993289" # Reemplaza con el ID del grupo
bot = Bot(token=TOKEN)
async def enviar_mensaje(mensaje: str):
await bot.send_message(chat_id=CHAT_ID, text=mensaje)
try:
await bot.send_message(chat_id=CHAT_ID, text=mensaje)
logging.info("Mensaje enviado: %s", mensaje)
except Exception as e:
logging.error("Error al enviar mensaje: %s", e)
def obtener_titulares_por_keyword():
"""
Consulta la base de datos para obtener titulares agrupados por keywords del día anterior.
Consulta la base de datos para obtener titulares agrupados por keywords del día actual.
"""
db: Session = next(get_db()) # Obtener la sesión de la DB
try:
# Calcular el rango de fechas (ayer)
hoy = datetime.now()
# ayer = hoy - timedelta(days=1)
# Obtener todas las keywords disponibles
keywords = (
db.query(NewsItem.keyword)
@ -42,16 +51,16 @@ def obtener_titulares_por_keyword():
.filter(
NewsItem.keyword == keyword,
NewsItem.fecha == hoy.date(),
# NewsItem.fecha < hoy.date()
)
.all()
)
resumen[keyword] = [r[0] for r in resultados] # Convertir en lista de strings
logging.info("Titulares obtenidos: %s", resumen)
return resumen
except Exception as e:
print(f"Error al obtener titulares: {e}")
logging.error("Error al obtener titulares: %s", e)
return {}
finally:
@ -64,9 +73,9 @@ def enviar_resumen_diario():
resumen = obtener_titulares_por_keyword()
if not resumen:
mensaje = "¡Hola! No se encontraron noticias relevantes ayer. ¡Mañana volvemos con más información!"
mensaje = "¡Hola! No se encontraron noticias relevantes hoy. ¡Mañana volvemos con más información!"
else:
mensaje = "¡Hola! Estos son los titulares más relevantes de las noticias de ayer:\n\n"
mensaje = "¡Hola! Estos son los titulares más relevantes de las noticias de hoy:\n\n"
for keyword, titulares in resumen.items():
mensaje += f"🔹 *{keyword}*:\n"
for titulo in titulares:
@ -75,4 +84,4 @@ def enviar_resumen_diario():
mensaje += "📢 ¡Mañana volveremos con más información!"
# Enviar mensaje de forma asíncrona
asyncio.run(enviar_mensaje(mensaje))
asyncio.create_task(enviar_mensaje(mensaje))