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
93 lines
2.6 KiB
Python
93 lines
2.6 KiB
Python
import json
|
||
import logging
|
||
|
||
import httpx
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
POST_URL = "https://m.api.weibo.com/2/messages/post.json"
|
||
REMIND_URL = "https://m.api.weibo.com/2/messages/send.json"
|
||
|
||
|
||
class WeiboSubscription:
|
||
"""订阅用户推送 + 消息提醒"""
|
||
|
||
def __init__(self, gateway=None):
|
||
self._gateway = gateway
|
||
self._http: httpx.AsyncClient | None = None
|
||
|
||
async def post_to_subscriber(
|
||
self,
|
||
receiver_id: int,
|
||
msg_type: str,
|
||
content: dict,
|
||
) -> dict | None:
|
||
"""
|
||
向已订阅用户发送私信消息（messages/post）。
|
||
|
||
仅对已订阅用户（发送过 DY）有效。
|
||
msg_type: text / articles / position / image / voice
|
||
"""
|
||
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:
|
||
logger.error("No access_token for subscription post")
|
||
return None
|
||
|
||
payload = {
|
||
"access_token": token,
|
||
"receiver_id": receiver_id,
|
||
"type": msg_type,
|
||
"data": json.dumps(content),
|
||
}
|
||
try:
|
||
resp = await self._http.post(POST_URL, data=payload)
|
||
data = resp.json()
|
||
if data.get("error_code"):
|
||
logger.error("Subscription post failed: %s", data)
|
||
return data
|
||
except Exception:
|
||
logger.exception("Subscription post error")
|
||
return None
|
||
|
||
async def send_reminder(
|
||
self,
|
||
receiver_id: int,
|
||
text: str,
|
||
) -> dict | None:
|
||
"""
|
||
发送消息提醒（messages/send）。
|
||
|
||
根据粉丝设置的提醒条件发送私信提醒。
|
||
"""
|
||
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:
|
||
logger.error("No access_token for reminder")
|
||
return None
|
||
|
||
payload = {
|
||
"access_token": token,
|
||
"receiver_id": receiver_id,
|
||
"type": "text",
|
||
"data": json.dumps({"text": text}),
|
||
}
|
||
try:
|
||
resp = await self._http.post(REMIND_URL, data=payload)
|
||
data = resp.json()
|
||
if data.get("error_code"):
|
||
logger.error("Reminder send failed: %s", data)
|
||
return data
|
||
except Exception:
|
||
logger.exception("Reminder send error")
|
||
return None
|
||
|
||
async def close(self):
|
||
if self._http:
|
||
await self._http.aclose()
|
||
self._http = None
|