tropii/blob
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Updated JoinRole

Browse Source
This commit is contained in:
tropii 2023-11-25 09:58:05 -05:00
parent ceb967eb33
commit ba426c9f6e
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
commands/joinrole.py
View File

@ -34,8 +34,14 @@ class joinrole(commands.Cog):
client = pymongo.MongoClient(os.getenv("mongo_url"))
db = client.servers
coll = db.roles
roles = coll.find_one({"_id": {"guild_id": ctx.guild.id}})
if not roles:
coll.insert_one({"_id": {"guild_id": ctx.guild.id}, "joinrole": role.id})
await ctx.send(f"Set the joinrole to {role.mention}!")
return
else:
coll.update_one({"_id": {"guild_id": ctx.guild.id}}, {"$set": {"joinrole": role.id}})
coll.update_one({"_id": {"guild_id": ctx.guild.id, "commands":"joinrole"}}, {"$set": {"joinrole": role.id}})
await ctx.send(f"Set the joinrole to {role.mention}!")
@commands.Cog.listener()
@ -44,7 +50,7 @@ class joinrole(commands.Cog):
db = client.servers
coll = db.roles
joinrole = coll.find_one({"_id": {"guild_id": member.guild.id, "commands":"joinrole"}})
joinrole = coll.find_one({"_id": {"guild_id": member.guild.id}})
if joinrole:
role = member.guild.get_role(joinrole["joinrole"])
await member.add_roles(role)