90 lines
2.9 KiB
Python
90 lines
2.9 KiB
Python
import asyncio
|
|
import logging
|
|
|
|
from twitchAPI.twitch import Twitch
|
|
from twitchAPI.type import AuthScope, ChatEvent
|
|
from twitchAPI.chat import Chat, ChatEvent, ChatMessage, EventData
|
|
|
|
from database.helpers import ConfigurationHelper
|
|
from twitchbot.live_alert import checkOnlineStreamer
|
|
from twitchbot.announcements import checkAndSendAnnouncements, incrementMessageCount
|
|
from webapp import webapp
|
|
|
|
USER_SCOPE = [AuthScope.CHAT_READ, AuthScope.CHAT_EDIT]
|
|
|
|
|
|
async def _onReady(ready_event: EventData):
|
|
logging.info('Bot Twitch prêt')
|
|
channel = ConfigurationHelper().getValue('twitch_channel')
|
|
webapp.config["BOT_STATUS"]["twitch_connected"] = True
|
|
webapp.config["BOT_STATUS"]["twitch_channel_name"] = channel
|
|
with webapp.app_context():
|
|
await ready_event.chat.join_room(channel)
|
|
asyncio.get_event_loop().create_task(twitchBot._checkOnlineStreamers())
|
|
asyncio.get_event_loop().create_task(twitchBot._runAnnouncements())
|
|
|
|
|
|
async def _onMessage(msg: ChatMessage):
|
|
logging.info(f'Dans {msg.room.name}, {msg.user.name} a dit : {msg.text}')
|
|
incrementMessageCount()
|
|
|
|
|
|
async def _helloCommand(msg: ChatMessage):
|
|
await msg.reply(f'Bonjour {msg.user.name}')
|
|
|
|
|
|
def _isConfigured():
|
|
helper = ConfigurationHelper()
|
|
return (helper.getValue('twitch_client_id') is not None and
|
|
helper.getValue('twitch_client_secret') is not None and
|
|
helper.getValue('twitch_access_token') is not None and
|
|
helper.getValue('twitch_refresh_token') is not None and
|
|
helper.getValue('twitch_channel') is not None)
|
|
|
|
|
|
class TwitchBot():
|
|
|
|
async def _connect(self):
|
|
with webapp.app_context():
|
|
if _isConfigured():
|
|
try:
|
|
helper = ConfigurationHelper()
|
|
self.twitch = await Twitch(helper.getValue('twitch_client_id'), helper.getValue('twitch_client_secret'))
|
|
await self.twitch.set_user_authentication(helper.getValue('twitch_access_token'), USER_SCOPE, helper.getValue('twitch_refresh_token'))
|
|
self.chat = await Chat(self.twitch)
|
|
self.chat.register_event(ChatEvent.READY, _onReady)
|
|
self.chat.register_event(ChatEvent.MESSAGE, _onMessage)
|
|
self.chat.register_command('hello', _helloCommand)
|
|
self.chat.start()
|
|
except Exception as e:
|
|
logging.error(f'Échec de l\'authentification Twitch : {e}')
|
|
else:
|
|
logging.info("Twitch n'est pas configuré")
|
|
|
|
async def _checkOnlineStreamers(self):
|
|
while True:
|
|
try:
|
|
await checkOnlineStreamer(self.twitch)
|
|
except Exception as e:
|
|
logging.error(f'Erreur check streamers online : {e}')
|
|
await asyncio.sleep(5 * 60)
|
|
|
|
async def _runAnnouncements(self):
|
|
channel = ConfigurationHelper().getValue('twitch_channel')
|
|
while True:
|
|
try:
|
|
await checkAndSendAnnouncements(self.chat, channel)
|
|
except Exception as e:
|
|
logging.error(f'Erreur envoi annonces : {e}')
|
|
await asyncio.sleep(60)
|
|
|
|
def begin(self):
|
|
asyncio.run(self._connect())
|
|
|
|
async def _close(self):
|
|
self.chat.stop()
|
|
await self.twitch.close()
|
|
|
|
|
|
twitchBot = TwitchBot()
|