LiteyukiStudio/nonebot-plugin-marshoai
4
4
Fork 0
mirror of https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git synced 2025-01-26 18:12:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

修复完成

Browse Source
This commit is contained in:
Twisuki 2024-12-11 18:37:36 +08:00
parent 086f21272f
commit 2aba05d9ae
1 changed files with 9 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
nonebot_plugin_marshoai/tools/marshoai-meogirl/mg_Search.py
View File

@ -4,26 +4,22 @@ import re
import httpx
import urllib.parse
from bs4 import BeautifulSoup
from watchfiles import awatch
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() as client:
return await client.get(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 = ""
# 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'
# }
url = "https://mzh.moegirl.org.cn/index.php?search=" + urllib.parse.quote_plus(msg)
# client = httpx.AsyncClient()
# async with client:
# response = await client.post(url, headers = headers)
response = await get_async_data(url)
logger.info(f"连接\"{response.headers.get('location')}\"中, 状态码 : {response.status_code}")
logger.success(f"连接{url}完成, 状态码 : {response.status_code}")
# 正常搜索
if response.status_code == 200:
@ -71,11 +67,11 @@ async def search(msg : str, num : int):
# 重定向
elif response.status_code == 302:
logger.info(f"\"{msg}\"已被重定向")
logger.info(f"\"{msg}\"已被重定向至\"{response.headers.get('location')}\"")
# 读取重定向结果
response = await get_async_data(response.headers.get('location'))
soup = BeautifulSoup(response.text, 'html.parser')
logger.success("重定向成功")
logger.info(response)
num = 0
"""
@ -94,7 +90,7 @@ async def search(msg : str, num : int):
p = str(p_tag)
p = re.sub(r'<.*?>', '', p)
if p != '':
result += str(p) + "\n"
result += str(p)
num += 1
if num >= 5:
@ -105,4 +101,3 @@ async def search(msg : str, num : int):
else:
logger.error(f"网络错误, 状态码 : {response.status_code}")
return f"网络错误, 状态码 : {response.status_code}"