Update bot.py and added events.py

This commit is contained in:
Yuuki 2024-12-25 20:01:36 +09:00
parent c34291a001
commit 81b740ea95
2 changed files with 45 additions and 1 deletions

6
bot.py
View file

@ -2,6 +2,7 @@ import os
import disnake
from disnake.ext import commands
from disnake.ext.commands import ExtensionAlreadyLoaded
from dotenv import load_dotenv
from logger import Logger
@ -26,7 +27,10 @@ if __name__ == '__main__':
@bot.event
async def on_ready():
pass
try:
bot.load_extension('cogs.events')
except ExtensionAlreadyLoaded as eal:
pass
@bot.event

40
cogs/events.py Normal file
View file

@ -0,0 +1,40 @@
import disnake
from disnake.ext import commands
import os
from logger import Logger
logger = Logger('Liza Events')
class EventsCog(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.JOIN_MESSAGE = os.getenv('JOIN_MESSAGE')
self.LEAVE_MESSAGE = os.getenv('LEAVE_MESSAGE')
self.CHANNEL = os.getenv('WELCOME_LEAVE_CHANNEL_ID')
@commands.Cog.listener()
async def on_member_join(self, member: disnake.Member):
c: disnake.guild = self.bot.get_channel(self.CHANNEL)
await c.send_message(self.JOIN_MESSAGE.format(member.mention))
@commands.Cog.listener()
async def on_member_update(self, before: disnake.Member, after: disnake.Member):
pass
@commands.Cog.listener()
async def on_member_remove(self, member: disnake.Member):
c: disnake.guild = self.bot.get_channel(self.CHANNEL)
await c.send_message(self.LEAVE_MESSAGE.format(member.mention))
@commands.Cog.listener()
async def on_member_ban(self, guild: disnake.Guild, member: disnake.Member or disnake.User):
pass
def setup(bot):
bot.add_cog(EventsCog(bot))