Skip to content

feat(wechat): add wechat_allowed_users gating, mirror QQ adapter pattern #97

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
shaun0927 wants to merge 1 commit into
base: main
Choose a base branch
from
Open

feat(wechat): add wechat_allowed_users gating, mirror QQ adapter pattern #97

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions frontends/wechatapp.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,10 @@
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
_TEMP_DIR = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'temp')
from agentmain import GeneraticAgent
from llmcore import mykeys
from chatapp_common import public_access

ALLOWED = {str(x).strip() for x in mykeys.get('wechat_allowed_users', []) if str(x).strip()}

# ── WxBotClient (inline from wx_bot_client.py) ──
API = 'https://ilinkai.weixin.qq.com'
Expand Down Expand Up @@ -233,6 +237,9 @@ def on_message(bot, msg):
text = bot.extract_text(msg).strip()
uid = msg.get('from_user_id', '')
ctx = msg.get('context_token', '')
if not public_access(ALLOWED) and uid not in ALLOWED:
print(f'[WX] unauthorized user: {uid}', file=sys.__stdout__)
return
media_paths = _dl_media(msg.get('item_list', []))
if not text and not media_paths: return
if media_paths:
Expand Down
1 change: 1 addition & 0 deletions mykey_template.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -392,6 +392,7 @@
# wecom_secret = 'your_bot_secret'
# wecom_allowed_users = ['your_user_id'] # 留空或 ['*'] 表示允许所有企业微信用户
# wecom_welcome_message = '你好，我在线上。'
# wechat_allowed_users = ['from_user_id'] # 留空或 ['*'] 表示允许所有微信用户
# dingtalk_client_id = 'your_app_key'
# dingtalk_client_secret = 'your_app_secret'
# dingtalk_allowed_users = ['your_staff_id'] # 留空或 ['*'] 表示允许所有钉钉用户