mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git
synced 2025-08-01 16:39:52 +00:00
完成消息体内 LaTeX 内容渲染功能 (#15)
* 确实，现在可以处理 LaTeX 渲染了，欢迎 PR 新的渲染网址。 * 意外的小问题 * 删掉一个小数点 * 单词拼错了，马上四级，不知道能不能过 * 我是傻逼 * ok，但我肚子痛，等去蹲个坑
This commit is contained in:
金羿ELS
GitHub
299
nonebot_plugin_marshoai/deal_latex.py
Normal file
299
nonebot_plugin_marshoai/deal_latex.py
Normal file
@ -0,0 +1,299 @@
|
||||
"""
|
||||
此文件援引并改编自 nonebot-plugin-latex 数据类
|
||||
源项目地址: https://github.com/EillesWan/nonebot-plugin-latex
|
||||
|
||||
|
||||
Copyright (c) 2024 金羿Eilles
|
||||
nonebot-plugin-latex is licensed under Mulan PSL v2.
|
||||
You can use this software according to the terms and conditions of the Mulan PSL v2.
|
||||
You may obtain a copy of Mulan PSL v2 at:
|
||||
http://license.coscl.org.cn/MulanPSL2
|
||||
THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
|
||||
EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
|
||||
MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
|
||||
See the Mulan PSL v2 for more details.
|
||||
"""
|
||||
|
||||
from typing import Optional, Literal, Tuple
|
||||
import httpx
|
||||
import time
|
||||
|
||||
|
||||
class ConvertChannel:
|
||||
URL: str
|
||||
|
||||
async def get_to_convert(
|
||||
self,
|
||||
latex_code: str,
|
||||
dpi: int = 600,
|
||||
fgcolour: str = "000000",
|
||||
timeout: int = 5,
|
||||
retry: int = 3,
|
||||
) -> Tuple[Literal[True], bytes] | Tuple[Literal[False], bytes | str]:
|
||||
return False, "请勿直接调用母类"
|
||||
|
||||
@staticmethod
|
||||
def channel_test() -> int:
|
||||
return -1
|
||||
|
||||
|
||||
class L2PChannel(ConvertChannel):
|
||||
|
||||
URL = "https://www.latex2png.com"
|
||||
|
||||
async def get_to_convert(
|
||||
self,
|
||||
latex_code: str,
|
||||
dpi: int = 600,
|
||||
fgcolour: str = "000000",
|
||||
timeout: int = 5,
|
||||
retry: int = 3,
|
||||
) -> Tuple[Literal[True], bytes] | Tuple[Literal[False], bytes | str]:
|
||||
|
||||
async with httpx.AsyncClient(
|
||||
timeout=timeout,
|
||||
) as client:
|
||||
while retry > 0:
|
||||
try:
|
||||
post_response = await client.post(
|
||||
self.URL + "/api/convert",
|
||||
json={
|
||||
"auth": {"user": "guest", "password": "guest"},
|
||||
"latex": latex_code,
|
||||
"resolution": dpi,
|
||||
"color": fgcolour,
|
||||
},
|
||||
)
|
||||
if post_response.status_code == 200:
|
||||
|
||||
if (json_response := post_response.json())[
|
||||
"result-message"
|
||||
] == "success":
|
||||
|
||||
# print("latex2png:", post_response.content)
|
||||
|
||||
if (
|
||||
get_response := await client.get(
|
||||
self.URL + json_response["url"]
|
||||
)
|
||||
).status_code == 200:
|
||||
return True, get_response.content
|
||||
else:
|
||||
return False, json_response["result-message"]
|
||||
retry -= 1
|
||||
except httpx.TimeoutException:
|
||||
retry -= 1
|
||||
raise ConnectionError("服务不可用")
|
||||
return False, "未知错误"
|
||||
|
||||
@staticmethod
|
||||
def channel_test() -> int:
|
||||
with httpx.Client(timeout=5) as client:
|
||||
try:
|
||||
start_time = time.time_ns()
|
||||
latex2png = (
|
||||
client.get(
|
||||
"https://www.latex2png.com{}"
|
||||
+ client.post(
|
||||
"https://www.latex2png.com/api/convert",
|
||||
json={
|
||||
"auth": {"user": "guest", "password": "guest"},
|
||||
"latex": "\\\\int_{a}^{b} x^2 \\\\, dx = \\\\frac{b^3}{3} - \\\\frac{a^3}{5}\n",
|
||||
"resolution": 600,
|
||||
"color": "000000",
|
||||
},
|
||||
).json()["url"]
|
||||
),
|
||||
time.time_ns() - start_time,
|
||||
)
|
||||
except:
|
||||
return 99999
|
||||
if latex2png[0].status_code == 200:
|
||||
return latex2png[1]
|
||||
else:
|
||||
return 99999
|
||||
|
||||
|
||||
class CDCChannel(ConvertChannel):
|
||||
|
||||
URL = "https://latex.codecogs.com"
|
||||
|
||||
async def get_to_convert(
|
||||
self,
|
||||
latex_code: str,
|
||||
dpi: int = 600,
|
||||
fgcolour: str = "000000",
|
||||
timeout: int = 5,
|
||||
retry: int = 3,
|
||||
) -> Tuple[Literal[True], bytes] | Tuple[Literal[False], bytes | str]:
|
||||
async with httpx.AsyncClient(
|
||||
timeout=timeout,
|
||||
) as client:
|
||||
|
||||
while retry > 0:
|
||||
try:
|
||||
response = await client.get(
|
||||
self.URL
|
||||
+ r"/png.image?\huge&space;\dpi{"
|
||||
+ str(dpi)
|
||||
+ r"}\fg{"
|
||||
+ fgcolour
|
||||
+ r"}"
|
||||
+ latex_code
|
||||
)
|
||||
# print("codecogs:", response)
|
||||
if response.status_code == 200:
|
||||
return True, response.content
|
||||
else:
|
||||
return False, response.content
|
||||
retry -= 1
|
||||
except httpx.TimeoutException:
|
||||
retry -= 1
|
||||
return False, "未知错误"
|
||||
|
||||
@staticmethod
|
||||
def channel_test() -> int:
|
||||
with httpx.Client(timeout=5) as client:
|
||||
try:
|
||||
start_time = time.time_ns()
|
||||
codecogs = (
|
||||
client.get(
|
||||
r"https://latex.codecogs.com/png.image?\huge%20\dpi{600}\\int_{a}^{b}x^2\\,dx=\\frac{b^3}{3}-\\frac{a^3}{5}"
|
||||
),
|
||||
time.time_ns() - start_time,
|
||||
)
|
||||
except:
|
||||
return 99999
|
||||
if codecogs[0].status_code == 200:
|
||||
return codecogs[1]
|
||||
else:
|
||||
return 99999
|
||||
|
||||
|
||||
class JRTChannel(ConvertChannel):
|
||||
|
||||
URL = "https://latex2image.joeraut.com"
|
||||
|
||||
async def get_to_convert(
|
||||
self,
|
||||
latex_code: str,
|
||||
dpi: int = 600,
|
||||
fgcolour: str = "000000", # 无效设置
|
||||
timeout: int = 5,
|
||||
retry: int = 3,
|
||||
) -> Tuple[Literal[True], bytes] | Tuple[Literal[False], bytes | str]:
|
||||
|
||||
async with httpx.AsyncClient(
|
||||
timeout=timeout,
|
||||
) as client:
|
||||
while retry > 0:
|
||||
try:
|
||||
post_response = await client.post(
|
||||
self.URL + "/default/latex2image",
|
||||
json={
|
||||
"latexInput": latex_code,
|
||||
"outputFormat": "PNG",
|
||||
"outputScale": "{}%".format(dpi / 3 * 5),
|
||||
},
|
||||
)
|
||||
print(post_response)
|
||||
if post_response.status_code == 200:
|
||||
|
||||
if not (json_response := post_response.json())["error"]:
|
||||
|
||||
# print("latex2png:", post_response.content)
|
||||
|
||||
if (
|
||||
get_response := await client.get(
|
||||
json_response["imageUrl"]
|
||||
)
|
||||
).status_code == 200:
|
||||
return True, get_response.content
|
||||
else:
|
||||
return False, json_response["error"]
|
||||
retry -= 1
|
||||
except httpx.TimeoutException:
|
||||
retry -= 1
|
||||
raise ConnectionError("服务不可用")
|
||||
return False, "未知错误"
|
||||
|
||||
@staticmethod
|
||||
def channel_test() -> int:
|
||||
with httpx.Client(timeout=5) as client:
|
||||
try:
|
||||
start_time = time.time_ns()
|
||||
joeraut = (
|
||||
client.get(
|
||||
client.post(
|
||||
"https://www.latex2png.com/api/convert",
|
||||
json={
|
||||
"latexInput": "\\\\int_{a}^{b} x^2 \\\\, dx = \\\\frac{b^3}{3} - \\\\frac{a^3}{5}",
|
||||
"outputFormat": "PNG",
|
||||
"outputScale": "1000%",
|
||||
},
|
||||
).json()["imageUrl"]
|
||||
),
|
||||
time.time_ns() - start_time,
|
||||
)
|
||||
except:
|
||||
return 99999
|
||||
if joeraut[0].status_code == 200:
|
||||
return joeraut[1]
|
||||
else:
|
||||
return 99999
|
||||
|
||||
|
||||
channel_list: list[type[ConvertChannel]] = [L2PChannel, CDCChannel, JRTChannel]
|
||||
|
||||
|
||||
class ConvertLatex:
|
||||
|
||||
channel: ConvertChannel
|
||||
|
||||
def __init__(self, channel: Optional[ConvertChannel] = None) -> None:
|
||||
|
||||
if channel is None:
|
||||
self.channel = self.auto_choose_channel()
|
||||
else:
|
||||
self.channel = channel
|
||||
|
||||
async def generate_png(
|
||||
self,
|
||||
latex: str,
|
||||
dpi: int = 600,
|
||||
foreground_colour: str = "000000",
|
||||
timeout_: int = 5,
|
||||
retry_: int = 3,
|
||||
) -> Tuple[Literal[True], bytes] | Tuple[Literal[False], bytes | str]:
|
||||
"""
|
||||
LaTeX 在线渲染
|
||||
|
||||
参数
|
||||
====
|
||||
|
||||
latex: str
|
||||
LaTeX 代码
|
||||
dpi: int
|
||||
分辨率
|
||||
foreground_colour: str
|
||||
文字前景色
|
||||
timeout_: int
|
||||
超时时间
|
||||
retry_: int
|
||||
重试次数
|
||||
返回
|
||||
====
|
||||
bytes
|
||||
图片
|
||||
"""
|
||||
return await self.channel.get_to_convert(
|
||||
latex, dpi, foreground_colour, timeout_, retry_
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def auto_choose_channel() -> ConvertChannel:
|
||||
|
||||
return min(
|
||||
channel_list,
|
||||
key=lambda channel: channel.channel_test(),
|
||||
)()
|
||||
Reference in New Issue
Block a user