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
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
import logging
|
|
|
|
from yuxi.channel.extensions.weibo.types import WeiboAccount
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class WeiboSecurity:
|
|
VALID_POLICIES = {"open", "pairing", "allowlist", "disabled"}
|
|
|
|
def __init__(self, account: WeiboAccount | None = None):
|
|
self._account = account
|
|
if account and account.dm_policy in self.VALID_POLICIES:
|
|
self._policy = account.dm_policy
|
|
else:
|
|
self._policy = "open"
|
|
self._allowlist: set[str] = set(account.allow_from) if account else set()
|
|
|
|
def resolve_dm_policy(self) -> str:
|
|
return self._policy
|
|
|
|
def check_allowlist(self, sender_id: str, channel_type: str = "weibo") -> bool:
|
|
if self._policy == "open":
|
|
return True
|
|
if self._policy == "disabled":
|
|
return False
|
|
if not self._allowlist:
|
|
logger.warning("allowlist is empty with policy=%s", self._policy)
|
|
return False
|
|
return sender_id in self._allowlist
|
|
|
|
def add_to_allowlist(self, sender_id: str):
|
|
self._allowlist.add(sender_id)
|
|
|
|
def remove_from_allowlist(self, sender_id: str):
|
|
self._allowlist.discard(sender_id)
|