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
|