Kaynağa Gözat

adding function to get server guild id

baumsplitter41 3 gün önce
ebeveyn
işleme
b32eabb76a
1 değiştirilmiş dosya ile 9 ekleme ve 1 silme
  1. 9 1
      VPD_BOT/cogs/matesearch.py

+ 9 - 1
VPD_BOT/cogs/matesearch.py

@@ -3,6 +3,8 @@ from discord.ext import commands
 from discord.commands import Option
 from discord.commands import Option
 from discord.commands import slash_command
 from discord.commands import slash_command
 import configparser
 import configparser
+import os
+from dotenv import load_dotenv
 
 
 
 
 class Matesearch(commands.Cog):
 class Matesearch(commands.Cog):
@@ -25,6 +27,12 @@ class Matesearch(commands.Cog):
             configFilePath = r'config.cfg'
             configFilePath = r'config.cfg'
             config.read(configFilePath)
             config.read(configFilePath)
             return config
             return config
+        def _get_guild(self):
+            load_dotenv()
+            debug_guilds_up = []
+            server_token = os.getenv("SERVER").split(",")
+            for i in range(len(server_token)):
+                debug_guilds_up.append(int(server_token[i]))
         def _get_channel(self):
         def _get_channel(self):
             config = self._load_config()
             config = self._load_config()
             search_channel_id = int(config["Matesearch"]["search_channel_id"])
             search_channel_id = int(config["Matesearch"]["search_channel_id"])
@@ -35,7 +43,7 @@ class Matesearch(commands.Cog):
             return search_channel
             return search_channel
         def _get_role(self):
         def _get_role(self):
             config = self._load_config()
             config = self._load_config()
-            guild = discord.Guild
+            guild = self._get_guild()
             search_role_id = int(config["Matesearch"]["search_role_id"])
             search_role_id = int(config["Matesearch"]["search_role_id"])
             search_role = guild.get_role(search_role_id)
             search_role = guild.get_role(search_role_id)
             if search_role is None:
             if search_role is None: