mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git
synced 2025-08-02 05:19:50 +00:00
✨ 更新 mypy 版本至 1.13.0，重构 marshoai-megakits 模块并添加随机数和计算功能
This commit is contained in:
16
nonebot_plugin_marshoai/tools/marshoai_meogirl/__init__.py
Normal file
16
nonebot_plugin_marshoai/tools/marshoai_meogirl/__init__.py
Normal file
@ -0,0 +1,16 @@
|
||||
from . import mg_Info, mg_Introduce, mg_Search
|
||||
|
||||
|
||||
# meogirl
|
||||
async def meogirl():
|
||||
return mg_Info.meogirl()
|
||||
|
||||
|
||||
# Search
|
||||
async def search(msg: str, num: int = 3):
|
||||
return str(await mg_Search.search(msg, num))
|
||||
|
||||
|
||||
# Show
|
||||
async def introduce(msg: str):
|
||||
return str(await mg_Introduce.introduce(msg))
|
||||
@ -0,0 +1,3 @@
|
||||
# Meogirl
|
||||
def meogirl():
|
||||
return 'Meogirl指的是"萌娘百科"(https://zh.moegirl.org.cn/ , 简称"萌百"), 是一个"万物皆可萌的百科全书！"; 同时, MarshoTools也配有"Meogirl"插件, 可调用萌百的api'
|
||||
@ -0,0 +1,81 @@
|
||||
import re
|
||||
import urllib.parse
|
||||
|
||||
import httpx
|
||||
from bs4 import BeautifulSoup
|
||||
from nonebot.log import logger
|
||||
|
||||
headers = {
|
||||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/100.0.4896.127 Safari/537.36"
|
||||
}
|
||||
|
||||
|
||||
async def get_async_data(url):
|
||||
async with httpx.AsyncClient(timeout=None) as client:
|
||||
return await client.get(url, headers=headers)
|
||||
|
||||
|
||||
async def introduce(msg: str):
|
||||
logger.info(f'介绍 : "{msg}" ...')
|
||||
result = ""
|
||||
|
||||
url = "https://mzh.moegirl.org.cn/" + urllib.parse.quote_plus(msg)
|
||||
response = await get_async_data(url)
|
||||
logger.success(f'连接"{url}"完成, 状态码 : {response.status_code}')
|
||||
|
||||
soup = BeautifulSoup(response.text, "html.parser")
|
||||
|
||||
# 正常页
|
||||
if response.status_code == 200:
|
||||
"""
|
||||
萌娘百科页面结构
|
||||
div#mw-content-text
|
||||
└── div#404search # 空白页面出现
|
||||
└── div.mw-parser-output # 正常页面
|
||||
└── div, p, table ... # 大量的解释项
|
||||
"""
|
||||
result += msg + "\n"
|
||||
|
||||
img = soup.find("img", class_="infobox-image")
|
||||
if img:
|
||||
result += f"![ {msg} ]( {img["src"]} ) \n"
|
||||
|
||||
div = soup.find("div", class_="mw-parser-output")
|
||||
if div:
|
||||
p_tags = div.find_all("p")
|
||||
num = 0
|
||||
for p_tag in p_tags:
|
||||
p = str(p_tag)
|
||||
p = re.sub(
|
||||
r"<script.*?</script>|<style.*?</style>", "", p, flags=re.DOTALL
|
||||
)
|
||||
p = re.sub(r"<.*?>", "", p, flags=re.DOTALL)
|
||||
p = re.sub(r"\[.*?]", "", p, flags=re.DOTALL)
|
||||
|
||||
if p != "":
|
||||
result += str(p)
|
||||
|
||||
num += 1
|
||||
if num >= 20:
|
||||
break
|
||||
return result
|
||||
|
||||
# 空白页
|
||||
elif response.status_code == 404:
|
||||
logger.info(f'未找到"{msg}", 进行搜索')
|
||||
|
||||
from . import mg_Search
|
||||
|
||||
context = await mg_Search.search(msg, 1)
|
||||
keyword = re.search(r".*?\n", context, flags=re.DOTALL).group()[:-1] # type: ignore
|
||||
|
||||
logger.success(f'搜索完成, 打开"{keyword}"')
|
||||
return await introduce(keyword)
|
||||
|
||||
# 搜索失败
|
||||
elif response.status_code == 301:
|
||||
return f"未找到{msg}"
|
||||
|
||||
else:
|
||||
logger.error(f"网络错误, 状态码 : {response.status_code}")
|
||||
return f"网络错误, 状态码 : {response.status_code}"
|
||||
85
nonebot_plugin_marshoai/tools/marshoai_meogirl/mg_Search.py
Normal file
85
nonebot_plugin_marshoai/tools/marshoai_meogirl/mg_Search.py
Normal file
@ -0,0 +1,85 @@
|
||||
import urllib.parse
|
||||
|
||||
import httpx
|
||||
from bs4 import BeautifulSoup
|
||||
from nonebot.log import logger
|
||||
|
||||
headers = {
|
||||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/100.0.4896.127 Safari/537.36"
|
||||
}
|
||||
|
||||
|
||||
async def get_async_data(url):
|
||||
async with httpx.AsyncClient(timeout=None) as client:
|
||||
return await client.get(url, headers=headers)
|
||||
|
||||
|
||||
async def search(msg: str, num: int):
|
||||
logger.info(f'搜索 : "{msg}" ...')
|
||||
result = ""
|
||||
|
||||
url = "https://mzh.moegirl.org.cn/index.php?search=" + urllib.parse.quote_plus(msg)
|
||||
response = await get_async_data(url)
|
||||
logger.success(f'连接"{url}"完成, 状态码 : {response.status_code}')
|
||||
|
||||
# 正常搜索
|
||||
if response.status_code == 200:
|
||||
"""
|
||||
萌娘百科搜索页面结构
|
||||
div.searchresults
|
||||
└── p ...
|
||||
└── ul.mw-search-results # 若无, 证明无搜索结果
|
||||
└── li # 一个搜索结果
|
||||
└── div.mw-search-result-heading > a # 标题
|
||||
└── div.mw-searchresult # 内容
|
||||
└── div.mw-search-result-data
|
||||
└── li ...
|
||||
└── li ...
|
||||
"""
|
||||
soup = BeautifulSoup(response.text, "html.parser")
|
||||
|
||||
# 检测ul.mw-search-results, 是否有结果
|
||||
ul_tag = soup.find("ul", class_="mw-search-results")
|
||||
if ul_tag:
|
||||
li_tags = ul_tag.find_all("li")
|
||||
for li_tag in li_tags:
|
||||
|
||||
div_heading = li_tag.find("div", class_="mw-search-result-heading")
|
||||
if div_heading:
|
||||
a_tag = div_heading.find("a")
|
||||
result += a_tag["title"] + "\n"
|
||||
logger.info(f'搜索到 : "{a_tag["title"]}"')
|
||||
|
||||
div_result = li_tag.find("div", class_="searchresult")
|
||||
if div_result:
|
||||
content = (
|
||||
str(div_result)
|
||||
.replace('<div class="searchresult">', "")
|
||||
.replace("</div>", "")
|
||||
)
|
||||
content = content.replace('<span class="searchmatch">', "").replace(
|
||||
"</span>", ""
|
||||
)
|
||||
result += content + "\n"
|
||||
|
||||
num -= 1
|
||||
if num == 0:
|
||||
break
|
||||
return result
|
||||
|
||||
# 无ul.mw-search-results, 无结果
|
||||
else:
|
||||
logger.info("无结果")
|
||||
return "无结果"
|
||||
|
||||
# 重定向
|
||||
elif response.status_code == 302:
|
||||
logger.info(f'"{msg}"已被重定向至"{response.headers.get("location")}"')
|
||||
# 读取重定向结果
|
||||
from . import mg_Introduce
|
||||
|
||||
return await mg_Introduce.introduce(msg)
|
||||
|
||||
else:
|
||||
logger.error(f"网络错误, 状态码 : {response.status_code}")
|
||||
return f"网络错误, 状态码 : {response.status_code}"
|
||||
51
nonebot_plugin_marshoai/tools/marshoai_meogirl/tools.json
Normal file
51
nonebot_plugin_marshoai/tools/marshoai_meogirl/tools.json
Normal file
@ -0,0 +1,51 @@
|
||||
[
|
||||
{
|
||||
"type" : "function",
|
||||
"function" : {
|
||||
"name" : "marshoai-meogirl__meogirl",
|
||||
"description" : "介绍Meogirl"
|
||||
}
|
||||
},
|
||||
{
|
||||
"type": "function",
|
||||
"function": {
|
||||
"name": "marshoai-meogirl__search",
|
||||
"description": "查找/搜索 某角色/事物 (使用萌娘百科)",
|
||||
"parameters": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"msg": {
|
||||
"type": "string",
|
||||
"description": "搜索关键词"
|
||||
},
|
||||
"num": {
|
||||
"type": "integer",
|
||||
"description": "数据显示条数, 默认3, 可留空"
|
||||
}
|
||||
}
|
||||
},
|
||||
"required": [
|
||||
"msg"
|
||||
]
|
||||
}
|
||||
},
|
||||
{
|
||||
"type" : "function",
|
||||
"function" : {
|
||||
"name" : "marshoai-meogirl__introduce",
|
||||
"description" : "介绍/展示 某角色/事物 (使用萌娘百科)",
|
||||
"parameters" : {
|
||||
"type" : "object",
|
||||
"properties" : {
|
||||
"msg" : {
|
||||
"type": "string",
|
||||
"description": "关键词"
|
||||
}
|
||||
}
|
||||
},
|
||||
"required": [
|
||||
"msg"
|
||||
]
|
||||
}
|
||||
}
|
||||
]
|
||||
Reference in New Issue
Block a user