Simple default role add on join

This commit is contained in:
Ethanell 2019-08-10 13:09:33 +02:00
parent 0e5297d02f
commit dc7131cbae

70
bot.py
View file

@ -1,10 +1,70 @@
import discord
import shelve
from discord.ext import commands
bot = commands.Bot(command_prefix="%")
class Bot(discord.Client):
async def on_ready(self):
print("FurrMula Bot online !")
def set_config(guild, conf):
print(f"Set config for {guild.name}")
conf[str(guild.id)] = dict()
for opt in ["default_role_id"]:
conf[str(guild.id)][opt] = ""
@bot.event
async def on_ready():
with shelve.open("config.conf", writeback=True) as conf:
for g in bot.guilds:
if str(g.id) not in conf:
set_config(g, conf)
print("FurrMula Bot online !")
@bot.event
async def on_guild_join(guild):
with shelve.open("config.conf", writeback=True) as conf:
set_config(guild, conf)
@bot.event
async def on_member_join(member):
print(1)
with shelve.open("config.conf") as conf:
print(2)
print(conf[str(member.guild.id)]["default_role_id"])
if "default_role_id" in conf[str(member.guild.id)] and conf[str(member.guild.id)]["default_role_id"]:
print(3)
role = member.guild.get_role(conf[str(member.guild.id)]["default_role_id"])
print(4)
await member.add_roles(role)
print(5)
@bot.command()
async def shutdown(ctx):
await ctx.send("Shutdown !")
await bot.logout()
await bot.close()
@bot.command()
async def set_default_role(ctx):
with shelve.open("config.conf", writeback=True) as conf:
if len(ctx.message.role_mentions) == 1:
conf[str(ctx.guild.id)]["default_role_id"] = ctx.message.role_mentions[0].id
await ctx.send(f"Role {ctx.message.role_mentions[0].name} set to default role")
elif len(ctx.message.role_mentions) == 0:
conf[str(ctx.guild.id)]["default_role_id"] = ""
await ctx.send("Default role disabled")
else:
await ctx.send("Too many roles !")
@bot.command()
async def default_role(ctx):
with shelve.open("config.conf") as conf:
await ctx.send(conf[str(ctx.guild.id)]["default_role_id"])
bot = Bot()
bot.run(open("token.ini").read())