mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-02-03 02:25:32 +08:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
import base64
|
||
from typing import Optional
|
||
|
||
from app.core.config import settings
|
||
from app.utils.http import RequestUtils
|
||
|
||
|
||
class OcrHelper:
|
||
|
||
_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):
|
||
"""
|
||
根据图片地址，获取验证码图片，并识别内容
|
||
:param image_url: 图片地址
|
||
:param image_b64: 图片base64，跳过图片地址下载
|
||
:param cookie: 下载图片使用的cookie
|
||
:param ua: 下载图片使用的ua
|
||
"""
|
||
if image_url:
|
||
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")
|
||
return ""
|