105 lines
3.3 KiB
Python
105 lines
3.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""在服务器上生成通知图片并上传到企微"""
|
||
|
|
import asyncio
|
||
|
|
import sys
|
||
|
|
|
||
|
|
sys.path.insert(0, '/app')
|
||
|
|
|
||
|
|
from PIL import Image, ImageDraw, ImageFont
|
||
|
|
from app.services.wecom_service import WecomService
|
||
|
|
|
||
|
|
|
||
|
|
async def generate_image():
|
||
|
|
"""生成通知图片"""
|
||
|
|
WIDTH, HEIGHT = 600, 300
|
||
|
|
|
||
|
|
img = Image.new('RGB', (WIDTH, HEIGHT))
|
||
|
|
draw = ImageDraw.Draw(img)
|
||
|
|
|
||
|
|
# 渐变背景
|
||
|
|
for y in range(HEIGHT):
|
||
|
|
ratio = y / HEIGHT
|
||
|
|
r = int(0 + (0 - 0) * ratio)
|
||
|
|
g = int(120 + (180 - 120) * ratio)
|
||
|
|
b = int(215 + (160 - 215) * ratio)
|
||
|
|
draw.line([(0, y), (WIDTH, y)], fill=(r, g, b))
|
||
|
|
|
||
|
|
# 字体
|
||
|
|
try:
|
||
|
|
title_font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 32)
|
||
|
|
subtitle_font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf", 20)
|
||
|
|
button_font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 24)
|
||
|
|
except:
|
||
|
|
title_font = ImageFont.load_default()
|
||
|
|
subtitle_font = ImageFont.load_default()
|
||
|
|
button_font = ImageFont.load_default()
|
||
|
|
|
||
|
|
# 标题
|
||
|
|
draw.text((30, 30), "IT咨询即将关闭", fill=(255, 255, 255), font=title_font)
|
||
|
|
draw.text((30, 85), "请尽快回复坐席，否则将自动结束", fill=(220, 220, 220), font=subtitle_font)
|
||
|
|
|
||
|
|
# 3D按钮
|
||
|
|
btn_x, btn_y = 30, 150
|
||
|
|
btn_w, btn_h = 540, 60
|
||
|
|
|
||
|
|
# 阴影
|
||
|
|
draw.rounded_rectangle([btn_x+3, btn_y+3, btn_x+btn_w+3, btn_y+btn_h+3], radius=12, fill=(100, 100, 100))
|
||
|
|
|
||
|
|
# 按钮主体
|
||
|
|
for i in range(btn_h):
|
||
|
|
ratio = i / btn_h
|
||
|
|
r = int(20 + (100 - 20) * ratio)
|
||
|
|
g = int(150 + (200 - 150) * ratio)
|
||
|
|
b = int(220 + (255 - 220) * ratio)
|
||
|
|
draw.rectangle([btn_x, btn_y+i, btn_x+btn_w, btn_y+i+1], fill=(r, g, b))
|
||
|
|
|
||
|
|
# 高光
|
||
|
|
draw.rounded_rectangle([btn_x, btn_y, btn_x+btn_w, btn_y+btn_h//3], radius=12, fill=(255, 255, 255, 60))
|
||
|
|
|
||
|
|
# 按钮文字
|
||
|
|
draw.text((btn_x + btn_w//2 - 60, btn_y + 18), "立即回复", fill=(255, 255, 255), font=button_font)
|
||
|
|
|
||
|
|
# 保存到内存
|
||
|
|
from io import BytesIO
|
||
|
|
buffer = BytesIO()
|
||
|
|
img.save(buffer, format='PNG')
|
||
|
|
buffer.seek(0)
|
||
|
|
return buffer.getvalue()
|
||
|
|
|
||
|
|
|
||
|
|
async def upload_and_send():
|
||
|
|
"""上传图片并发送消息"""
|
||
|
|
print("🎨 正在生成通知图片...")
|
||
|
|
image_data = await generate_image()
|
||
|
|
print(f" 图片已生成, 大小: {len(image_data)} bytes")
|
||
|
|
|
||
|
|
# 创建企微服务
|
||
|
|
wecom = WecomService()
|
||
|
|
|
||
|
|
try:
|
||
|
|
print("📤 正在上传图片到企微...")
|
||
|
|
# 上传临时素材 - 传入二进制数据
|
||
|
|
media_id = await wecom.upload_temp_media("image", image_data, "reminder.png")
|
||
|
|
|
||
|
|
if not media_id or media_id.startswith("ERROR"):
|
||
|
|
print(f"❌ 上传失败: {media_id}")
|
||
|
|
return
|
||
|
|
|
||
|
|
print(f" 上传成功! media_id: {media_id}")
|
||
|
|
|
||
|
|
print("📨 正在发送图片消息...")
|
||
|
|
# 发送图片消息
|
||
|
|
result = await wecom.send_image_message("sxn", media_id)
|
||
|
|
|
||
|
|
if result.get("errcode") == 0:
|
||
|
|
print(f"✅ 发送成功! msgid: {result.get('msgid')}")
|
||
|
|
else:
|
||
|
|
print(f"❌ 发送失败: {result}")
|
||
|
|
|
||
|
|
finally:
|
||
|
|
await wecom.close()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
asyncio.run(upload_and_send())
|