mirror of
https://github.com/skylanix/MamieHenriette.git
synced 2026-02-06 06:40:35 +01:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
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
|
|
|