Update bot.py and added events.py
This commit is contained in:
parent
c34291a001
commit
81b740ea95
2 changed files with 45 additions and 1 deletions
4
bot.py
4
bot.py
|
@ -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,6 +27,9 @@ if __name__ == '__main__':
|
|||
|
||||
@bot.event
|
||||
async def on_ready():
|
||||
try:
|
||||
bot.load_extension('cogs.events')
|
||||
except ExtensionAlreadyLoaded as eal:
|
||||
pass
|
||||
|
||||
|
||||
|
|
40
cogs/events.py
Normal file
40
cogs/events.py
Normal 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))
|
Loading…
Add table
Reference in a new issue