feat black_canned_message chanel options

mc-cloud-town · Feb 1, 2024 · ae8f5a4 · ae8f5a4
1 parent aec4cb1
commit ae8f5a4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/plugins/discord/client.py b/plugins/discord/client.py
@@ -186,7 +186,13 @@ async def on_message(self, msg: Message):
             "canned_message",
             {},
         )
-        if (key := msg.content) in canned_message:
+        if (
+            (key := msg.content) in canned_message
+            and msg.channel.id
+            not in (self.config.get("black_canned_message_channel", {})).get(key, [])
+            and msg.channel.parent_id
+            not in (self.config.get("black_canned_message_parent", {})).get(key, [])
+        ):
             await msg.channel.send(
                 random.choice(value)
                 if type(value := canned_message[key]) is list

diff --git a/plugins/discord/main.py b/plugins/discord/main.py
@@ -31,6 +31,8 @@ class DiscordConfig(Config):
     online_display_name = "雲鎮工藝"
     online_icon_url = "AUTO"
     canned_message: dict[str, str | list[str]] = {}
+    black_canned_message_channel: dict[str, list[int]] = {}
+    black_canned_message_parent: dict[str, list[int]] = {}
 
 
 class Discord(Plugin, config=DiscordConfig):