suihf
0625f9b13b
- New uploader/alipay_uploader module: QR-code cookie login, upload page flow, title/desc/tags fill, thumbnail, collection and publish - sau_cli: add 'alipay login/check/upload-video' subcommands - examples: get_alipay_cookie.py and test_alipay_upload.py - utils/log.py: register alipay logger - utils/login_qrcode.py: robust QR decode on non-ASCII Windows paths (numpy imdecode with cv2.imread fallback)
90 lines
2.7 KiB
Python
90 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
from datetime import datetime
|
|
import base64
|
|
from pathlib import Path
|
|
import sys
|
|
|
|
import cv2
|
|
import numpy as np
|
|
import segno
|
|
|
|
|
|
def build_login_qrcode_path(account_file: str, suffix: str = "login_qrcode") -> Path:
|
|
account_path = Path(account_file)
|
|
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
|
return account_path.with_name(f"{account_path.stem}_{suffix}_{timestamp}.png")
|
|
|
|
|
|
def save_data_url_image(data_url: str, output_path: Path) -> Path:
|
|
if not data_url.startswith("data:image/"):
|
|
raise ValueError("二维码地址不是 data:image 格式")
|
|
|
|
header, encoded = data_url.split(",", 1)
|
|
if ";base64" not in header:
|
|
raise ValueError("二维码图片不是 base64 编码")
|
|
|
|
output_path.parent.mkdir(parents=True, exist_ok=True)
|
|
output_path.write_bytes(base64.b64decode(encoded))
|
|
return output_path
|
|
|
|
|
|
def remove_qrcode_file(qrcode_path: Path | None) -> bool:
|
|
if qrcode_path and qrcode_path.exists():
|
|
qrcode_path.unlink()
|
|
return True
|
|
return False
|
|
|
|
|
|
def decode_qrcode_from_path(qrcode_path: Path) -> str | None:
|
|
# Windows 下 cv2.imread 对中文路径不稳定，优先走 numpy+imdecode
|
|
image = None
|
|
try:
|
|
data = np.fromfile(str(qrcode_path), dtype=np.uint8)
|
|
if data.size > 0:
|
|
image = cv2.imdecode(data, cv2.IMREAD_COLOR)
|
|
except Exception:
|
|
image = None
|
|
if image is None:
|
|
image = cv2.imread(str(qrcode_path))
|
|
if image is None:
|
|
return None
|
|
|
|
detector = cv2.QRCodeDetector()
|
|
qrcode_content, _, _ = detector.detectAndDecode(image)
|
|
return qrcode_content or None
|
|
|
|
|
|
def _print_ascii_qrcode(qrcode) -> None:
|
|
border = 1
|
|
rows = list(qrcode.matrix)
|
|
empty_line = " " * (len(rows[0]) + border * 2)
|
|
print(empty_line)
|
|
for row in rows:
|
|
line = [" "] * border
|
|
line.extend("##" if cell else " " for cell in row)
|
|
line.extend([" "] * border)
|
|
print("".join(line))
|
|
print(empty_line)
|
|
|
|
|
|
def print_terminal_qrcode(
|
|
qrcode_content: str,
|
|
qrcode_path: Path,
|
|
app_name: str,
|
|
compact: bool = True,
|
|
border: int = 0,
|
|
) -> None:
|
|
print()
|
|
print(f"请使用{app_name}扫描下方二维码登录：")
|
|
qrcode = segno.make(qrcode_content, error="L", boost_error=False)
|
|
try:
|
|
if hasattr(sys.stdout, "reconfigure"):
|
|
sys.stdout.reconfigure(encoding="utf-8")
|
|
qrcode.terminal(compact=compact, border=border)
|
|
except (UnicodeEncodeError, OSError):
|
|
print("当前终端不支持 Unicode 二维码字符，已切换为 ASCII 打印：")
|
|
_print_ascii_qrcode(qrcode)
|
|
print("在 Windows 下建议使用 Windows Terminal（支持 UTF-8，可完整显示二维码）")
|
|
print(f"否则请打开 {qrcode_path} 扫码")
|
|
print()
|