fix: check if event is subscribe

tudou2 · Apr 21, 2023 · f6d370a · f6d370a
1 parent c9131b3
commit f6d370a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/channel/wechatmp/active_reply.py b/channel/wechatmp/active_reply.py
@@ -30,9 +30,12 @@ def POST(self):
                         msg.event, msg.source
                     )
                 )
-                reply_text = subscribe_msg()
-                replyPost = create_reply(reply_text, msg)
-                return replyPost.render()
+                if msg.event in ["subscribe", "subscribe_scan"]:
+                    reply_text = subscribe_msg()
+                    replyPost = create_reply(reply_text, msg)
+                    return replyPost.render()
+                else:
+                    return "success"
             wechatmp_msg = WeChatMPMessage(msg, client=channel.client)
             if wechatmp_msg.ctype in [ContextType.TEXT, ContextType.IMAGE, ContextType.VOICE]:
                 from_user = wechatmp_msg.from_user_id

diff --git a/channel/wechatmp/passive_reply.py b/channel/wechatmp/passive_reply.py
@@ -25,16 +25,18 @@ def POST(self):
             message = web.data() # todo crypto
             msg = parse_message(message)
             logger.debug("[wechatmp] Receive post data:\n" + message.decode("utf-8"))
-
             if msg.type == "event":
                 logger.info(
                     "[wechatmp] Event {} from {}".format(
                         msg.event, msg.source
                     )
                 )
-                reply_text = subscribe_msg()
-                replyPost = create_reply(reply_text, msg)
-                return replyPost.render()
+                if msg.event in ["subscribe", "subscribe_scan"]:
+                    reply_text = subscribe_msg()
+                    replyPost = create_reply(reply_text, msg)
+                    return replyPost.render()
+                else:
+                    return "success"
 
             wechatmp_msg = WeChatMPMessage(msg, client=channel.client)
             if wechatmp_msg.ctype in [ContextType.TEXT, ContextType.IMAGE, ContextType.VOICE]: