mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-06-19 14:40:23 +08:00
69 lines
2.4 KiB
Python
69 lines
2.4 KiB
Python
import base64
|
||
from typing import Optional
|
||
|
||
from app.core.config import settings
|
||
from app.utils.http import RequestUtils
|
||
|
||
|
||
class OcrHelper:
|
||
"""
|
||
OCR 辅助类，负责获取验证码图片并调用 OCR 服务识别文本。
|
||
"""
|
||
|
||
_ocr_b64_url = f"{settings.OCR_HOST}/captcha/base64"
|
||
|
||
def get_captcha_text(
|
||
self,
|
||
image_url: Optional[str] = None,
|
||
image_b64: Optional[str] = None,
|
||
cookie: Optional[str] = None,
|
||
ua: Optional[str] = None,
|
||
) -> str:
|
||
"""
|
||
根据图片地址，获取验证码图片，并识别内容
|
||
:param image_url: 图片地址
|
||
:param image_b64: 图片base64，跳过图片地址下载
|
||
:param cookie: 下载图片使用的cookie
|
||
:param ua: 下载图片使用的ua
|
||
:return: 验证码识别结果，失败时返回空字符串
|
||
"""
|
||
image_b64 = self._normalize_image_base64(image_b64)
|
||
if image_url:
|
||
data_url_b64 = self._extract_data_url_base64(image_url)
|
||
if data_url_b64:
|
||
image_b64 = data_url_b64
|
||
else:
|
||
ret = RequestUtils(ua=ua,
|
||
cookies=cookie).get_res(image_url)
|
||
if ret is not None:
|
||
image_bin = ret.content
|
||
if not image_bin:
|
||
return ""
|
||
image_b64 = base64.b64encode(image_bin).decode()
|
||
if not image_b64:
|
||
return ""
|
||
ret = RequestUtils(content_type="application/json").post_res(
|
||
url=self._ocr_b64_url,
|
||
json={"base64_img": image_b64})
|
||
if ret:
|
||
return ret.json().get("result") or ""
|
||
return ""
|
||
|
||
@staticmethod
|
||
def _normalize_image_base64(image_b64: Optional[str]) -> str:
|
||
"""规范化外部传入的图片 base64 内容。"""
|
||
if not image_b64:
|
||
return ""
|
||
return OcrHelper._extract_data_url_base64(image_b64) or image_b64.strip()
|
||
|
||
@staticmethod
|
||
def _extract_data_url_base64(image_url: Optional[str]) -> str:
|
||
"""从 data:image/...;base64,... 地址中提取纯 base64 内容。"""
|
||
image_url = (image_url or "").strip()
|
||
if not image_url.lower().startswith("data:image/"):
|
||
return ""
|
||
metadata, separator, data = image_url.partition(",")
|
||
if not separator or ";base64" not in metadata.lower():
|
||
return ""
|
||
return data.strip()
|