Alerte de live

twitchbot/live_alert.py

@@ -0,0 +1,36 @@
+from twitchAPI.twitch import Twitch
+from twitchAPI.object.api import Stream
+
+from database import db
+from database.models import LiveAlert
+from discordbot import bot
+from webapp import webapp
+
+
+async def checkOnlineStreamer(twitch: Twitch) :
+	with webapp.app_context() : 
+		alerts : list[LiveAlert] = LiveAlert.query.filter_by(enable = True).all()
+		streams = await _retreiveStreams(twitch, alerts)
+		for alert in alerts : 
+			stream = next((s for s in streams if s.user_login == alert.login), None)
+			if stream : 
+				if not alert.online :
+					await _notifyAlert(alert, stream)
+				alert.online = True
+			else :
+				alert.online = False
+		db.session.commit()
+
+async def _notifyAlert(alert : LiveAlert, stream : Stream):
+	message : str = alert.message.format(stream)
+	bot.loop.create_task(_sendMessage(alert.notify_channel, message))
+
+async def _sendMessage(channel : int, message : str) : 
+	await bot.get_channel(channel).send(message)
+
+async def _retreiveStreams(twitch: Twitch, alerts : list[LiveAlert]) -> list[Stream] : 
+	streams : list[Stream] = []
+	async for stream in twitch.get_streams(user_login = [alert.login for alert in alerts]):
+		streams.append(stream)
+	return streams
+