|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178 |
-
- import io
- import os
- import time
-
- import requests
- import web
- from wechatpy.enterprise import create_reply, parse_message
- from wechatpy.enterprise.crypto import WeChatCrypto
- from wechatpy.enterprise.exceptions import InvalidCorpIdException
- from wechatpy.exceptions import InvalidSignatureException, WeChatClientException
-
- from bridge.context import Context
- from bridge.reply import Reply, ReplyType
- from channel.chat_channel import ChatChannel
- from channel.wechatcom.wechatcomapp_client import WechatComAppClient
- from channel.wechatcom.wechatcomapp_message import WechatComAppMessage
- from common.log import logger
- from common.singleton import singleton
- from common.utils import compress_imgfile, fsize, split_string_by_utf8_length
- from config import conf, subscribe_msg
- from voice.audio_convert import any_to_amr, split_audio
-
- MAX_UTF8_LEN = 2048
-
-
- @singleton
- class WechatComAppChannel(ChatChannel):
- NOT_SUPPORT_REPLYTYPE = []
-
- def __init__(self):
- super().__init__()
- self.corp_id = conf().get("wechatcom_corp_id")
- self.secret = conf().get("wechatcomapp_secret")
- self.agent_id = conf().get("wechatcomapp_agent_id")
- self.token = conf().get("wechatcomapp_token")
- self.aes_key = conf().get("wechatcomapp_aes_key")
- print(self.corp_id, self.secret, self.agent_id, self.token, self.aes_key)
- logger.info(
- "[wechatcom] init: corp_id: {}, secret: {}, agent_id: {}, token: {}, aes_key: {}".format(self.corp_id, self.secret, self.agent_id, self.token, self.aes_key)
- )
- self.crypto = WeChatCrypto(self.token, self.aes_key, self.corp_id)
- self.client = WechatComAppClient(self.corp_id, self.secret)
-
- def startup(self):
-
- urls = ("/wxcomapp", "channel.wechatcom.wechatcomapp_channel.Query")
- app = web.application(urls, globals(), autoreload=False)
- port = conf().get("wechatcomapp_port", 9898)
- web.httpserver.runsimple(app.wsgifunc(), ("0.0.0.0", port))
-
- def send(self, reply: Reply, context: Context):
- receiver = context["receiver"]
- if reply.type in [ReplyType.TEXT, ReplyType.ERROR, ReplyType.INFO]:
- reply_text = reply.content
- texts = split_string_by_utf8_length(reply_text, MAX_UTF8_LEN)
- if len(texts) > 1:
- logger.info("[wechatcom] text too long, split into {} parts".format(len(texts)))
- for i, text in enumerate(texts):
- self.client.message.send_text(self.agent_id, receiver, text)
- if i != len(texts) - 1:
- time.sleep(0.5)
- logger.info("[wechatcom] Do send text to {}: {}".format(receiver, reply_text))
- elif reply.type == ReplyType.VOICE:
- try:
- media_ids = []
- file_path = reply.content
- amr_file = os.path.splitext(file_path)[0] + ".amr"
- any_to_amr(file_path, amr_file)
- duration, files = split_audio(amr_file, 60 * 1000)
- if len(files) > 1:
- logger.info("[wechatcom] voice too long {}s > 60s , split into {} parts".format(duration / 1000.0, len(files)))
- for path in files:
- response = self.client.media.upload("voice", open(path, "rb"))
- logger.debug("[wechatcom] upload voice response: {}".format(response))
- media_ids.append(response["media_id"])
- except WeChatClientException as e:
- logger.error("[wechatcom] upload voice failed: {}".format(e))
- return
- try:
- os.remove(file_path)
- if amr_file != file_path:
- os.remove(amr_file)
- except Exception:
- pass
- for media_id in media_ids:
- self.client.message.send_voice(self.agent_id, receiver, media_id)
- time.sleep(1)
- logger.info("[wechatcom] sendVoice={}, receiver={}".format(reply.content, receiver))
- elif reply.type == ReplyType.IMAGE_URL:
- img_url = reply.content
- pic_res = requests.get(img_url, stream=True)
- image_storage = io.BytesIO()
- for block in pic_res.iter_content(1024):
- image_storage.write(block)
- sz = fsize(image_storage)
- if sz >= 10 * 1024 * 1024:
- logger.info("[wechatcom] image too large, ready to compress, sz={}".format(sz))
- image_storage = compress_imgfile(image_storage, 10 * 1024 * 1024 - 1)
- logger.info("[wechatcom] image compressed, sz={}".format(fsize(image_storage)))
- image_storage.seek(0)
- try:
- response = self.client.media.upload("image", image_storage)
- logger.debug("[wechatcom] upload image response: {}".format(response))
- except WeChatClientException as e:
- logger.error("[wechatcom] upload image failed: {}".format(e))
- return
-
- self.client.message.send_image(self.agent_id, receiver, response["media_id"])
- logger.info("[wechatcom] sendImage url={}, receiver={}".format(img_url, receiver))
- elif reply.type == ReplyType.IMAGE:
- image_storage = reply.content
- sz = fsize(image_storage)
- if sz >= 10 * 1024 * 1024:
- logger.info("[wechatcom] image too large, ready to compress, sz={}".format(sz))
- image_storage = compress_imgfile(image_storage, 10 * 1024 * 1024 - 1)
- logger.info("[wechatcom] image compressed, sz={}".format(fsize(image_storage)))
- image_storage.seek(0)
- try:
- response = self.client.media.upload("image", image_storage)
- logger.debug("[wechatcom] upload image response: {}".format(response))
- except WeChatClientException as e:
- logger.error("[wechatcom] upload image failed: {}".format(e))
- return
- self.client.message.send_image(self.agent_id, receiver, response["media_id"])
- logger.info("[wechatcom] sendImage, receiver={}".format(receiver))
-
-
- class Query:
- def GET(self):
- channel = WechatComAppChannel()
- params = web.input()
- logger.info("[wechatcom] receive params: {}".format(params))
- try:
- signature = params.msg_signature
- timestamp = params.timestamp
- nonce = params.nonce
- echostr = params.echostr
- echostr = channel.crypto.check_signature(signature, timestamp, nonce, echostr)
- except InvalidSignatureException:
- raise web.Forbidden()
- return echostr
-
- def POST(self):
- channel = WechatComAppChannel()
- params = web.input()
- logger.info("[wechatcom] receive params: {}".format(params))
- try:
- signature = params.msg_signature
- timestamp = params.timestamp
- nonce = params.nonce
- message = channel.crypto.decrypt_message(web.data(), signature, timestamp, nonce)
- except (InvalidSignatureException, InvalidCorpIdException):
- raise web.Forbidden()
- msg = parse_message(message)
- logger.debug("[wechatcom] receive message: {}, msg= {}".format(message, msg))
- if msg.type == "event":
- if msg.event == "subscribe":
- reply_content = subscribe_msg()
- if reply_content:
- reply = create_reply(reply_content, msg).render()
- res = channel.crypto.encrypt_message(reply, nonce, timestamp)
- return res
- else:
- try:
- wechatcom_msg = WechatComAppMessage(msg, client=channel.client)
- except NotImplementedError as e:
- logger.debug("[wechatcom] " + str(e))
- return "success"
- context = channel._compose_context(
- wechatcom_msg.ctype,
- wechatcom_msg.content,
- isgroup=False,
- msg=wechatcom_msg,
- )
- if context:
- channel.produce(context)
- return "success"
|