Ajout de la surveillance des nouvelles vidéos YouTube

youtubebot/youtube_alert.py

@@ -0,0 +1,100 @@
+import logging
+import feedparser
+from dataclasses import dataclass
+from datetime import datetime
+
+from database import db
+from database.models import YoutubeAlert
+from webapp import webapp
+
+logger = logging.getLogger('youtube-alert')
+logger.setLevel(logging.INFO)
+
+YOUTUBE_RSS_URL = "https://www.youtube.com/feeds/videos.xml?channel_id={channel_id}"
+
+
+@dataclass
+class YoutubeVideo:
+	video_id: str
+	title: str
+	author: str
+	link: str
+	published: datetime
+	thumbnail: str
+
+
+def _fetch_latest_video(channel_id: str) -> YoutubeVideo | None:
+	feed_url = YOUTUBE_RSS_URL.format(channel_id=channel_id)
+	try:
+		feed = feedparser.parse(feed_url)
+		if feed.entries:
+			entry = feed.entries[0]
+			video_id = entry.yt_videoid
+			return YoutubeVideo(
+				video_id=video_id,
+				title=entry.title,
+				author=entry.author,
+				link=entry.link,
+				published=datetime(*entry.published_parsed[:6]),
+				thumbnail=f"https://i.ytimg.com/vi/{video_id}/maxresdefault.jpg"
+			)
+	except Exception as e:
+		logger.error(f"Erreur lors de la récupération du flux RSS pour {channel_id}: {e}")
+	return None
+
+
+def _get_channel_name(channel_id: str) -> str | None:
+	feed_url = YOUTUBE_RSS_URL.format(channel_id=channel_id)
+	try:
+		feed = feedparser.parse(feed_url)
+		if feed.feed and hasattr(feed.feed, 'author'):
+			return feed.feed.author
+		if feed.entries:
+			return feed.entries[0].author
+	except Exception as e:
+		logger.error(f"Erreur lors de la récupération du nom de chaîne pour {channel_id}: {e}")
+	return None
+
+
+async def checkNewVideos(bot):
+	with webapp.app_context():
+		alerts: list[YoutubeAlert] = YoutubeAlert.query.filter_by(enable=True).all()
+		
+		for alert in alerts:
+			logger.info(f"Vérification de la chaîne : {alert.channel_name or alert.channel_id}")
+			video = _fetch_latest_video(alert.channel_id)
+			
+			if video:
+				if not alert.channel_name:
+					alert.channel_name = video.author
+				
+				if alert.last_video_id != video.video_id:
+					logger.info(f"Nouvelle vidéo détectée : {video.title}")
+					
+					if alert.last_video_id is not None:
+						await _notifyAlert(bot, alert, video)
+					
+					alert.last_video_id = video.video_id
+			else:
+				logger.warning(f"Impossible de récupérer les vidéos pour {alert.channel_id}")
+		
+		db.session.commit()
+
+
+async def _notifyAlert(bot, alert: YoutubeAlert, video: YoutubeVideo):
+	try:
+		message = alert.message.format(video)
+		logger.info(f"Message de notification : {message}")
+		bot.loop.create_task(_sendMessage(bot, alert.notify_channel, message))
+	except Exception as e:
+		logger.error(f"Erreur lors du formatage du message : {e}")
+
+
+async def _sendMessage(bot, channel: int, message: str):
+	logger.info(f"Envoi de notification : {message}")
+	channel_obj = bot.get_channel(channel)
+	if channel_obj:
+		await channel_obj.send(message)
+		logger.info("Notification envoyée")
+	else:
+		logger.error(f"Canal Discord non trouvé : {channel}")