config fixes

bugsbirb · Mar 13, 2024 · c86d1f0 · c86d1f0
1 parent 357b0c5
commit c86d1f0
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 14 deletions.
diff --git a/Cogs/Configuration/Views/loaview.py b/Cogs/Configuration/Views/loaview.py
@@ -81,12 +81,8 @@ async def callback(self, interaction: discord.Interaction):
             'staffrole': selected_role_id.id  
         }
         try:
-            existing_record = LOARole.find_one(filter)
 
-            if existing_record:
-                await LOARole.update_one(filter, {'$set': data})
-            else:
-                await LOARole.insert_one(data)
+            await LOARole.update_one(filter, {'$set': data}, upsert=True)
             await interaction.response.edit_message(content=None)
             await refreshembed(interaction)
         except Exception as e:

diff --git a/Cogs/Configuration/Views/partnershipsview.py b/Cogs/Configuration/Views/partnershipsview.py
@@ -42,7 +42,7 @@ async def callback(self, interaction: discord.Interaction):
         }
 
         try:
-            existing_record = partnershipsch.find_one(filter)
+            existing_record = await partnershipsch.find_one(filter)
 
             if existing_record:
                 await partnershipsch.update_one(filter, {'$set': data})

diff --git a/Cogs/Configuration/Views/promotionsview.py b/Cogs/Configuration/Views/promotionsview.py
@@ -43,13 +43,7 @@ async def callback(self, interaction: discord.Interaction):
         }
 
         try:
-            existing_record = promochannel.find_one(filter)
-
-            if existing_record:
-                await promochannel.update_one(filter, {'$set': data})
-            else:
-                await promochannel.insert_one(data)
-
+            await promochannel.update_one(filter, {'$set': data}, upsert=True)
             await refreshembed(interaction) 
             await interaction.response.edit_message(content=None)
         except Exception as e:

diff --git a/Cogs/Configuration/Views/reportview.py b/Cogs/Configuration/Views/reportview.py
@@ -45,7 +45,7 @@ async def callback(self, interaction: discord.Interaction):
         }
 
         try:
-            existing_record = repchannel.find_one(filter)
+            existing_record = await repchannel.find_one(filter)
 
             if existing_record:
                 repchannel.update_one(filter, {'$set': data})