mirror of
https://github.com/zhayujie/bot-on-anything.git
synced 2026-02-18 16:16:59 +08:00
95 lines
3.2 KiB
Python
95 lines
3.2 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding=utf-8 -*-
|
||
"""
|
||
@time: 2023/4/10 22:24
|
||
@Project ：bot-on-anything
|
||
@file: wechat_com_channel.py
|
||
|
||
"""
|
||
from channel.channel import Channel
|
||
from concurrent.futures import ThreadPoolExecutor
|
||
from common.log import logger
|
||
from config import conf
|
||
|
||
from wechatpy.enterprise.crypto import WeChatCrypto
|
||
from wechatpy.enterprise import WeChatClient
|
||
from wechatpy.exceptions import InvalidSignatureException
|
||
from wechatpy.enterprise.exceptions import InvalidCorpIdException
|
||
from wechatpy.enterprise import parse_message
|
||
from flask import Flask, request, abort
|
||
|
||
thread_pool = ThreadPoolExecutor(max_workers=8)
|
||
app = Flask(__name__)
|
||
|
||
|
||
@app.route('/wechat', methods=['GET', 'POST'])
|
||
def handler_msg():
|
||
return WechatEnterpriseChannel().handle()
|
||
|
||
|
||
_conf = conf().get("channel").get("wechat_com")
|
||
|
||
|
||
class WechatEnterpriseChannel(Channel):
|
||
def __init__(self):
|
||
self.CorpId = _conf.get('wechat_corp_id')
|
||
self.Secret = _conf.get('secret')
|
||
self.AppId = _conf.get('appid')
|
||
self.TOKEN = _conf.get('wechat_token')
|
||
self.EncodingAESKey = _conf.get('wechat_encoding_aes_key')
|
||
self.crypto = WeChatCrypto(self.TOKEN, self.EncodingAESKey, self.CorpId)
|
||
self.client = WeChatClient(self.CorpId, self.Secret, self.AppId)
|
||
|
||
def startup(self):
|
||
# start message listener
|
||
app.run(host='0.0.0.0', port=_conf.get('port'))
|
||
|
||
def send(self, msg, receiver):
|
||
logger.info('[WXCOM] sendMsg={}, receiver={}'.format(msg, receiver))
|
||
self.client.message.send_text(self.AppId, receiver, msg)
|
||
|
||
def _do_send(self, query, reply_user_id):
|
||
try:
|
||
if not query:
|
||
return
|
||
context = dict()
|
||
context['from_user_id'] = reply_user_id
|
||
reply_text = super().build_reply_content(query, context)
|
||
if reply_text:
|
||
self.send(reply_text, reply_user_id)
|
||
except Exception as e:
|
||
logger.exception(e)
|
||
|
||
def handle(self):
|
||
query_params = request.args
|
||
signature = query_params.get('msg_signature', '')
|
||
timestamp = query_params.get('timestamp', '')
|
||
nonce = query_params.get('nonce', '')
|
||
if request.method == 'GET':
|
||
# 处理验证请求
|
||
echostr = query_params.get('echostr', '')
|
||
try:
|
||
echostr = self.crypto.check_signature(signature, timestamp, nonce, echostr)
|
||
except InvalidSignatureException:
|
||
abort(403)
|
||
print(echostr)
|
||
return echostr
|
||
elif request.method == 'POST':
|
||
try:
|
||
message = self.crypto.decrypt_message(
|
||
request.data,
|
||
signature,
|
||
timestamp,
|
||
nonce
|
||
)
|
||
except (InvalidSignatureException, InvalidCorpIdException):
|
||
abort(403)
|
||
msg = parse_message(message)
|
||
if msg.type == 'text':
|
||
thread_pool.submit(self._do_send, msg.content, msg.source)
|
||
else:
|
||
reply = 'Can not handle this for now'
|
||
# 未能处理的消息或菜单事件暂不做响应优化用户体验
|
||
# self.client.message.send_text(self.AppId, msg.source, reply)
|
||
return 'success'
|