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
92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
import json
|
|
import logging
|
|
|
|
import httpx
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
SEND_ALL_URL = "https://m.api.weibo.com/2/messages/sendall.json"
|
|
DELETE_URL = "https://m.api.weibo.com/2/messages/delete"
|
|
|
|
|
|
class WeiboBroadcast:
|
|
def __init__(self, gateway=None):
|
|
self._gateway = gateway
|
|
self._http: httpx.AsyncClient | None = None
|
|
|
|
async def send_by_group(
|
|
self,
|
|
group_id: int,
|
|
msg_type: str,
|
|
content: dict,
|
|
) -> dict | None:
|
|
"""
|
|
按分组群发。
|
|
|
|
msg_type: text / articles / news
|
|
content:
|
|
- text: {"text": "消息内容"}
|
|
- articles: {"articles": [{"display_name": "标题", "summary": "摘要", "image_url": "...", "url": "..."}]}
|
|
- news: {"news": [{"display_name": "标题", "summary": "摘要", "image_url": "...", "url": "..."}]}
|
|
"""
|
|
return await self._send_all({"filter": {"group_id": group_id}}, msg_type, content)
|
|
|
|
async def send_by_users(
|
|
self,
|
|
user_ids: list[int],
|
|
msg_type: str,
|
|
content: dict,
|
|
) -> dict | None:
|
|
"""按 UID 列表群发"""
|
|
return await self._send_all({"touser": user_ids}, msg_type, content)
|
|
|
|
async def _send_all(self, target: dict, msg_type: str, content: dict) -> dict | None:
|
|
if self._http is None:
|
|
self._http = httpx.AsyncClient(timeout=30.0)
|
|
|
|
token = self._gateway.access_token if self._gateway else None
|
|
if not token:
|
|
logger.error("No access_token for broadcast")
|
|
return None
|
|
|
|
payload = {
|
|
"access_token": token,
|
|
**target,
|
|
"type": msg_type,
|
|
"data": json.dumps(content),
|
|
}
|
|
try:
|
|
resp = await self._http.post(SEND_ALL_URL, data=payload)
|
|
data = resp.json()
|
|
if data.get("error_code"):
|
|
logger.error("Broadcast failed: %s", data)
|
|
return data
|
|
except Exception:
|
|
logger.exception("Broadcast error")
|
|
return None
|
|
|
|
async def delete_broadcast(self, msg_id: str) -> bool:
|
|
"""删除已发送的群发消息（仅 news 和 image 类型支持）"""
|
|
if self._http is None:
|
|
self._http = httpx.AsyncClient(timeout=15.0)
|
|
|
|
token = self._gateway.access_token if self._gateway else None
|
|
if not token:
|
|
return False
|
|
|
|
try:
|
|
resp = await self._http.post(
|
|
DELETE_URL,
|
|
data={"access_token": token, "id": msg_id},
|
|
)
|
|
data = resp.json()
|
|
return "error_code" not in data
|
|
except Exception:
|
|
logger.exception("Delete broadcast error")
|
|
return False
|
|
|
|
async def close(self):
|
|
if self._http:
|
|
await self._http.aclose()
|
|
self._http = None
|