Added guild db management on start/join/kick guild

main
Anorak_1 2023-08-10 22:29:21 +02:00
parent f73a6612e2
commit 75ff784301
1 changed files with 12 additions and 0 deletions

12
bot.py
View File

@ -9,6 +9,7 @@ from discord.ui import Modal, InputText
from discord.utils import get from discord.utils import get
from discord.ext import commands from discord.ext import commands
from dbutil import MessageDB from dbutil import MessageDB
from dbutil import GuildAppDB
load_dotenv() load_dotenv()
@ -39,16 +40,27 @@ async def on_ready():
activity = discord.Activity(name=f"{len(bot.guilds)} guilds", type=discord.ActivityType.listening) activity = discord.Activity(name=f"{len(bot.guilds)} guilds", type=discord.ActivityType.listening)
await bot.change_presence(activity=activity, status = discord.Status.online) await bot.change_presence(activity=activity, status = discord.Status.online)
print(f"Logged in as {bot.user}") print(f"Logged in as {bot.user}")
for i in bot.guilds:
if str(i.id) not in GuildAppDB.get_all_guilds():
print(f"{i.id} is not in guilds")
GuildAppDB.create_guild(str(i.id), i.name)
print(f"entry for {i.id} created")
else:
print(f"{i.id} is in guilds")
print(GuildAppDB.get_all_guilds())
@bot.event @bot.event
async def on_guild_join(guild): async def on_guild_join(guild):
activity = discord.Activity(name=f"{len(bot.guilds)} guilds", type=discord.ActivityType.listening) activity = discord.Activity(name=f"{len(bot.guilds)} guilds", type=discord.ActivityType.listening)
await bot.change_presence(activity=activity, status = discord.Status.online) await bot.change_presence(activity=activity, status = discord.Status.online)
GuildAppDB.create_guild(str(guild.id), guild.name)
print(f"Joined guild {guild.name}: {guild.id}")
@bot.event @bot.event
async def on_guild_remove(guild): async def on_guild_remove(guild):
activity = discord.Activity(name=f"{len(bot.guilds)} guilds", type=discord.ActivityType.listening) activity = discord.Activity(name=f"{len(bot.guilds)} guilds", type=discord.ActivityType.listening)
await bot.change_presence(activity=activity, status = discord.Status.online) await bot.change_presence(activity=activity, status = discord.Status.online)
print(f"Removed from guild {guild.name}: {guild.id}")
@bot.slash_command(description = "Command used to apply") @bot.slash_command(description = "Command used to apply")
async def apply(ctx): async def apply(ctx):