78 lines
1.9 KiB
Python
78 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
import sys
|
|
import urllib.error
|
|
import urllib.request
|
|
|
|
|
|
FETCH_API_URL = "https://api.pearktrue.cn/api/today_wife"
|
|
FALLBACK_TEXT = "今天的美女图片暂时没拿到，等我再找找。"
|
|
|
|
|
|
def fetch_image_url() -> str | None:
|
|
try:
|
|
with urllib.request.urlopen(FETCH_API_URL, timeout=10) as response:
|
|
payload = json.load(response)
|
|
except (urllib.error.URLError, TimeoutError, json.JSONDecodeError):
|
|
return None
|
|
|
|
data = payload.get("data")
|
|
if not isinstance(data, dict):
|
|
return None
|
|
|
|
image_url = data.get("image_url")
|
|
if isinstance(image_url, str) and image_url.strip():
|
|
return image_url.strip()
|
|
return None
|
|
|
|
|
|
def send_image(image_url: str) -> bool:
|
|
robot_port = os.environ.get("ROBOT_WECHAT_CLIENT_PORT", "").strip()
|
|
to_wxid = os.environ.get("ROBOT_FROM_WX_ID", "").strip()
|
|
if not robot_port or not to_wxid:
|
|
return False
|
|
|
|
api_url = (
|
|
f"http://127.0.0.1:{robot_port}/api/v1/robot/message/send/image/url"
|
|
)
|
|
body = json.dumps(
|
|
{
|
|
"to_wxid": to_wxid,
|
|
"image_urls": [image_url],
|
|
}
|
|
).encode("utf-8")
|
|
request = urllib.request.Request(
|
|
api_url,
|
|
data=body,
|
|
headers={"Content-Type": "application/json"},
|
|
method="POST",
|
|
)
|
|
|
|
try:
|
|
with urllib.request.urlopen(request, timeout=10) as response:
|
|
if 200 <= response.status < 300:
|
|
return True
|
|
payload = json.load(response)
|
|
except (urllib.error.URLError, TimeoutError, json.JSONDecodeError):
|
|
return False
|
|
|
|
code = payload.get("code")
|
|
return code == 200 or code == 0
|
|
|
|
|
|
def main() -> int:
|
|
image_url = fetch_image_url()
|
|
if image_url and send_image(image_url):
|
|
return 0
|
|
|
|
sys.stdout.write(FALLBACK_TEXT)
|
|
sys.stdout.write("\n")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main()) |