Kris
7215418610
新增企业微信、微博、WhatsApp、Workplace 四个渠道扩展。 企业微信渠道扩展主要模块：config, gateway, webhook, webhook_bot, outbound, streaming, pairing, security, crypto, dedupe, persistent_dedupe, card, directory, events, externalcontact, media, mentions, menu, message, oauth, status 微博渠道扩展主要模块：config, gateway, webhook, outbound, streaming, pairing, security, dedupe, passive_reply, broadcast, message, menu, media, subscription, template, status WhatsApp 渠道扩展主要模块：config, gateway, webhook, outbound, streaming, pairing, security, dedupe, actions, monitor, status Workplace 渠道扩展主要模块：config, gateway, webhook, outbound, streaming, pairing, security, dedupe, actions, challenge, groups, media, mentions, menu, monitor, persona, quick_reply, signature, subscriptions, template, threading, users, status
32 lines
979 B
Python
32 lines
979 B
Python
from yuxi.channel.extensions.wecom.types import WeComAccount
|
|
|
|
|
|
class WeComSecurity:
|
|
def __init__(self, account: WeComAccount):
|
|
self._account = account
|
|
self._allowlist: list[str] = list(account.allow_from or [])
|
|
|
|
def resolve_dm_policy(self) -> dict:
|
|
return {
|
|
"mode": self._account.dm_policy,
|
|
"allow_from": self._allowlist,
|
|
}
|
|
|
|
def resolve_group_policy(self) -> dict:
|
|
return {
|
|
"mode": self._account.group_policy,
|
|
}
|
|
|
|
def check_allowlist(self, peer_id: str, channel_type: str = "direct") -> bool:
|
|
if channel_type == "group":
|
|
return True
|
|
return peer_id in self._allowlist
|
|
|
|
def add_to_allowlist(self, peer_id: str) -> None:
|
|
if peer_id not in self._allowlist:
|
|
self._allowlist.append(peer_id)
|
|
|
|
def remove_from_allowlist(self, peer_id: str) -> None:
|
|
if peer_id in self._allowlist:
|
|
self._allowlist.remove(peer_id)
|