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
|