Update task_hook.py

HibiKier · Feb 25, 2022 · 3c8983e · 3c8983e
1 parent 1fe41dd
commit 3c8983e
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/basic_plugins/hooks/task_hook.py b/basic_plugins/hooks/task_hook.py
@@ -14,10 +14,20 @@ async def handle_api_call(bot: Bot, api: str, data: Dict[str, Any]):
             or api == "send_group_msg"
         )
         and (
-            (r := re.search("^\[\[_task\|(.*)]]", str(data["message"]).strip()))
+            (
+                r := re.search(
+                    "^\[\[_task\|(.*)]]",
+                    data["message"].strip()
+                    if isinstance(data["message"], str)
+                    else str(data["message"]["text"]).strip(),
+                )
+            )
             or (
                 r := re.search(
-                    "^&#91;&#91;_task\|(.*)&#93;&#93;", str(data["message"]).strip()
+                    "^&#91;&#91;_task\|(.*)&#93;&#93;",
+                    data["message"].strip()
+                    if isinstance(data["message"], str)
+                    else str(data["message"]["text"]).strip(),
                 )
             )
         )