Kris
f6df6e2c95
新增快手（Kuaishou）渠道扩展，支持在 Yuxi 平台中集成快手客服渠道。 包含以下功能模块： - api: 快手 API 客户端封装 - accounts: 账户管理 - config: 渠道配置管理 - gateway: SSE/WebSocket 网关接入 - webhook: Webhook 事件处理 - outbound: 外发消息管理 - streaming: 流式消息处理 - pairing: 用户配对与绑定 - security: 安全校验 - signature: 请求签名验证 - dedupe: 消息去重 - monitor: 渠道状态监控 - status: 会话状态管理 - media: 媒体资源处理 - types: 类型定义
32 lines
866 B
Python
32 lines
866 B
Python
from __future__ import annotations
|
|
|
|
from enum import Enum
|
|
|
|
|
|
class DMPolicyMode(Enum):
|
|
OPEN = "open"
|
|
PAIRING = "pairing"
|
|
ALLOWLIST = "allowlist"
|
|
DISABLED = "disabled"
|
|
|
|
|
|
class KuaishouSecurity:
|
|
def __init__(self, policy: DMPolicyMode = DMPolicyMode.OPEN, allow_from: list[str] | None = None):
|
|
self._policy = policy
|
|
self._allowlist: set[str] = set(allow_from or [])
|
|
|
|
def check_allow(self, open_id: str) -> bool:
|
|
if self._policy == DMPolicyMode.OPEN:
|
|
return True
|
|
if self._policy == DMPolicyMode.DISABLED:
|
|
return False
|
|
if self._policy == DMPolicyMode.ALLOWLIST:
|
|
return open_id in self._allowlist
|
|
return True
|
|
|
|
def resolve_policy(self) -> dict:
|
|
return {
|
|
"mode": self._policy.value,
|
|
"allow_from": list(self._allowlist),
|
|
}
|