259 lines
8.1 KiB
Python
259 lines
8.1 KiB
Python
"""
|
||
二维码生成路由
|
||
生成项目绑定二维码，供手机APP扫描绑定
|
||
|
||
NAS armv7 等环境无 Pillow 编译链时，自动降级为 qrcode + pypng（纯 Python PNG）。
|
||
"""
|
||
|
||
from fastapi import APIRouter, Response
|
||
from pydantic import BaseModel
|
||
from typing import Optional, Type
|
||
import os
|
||
import json
|
||
import base64
|
||
import io
|
||
|
||
# qrcode 核心 + 图像后端（PIL 优先，pypng 兜底 — 适配 DS213j 等无 zlib-dev 的 NAS）
|
||
try:
|
||
import qrcode
|
||
from qrcode.constants import ERROR_CORRECT_H
|
||
|
||
_QR_IMAGE_FACTORY: Optional[Type] = None
|
||
_QR_USE_PIL_COLORS = False
|
||
try:
|
||
from PIL import Image # noqa: F401
|
||
from qrcode.image.pil import PilImage
|
||
|
||
_QR_IMAGE_FACTORY = PilImage
|
||
_QR_USE_PIL_COLORS = True
|
||
except ImportError:
|
||
try:
|
||
from qrcode.image.pure import PyPNGImage
|
||
|
||
_QR_IMAGE_FACTORY = PyPNGImage
|
||
except ImportError:
|
||
pass
|
||
QRCODE_AVAILABLE = _QR_IMAGE_FACTORY is not None
|
||
except ImportError:
|
||
QRCODE_AVAILABLE = False
|
||
_QR_IMAGE_FACTORY = None
|
||
_QR_USE_PIL_COLORS = False
|
||
|
||
router = APIRouter(prefix="/qrcode", tags=["二维码"])
|
||
|
||
_QR_INSTALL_HINT = (
|
||
"请安装: pip install qrcode pypng （或 pip install 'qrcode[pil]'）"
|
||
)
|
||
|
||
|
||
class ProjectBindConfig(BaseModel):
|
||
"""项目绑定配置"""
|
||
project_id: str
|
||
project_name: Optional[str] = ""
|
||
server: Optional[str] = "wss://wpsdk.quwanzhi.com/ws/device"
|
||
pwa: Optional[str] = ""
|
||
ai_api_key: Optional[str] = ""
|
||
ai_api_url: Optional[str] = "https://ckbapi.quwanzhi.com"
|
||
# BIND-03/05：公网主服有序回退列表（逗号分隔）+ 开机自动寻服
|
||
public_servers: Optional[str] = ""
|
||
auto_discover: Optional[bool] = False
|
||
pairing_token: Optional[str] = ""
|
||
|
||
|
||
def _build_qr_content(
|
||
project_id: str,
|
||
project_name: str = "",
|
||
server: str = "wss://wpsdk.quwanzhi.com/ws/device",
|
||
pwa: str = "",
|
||
ai_api_key: str = "",
|
||
ai_api_url: str = "",
|
||
public_servers: str = "",
|
||
auto_discover: bool = False,
|
||
pairing_token: str = "",
|
||
) -> str:
|
||
qr_data = {"server": server, "project": project_id, "project_name": project_name}
|
||
if pwa:
|
||
qr_data["pwa"] = pwa
|
||
if ai_api_key:
|
||
qr_data["ai_api_key"] = ai_api_key
|
||
if ai_api_url:
|
||
qr_data["ai_api_url"] = ai_api_url
|
||
if public_servers:
|
||
qr_data["public_servers"] = public_servers
|
||
if auto_discover:
|
||
qr_data["auto_discover"] = "1"
|
||
token = pairing_token or os.environ.get("DEVICE_PAIRING_TOKEN", "")
|
||
if token:
|
||
qr_data["pairing_token"] = token
|
||
return json.dumps(qr_data, ensure_ascii=False)
|
||
|
||
|
||
def _render_qr_png_bytes(qr_content: str) -> bytes:
|
||
qr = qrcode.QRCode(
|
||
version=1,
|
||
error_correction=ERROR_CORRECT_H,
|
||
box_size=10,
|
||
border=4,
|
||
)
|
||
qr.add_data(qr_content)
|
||
qr.make(fit=True)
|
||
|
||
if _QR_USE_PIL_COLORS:
|
||
img = qr.make_image(
|
||
fill_color="black",
|
||
back_color="white",
|
||
image_factory=_QR_IMAGE_FACTORY,
|
||
)
|
||
else:
|
||
img = qr.make_image(image_factory=_QR_IMAGE_FACTORY)
|
||
|
||
buffer = io.BytesIO()
|
||
img.save(buffer)
|
||
return buffer.getvalue()
|
||
|
||
|
||
@router.post("/generate")
|
||
async def generate_qrcode(config: ProjectBindConfig):
|
||
"""
|
||
生成项目绑定二维码
|
||
|
||
返回: Base64编码的PNG图片
|
||
"""
|
||
if not QRCODE_AVAILABLE:
|
||
return {
|
||
"success": False,
|
||
"message": f"服务器未安装 qrcode 图像库，{_QR_INSTALL_HINT}",
|
||
}
|
||
|
||
qr_content = _build_qr_content(
|
||
config.project_id,
|
||
config.project_name or "",
|
||
config.server or "",
|
||
config.pwa or "",
|
||
config.ai_api_key or "",
|
||
config.ai_api_url or "",
|
||
config.public_servers or "",
|
||
bool(config.auto_discover),
|
||
config.pairing_token or "",
|
||
)
|
||
png_bytes = _render_qr_png_bytes(qr_content)
|
||
img_base64 = base64.b64encode(png_bytes).decode()
|
||
|
||
return {
|
||
"success": True,
|
||
"project_id": config.project_id,
|
||
"project_name": config.project_name,
|
||
"qr_content": qr_content,
|
||
"image_base64": img_base64,
|
||
"image_data_url": f"data:image/png;base64,{img_base64}",
|
||
}
|
||
|
||
|
||
@router.get("/image/{project_id}")
|
||
async def get_qrcode_image(
|
||
project_id: str,
|
||
project_name: str = "",
|
||
server: str = "wss://wpsdk.quwanzhi.com/ws/device",
|
||
pwa: str = "",
|
||
):
|
||
"""直接获取二维码 PNG 图片"""
|
||
if not QRCODE_AVAILABLE:
|
||
return Response(
|
||
content=f"QRCode not available. {_QR_INSTALL_HINT}",
|
||
media_type="text/plain",
|
||
status_code=500,
|
||
)
|
||
|
||
qr_content = _build_qr_content(project_id, project_name, server, pwa)
|
||
return Response(
|
||
content=_render_qr_png_bytes(qr_content),
|
||
media_type="image/png",
|
||
)
|
||
|
||
|
||
@router.get("/html/{project_id}")
|
||
async def get_qrcode_html(
|
||
project_id: str,
|
||
project_name: str = "",
|
||
server: str = "wss://wpsdk.quwanzhi.com/ws/device",
|
||
pwa: str = "",
|
||
):
|
||
"""获取带样式的二维码 HTML 页面"""
|
||
if not QRCODE_AVAILABLE:
|
||
return Response(
|
||
content=f"<h1>QRCode not available</h1><p>{_QR_INSTALL_HINT}</p>",
|
||
media_type="text/html",
|
||
status_code=500,
|
||
)
|
||
|
||
qr_content = _build_qr_content(project_id, project_name, server, pwa)
|
||
img_base64 = base64.b64encode(_render_qr_png_bytes(qr_content)).decode()
|
||
|
||
html = f"""
|
||
<!DOCTYPE html>
|
||
<html lang="zh-CN">
|
||
<head>
|
||
<meta charset="UTF-8">
|
||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||
<title>项目绑定二维码 - {project_name or project_id}</title>
|
||
<style>
|
||
* {{ margin: 0; padding: 0; box-sizing: border-box; }}
|
||
body {{
|
||
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
|
||
background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
|
||
min-height: 100vh;
|
||
display: flex;
|
||
justify-content: center;
|
||
align-items: center;
|
||
padding: 20px;
|
||
}}
|
||
.card {{
|
||
background: white;
|
||
border-radius: 20px;
|
||
padding: 40px;
|
||
text-align: center;
|
||
box-shadow: 0 20px 60px rgba(0,0,0,0.3);
|
||
max-width: 400px;
|
||
}}
|
||
.logo {{ font-size: 48px; margin-bottom: 20px; }}
|
||
h1 {{ color: #333; font-size: 24px; margin-bottom: 10px; }}
|
||
.project-name {{
|
||
color: #4CAF50; font-size: 20px; font-weight: bold; margin-bottom: 30px;
|
||
}}
|
||
.qr-container {{
|
||
background: #f5f5f5; padding: 20px; border-radius: 16px; margin-bottom: 30px;
|
||
}}
|
||
.qr-container img {{ max-width: 100%; height: auto; }}
|
||
.instructions {{
|
||
color: #666; font-size: 14px; line-height: 1.8; text-align: left;
|
||
}}
|
||
.instructions li {{ margin-bottom: 8px; }}
|
||
.project-id {{
|
||
background: #e8f5e9; color: #2e7d32; padding: 8px 16px;
|
||
border-radius: 20px; display: inline-block; font-size: 12px; margin-top: 20px;
|
||
}}
|
||
@media print {{ body {{ background: white; }} .card {{ box-shadow: none; }} }}
|
||
</style>
|
||
</head>
|
||
<body>
|
||
<div class="card">
|
||
<div class="logo">📱</div>
|
||
<h1>工作手机Agent</h1>
|
||
<div class="project-name">{project_name or project_id}</div>
|
||
<div class="qr-container">
|
||
<img src="data:image/png;base64,{img_base64}" alt="绑定二维码">
|
||
</div>
|
||
<ul class="instructions">
|
||
<li>打开手机上的「工作手机Agent」APP</li>
|
||
<li>点击「📷 扫码绑定项目」按钮</li>
|
||
<li>扫描上方二维码完成绑定</li>
|
||
<li>点击「连接服务器」开始使用</li>
|
||
</ul>
|
||
<div class="project-id">项目ID: {project_id}</div>
|
||
</div>
|
||
</body>
|
||
</html>
|
||
"""
|
||
|
||
return Response(content=html, media_type="text/html")
|