From ba426c9f6eba4f54174da168adac85c0ea3e746e Mon Sep 17 00:00:00 2001
From: tropii <97747461+TropiiDev@users.noreply.github.com>
Date: Sat, 25 Nov 2023 09:58:05 -0500
Subject: [PATCH] Updated JoinRole

---
 commands/joinrole.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/commands/joinrole.py b/commands/joinrole.py
index 6b84049..bce6fa3 100644
--- a/commands/joinrole.py
+++ b/commands/joinrole.py
@@ -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)