spawnrys/flanabot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add remembered user roles at rejoin

Browse Source
This commit is contained in:
AlberLC
2022-07-16 06:29:27 +02:00
parent 4ecf847686
commit 5a923b01c7
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
flanabot/bots/flana_disc_bot.py
View File

@@ -6,6 +6,7 @@ from typing import Sequence
import discord import discord
import flanautils import flanautils
from flanautils import NotFoundError
from multibot import BadRoleError, DiscordBot, User, constants as multibot_constants from multibot import BadRoleError, DiscordBot, User, constants as multibot_constants
from flanabot import constants from flanabot import constants
@@ -43,6 +44,8 @@ class FlanaDiscBot(DiscordBot, FlanaBot):
# ----------------------------------------------------------- # # ----------------------------------------------------------- #
def _add_handlers(self): def _add_handlers(self):
super()._add_handlers() super()._add_handlers()
self.client.add_listener(self._on_member_join, 'on_member_join')
self.client.add_listener(self._on_member_remove, 'on_member_remove')
self.client.add_listener(self._on_voice_state_update, 'on_voice_state_update') self.client.add_listener(self._on_voice_state_update, 'on_voice_state_update')
# noinspection PyTypeChecker # noinspection PyTypeChecker
@@ -88,6 +91,18 @@ class FlanaDiscBot(DiscordBot, FlanaBot):
# ---------------------------------------------- # # ---------------------------------------------- #
# HANDLERS # # HANDLERS #
# ---------------------------------------------- # # ---------------------------------------------- #
async def _on_member_join(self, member: discord.Member):
user = self._create_user_from_discord_user(member)
user.pull_from_database(overwrite_fields=('roles',))
for role in user.roles:
try:
await self.add_role(user, member.guild.id, role.id)
except NotFoundError:
pass
async def _on_member_remove(self, member: discord.Member):
self._create_user_from_discord_user(member).save()
async def _on_voice_state_update(self, _: discord.Member, before: discord.VoiceState, after: discord.VoiceState): async def _on_voice_state_update(self, _: discord.Member, before: discord.VoiceState, after: discord.VoiceState):
if getattr(before.channel, 'id', None) == HOT_CHANNEL_ID: if getattr(before.channel, 'id', None) == HOT_CHANNEL_ID:
channel = before.channel channel = before.channel