multiple auto translate groups

translate/api.py

@@ -149,10 +149,7 @@ class GoogleTranslateAPI:
         self.cache["guild_whitelist"][guild.id] = await self.config.guild(guild).whitelist()
 
     async def check_bw_list(
-        self,
-        guild: discord.Guild,
-        channel: discord.TextChannel,
-        member: Union[discord.Member, discord.User],
+        self, guild: discord.Guild, channel: discord.TextChannel, member: Union[discord.Member, discord.User],
     ) -> bool:
         can_run = True
         if guild.id not in self.cache["guild_blacklist"]:
@@ -266,7 +263,9 @@ class GoogleTranslateAPI:
         author = cast(discord.Member, message.author)
         channel = cast(discord.TextChannel, message.channel)
         links = await self.config.guild(channel.guild).autosend()
-        link_channels = [int(r) for r in links.keys()]
+        link_channels = []
+        for l in links:
+            link_channels += [int(r) for r in l.keys()]  # for checks
         guild = message.guild
         if version_info >= VersionInfo.from_str("3.4.0"):
             if await self.bot.cog_disabled_in_guild(self, guild):
@@ -345,9 +344,12 @@ class GoogleTranslateAPI:
             return
         await self.translate_message(message, str(payload.emoji), reacted_user)
 
-    async def translate_automessage(self, message: discord.Message, links: dict) -> None:
+    async def translate_automessage(self, message: discord.Message, all_links: list) -> None:
         guild = cast(discord.Guild, message.guild)
         channel = cast(discord.TextChannel, message.channel)
+        for links in all_links:
+            if str(channel.id) not in links:
+                continue
             # remove sent channel from Links
             try:
                 del links[str(channel.id)]
@@ -425,10 +427,7 @@ class GoogleTranslateAPI:
                     translated_msg = await ch.send(msg, files=files)
 
     async def translate_message(
-        self,
-        message: discord.Message,
-        flag: str,
-        reacted_user: Optional[discord.Member] = None,
+        self, message: discord.Message, flag: str, reacted_user: Optional[discord.Member] = None,
     ) -> None:
         guild = cast(discord.Guild, message.guild)
         channel = cast(discord.TextChannel, message.channel)

translate/translate.py

@@ -48,7 +48,7 @@ class Translate(GoogleTranslateAPI, commands.Cog):
             "whitelist": [],
             "blacklist": [],
             "count": {"characters": 0, "requests": 0, "detect": 0},
-            "autosend": {},
+            "autosend": [],
         }
         default = {
             "cooldown": {"past_flags": [], "timeout": 0, "multiple": False},
@@ -108,11 +108,7 @@ class Translate(GoogleTranslateAPI, commands.Cog):
 
     @commands.command()
     async def translate(
-        self,
-        ctx: commands.Context,
-        to_language: FlagTranslation,
-        *,
-        message: Union[discord.Message, str],
+        self, ctx: commands.Context, to_language: FlagTranslation, *, message: Union[discord.Message, str],
     ) -> None:
         """
         Translate messages with Google Translate
@@ -191,8 +187,15 @@ class Translate(GoogleTranslateAPI, commands.Cog):
         """
         pass
 
-    @translateset.command(name="auto")
-    async def translate_auto(self, ctx: commands.Context, languages: str, *links: discord.TextChannel) -> None:
+    @translateset.group(name="auto")
+    async def translate_auto(self, ctx):
+        """
+        Set channels to auto translate messages from and to
+        """
+        pass
+
+    @translate_auto.command(name="add")
+    async def translate_auto_add(self, ctx: commands.Context, languages: str, *links: discord.TextChannel) -> None:
         """
         Set channels to auto translate messages from and to
 
@@ -206,7 +209,31 @@ class Translate(GoogleTranslateAPI, commands.Cog):
         if len(langs) != len(links):
             return await ctx.send(error("The number of lanuages and link channels don't match!"), delete_after=30)
 
-        await self.config.guild(ctx.guild).autosend.set({l.id: lang for l, lang in zip(links, langs)})
+        async with self.config.guild(ctx.guild).autosend() as autosend:
+            autosend.append({l.id: lang for l, lang in zip(links, langs)})
+
+        await ctx.tick()
+
+    @translate_auto.command(name="del")
+    async def translate_auto_del(self, ctx: commands.Context, *links: discord.TextChannel) -> None:
+        """
+        Delete linked channels
+        """
+        if not links:
+            await ctx.send("Please specify the channel links.")
+            return
+
+        links = {r.id for r in links}
+        async with self.config.guild(ctx.guild).autosend() as autosend:
+            to_delete_i = -1
+            for i, channels in enumerate(autosend):
+                ch = {int(k) for k in channels.keys()}
+                if ch == links:
+                    to_delete_i = i
+                    break
+
+            del autosend[to_delete_i]
+
         await ctx.tick()
 
     @translateset.command(name="stats")