Browse Source

Update mq_sender: new message type

StrayWarrior 2 weeks ago
parent
commit
76af22ae26
1 changed files with 6 additions and 3 deletions
  1. 6 3
      scripts/mq_sender.py

+ 6 - 3
scripts/mq_sender.py

@@ -33,9 +33,12 @@ if __name__ == '__main__':
         message_id += 1
         sender = args.user_id
         receiver = args.staff_id
-        if text == MessageType.AGGREGATION_TRIGGER.name:
-            message = Message.build(MessageType.AGGREGATION_TRIGGER, MessageChannel.CORP_WECHAT,
-                                    sender, receiver, None, int(time.time() * 1000))
+        if text in (MessageType.AGGREGATION_TRIGGER.name,
+                    MessageType.HUMAN_INTERVENTION_END.name):
+            message = Message.build(
+                MessageType.__members__.get(text),
+                MessageChannel.CORP_WECHAT,
+                sender, receiver, None, int(time.time() * 1000))
         else:
             message = Message.build(MessageType.TEXT, MessageChannel.CORP_WECHAT,
                                     sender,receiver, text, int(time.time() * 1000)