kurbezz/discord-bot
1
0
Fork 0
mirror of https://github.com/kurbezz/discord-bot.git synced 2025-12-06 15:15:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add lock

Browse Source
This commit is contained in:
Bulat Kurbanov
2024-11-15 13:16:07 +01:00
parent d68f6d6f3c
commit b5a64c02ce
1 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/services/twitch.py
View File

@@ -41,6 +41,8 @@ class TokenStorage:
class TwitchService: class TwitchService:
lock = Lock()
SCOPES = [ SCOPES = [
AuthScope.CHAT_READ, AuthScope.CHAT_READ,
AuthScope.CHAT_EDIT, AuthScope.CHAT_EDIT,
@@ -126,17 +128,18 @@ class TwitchService:
if stream is None: if stream is None:
return return
current_state = self.state.get(brodcaster_id) async with self.lock:
if current_state is None: current_state = self.state.get(brodcaster_id)
return if current_state is None:
return
changed = current_state.category != event.event.category_name changed = current_state.category != event.event.category_name
current_state.title = event.event.title current_state.title = event.event.title
current_state.category = event.event.category_name current_state.category = event.event.category_name
current_state.last_live_at = datetime.now() current_state.last_live_at = datetime.now()
self.state[brodcaster_id] = current_state self.state[brodcaster_id] = current_state
if changed: if changed:
await self.notify_change_category(brodcaster_id) await self.notify_change_category(brodcaster_id)
@@ -152,12 +155,12 @@ class TwitchService:
last_live_at=datetime.now() last_live_at=datetime.now()
) )
current_state = self.state.get(streamer_id) async with self.lock:
current_state = self.state.get(streamer_id)
is_need_notify = current_state is None or (datetime.now() - current_state.last_live_at).seconds >= self.ONLINE_NOTIFICATION_DELAY
is_need_notify = current_state is None or (datetime.now() - current_state.last_live_at).seconds >= self.ONLINE_NOTIFICATION_DELAY self.state[streamer_id] = state
self.state[streamer_id] = state
if is_need_notify: if is_need_notify:
await self.notify_online(streamer_id) await self.notify_online(streamer_id)