Ajout de la surveillance des nouvelles vidéos YouTube

webapp/youtube_alert.py

@@ -0,0 +1,88 @@
+from flask import render_template, request, redirect, url_for, flash
+
+from webapp import webapp
+from database import db
+from database.models import YoutubeAlert
+from discordbot import bot
+from youtubebot.youtube_alert import _get_channel_name
+
+
+@webapp.route("/youtube-alert")
+def openYoutubeAlert():
+	alerts: list[YoutubeAlert] = YoutubeAlert.query.all()
+	channels = bot.getAllTextChannel()
+	for alert in alerts:
+		for channel in channels:
+			if alert.notify_channel == channel.id:
+				alert.notify_channel_name = channel.name
+	return render_template("youtube-alert.html", alerts=alerts, channels=channels)
+
+
+@webapp.route("/youtube-alert/add", methods=['POST'])
+def addYoutubeAlert():
+	channel_id = request.form.get('channel_id').strip()
+	
+	existing = YoutubeAlert.query.filter_by(channel_id=channel_id).first()
+	if existing:
+		flash("Cette chaîne YouTube est déjà surveillée.", "error")
+		return redirect(url_for("openYoutubeAlert"))
+	
+	channel_name = _get_channel_name(channel_id)
+	if not channel_name:
+		flash("Impossible de trouver cette chaîne YouTube. Vérifiez l'ID de la chaîne.", "error")
+		return redirect(url_for("openYoutubeAlert"))
+	
+	alert = YoutubeAlert(
+		enable=True,
+		channel_id=channel_id,
+		channel_name=channel_name,
+		notify_channel=request.form.get('notify_channel'),
+		message=request.form.get('message')
+	)
+	db.session.add(alert)
+	db.session.commit()
+	flash(f"Alerte ajoutée pour la chaîne {channel_name}.", "success")
+	return redirect(url_for("openYoutubeAlert"))
+
+
+@webapp.route("/youtube-alert/toggle/<int:id>")
+def toggleYoutubeAlert(id):
+	alert: YoutubeAlert = YoutubeAlert.query.get_or_404(id)
+	alert.enable = not alert.enable
+	db.session.commit()
+	return redirect(url_for("openYoutubeAlert"))
+
+
+@webapp.route("/youtube-alert/edit/<int:id>")
+def openEditYoutubeAlert(id):
+	alert = YoutubeAlert.query.get_or_404(id)
+	channels = bot.getAllTextChannel()
+	return render_template("youtube-alert.html", alert=alert, channels=channels)
+
+
+@webapp.route("/youtube-alert/edit/<int:id>", methods=['POST'])
+def submitEditYoutubeAlert(id):
+	alert: YoutubeAlert = YoutubeAlert.query.get_or_404(id)
+	new_channel_id = request.form.get('channel_id').strip()
+	
+	if new_channel_id != alert.channel_id:
+		channel_name = _get_channel_name(new_channel_id)
+		if not channel_name:
+			flash("Impossible de trouver cette chaîne YouTube. Vérifiez l'ID de la chaîne.", "error")
+			return redirect(url_for("openEditYoutubeAlert", id=id))
+		alert.channel_id = new_channel_id
+		alert.channel_name = channel_name
+		alert.last_video_id = None
+	
+	alert.notify_channel = request.form.get('notify_channel')
+	alert.message = request.form.get('message')
+	db.session.commit()
+	return redirect(url_for("openYoutubeAlert"))
+
+
+@webapp.route("/youtube-alert/del/<int:id>")
+def delYoutubeAlert(id):
+	alert = YoutubeAlert.query.get_or_404(id)
+	db.session.delete(alert)
+	db.session.commit()
+	return redirect(url_for("openYoutubeAlert"))