mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
97 lines
3.2 KiB
Python
97 lines
3.2 KiB
Python
from util.user_agent import get_user_agent
|
|
import aiohttp
|
|
from lxml import etree
|
|
from lxml.etree import Element
|
|
from configs.config import MAXINFO_BT
|
|
from urllib import parse
|
|
from html import unescape
|
|
from util.utils import get_local_proxy, is_number
|
|
import time
|
|
import platform
|
|
if platform.system() == 'Windows':
|
|
import asyncio
|
|
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
|
|
|
|
|
url = 'https://www.btmet.com/search.php'
|
|
|
|
|
|
async def get_bt_info(keyword: str, args: str, r18: str = '1') -> str:
|
|
cookiesDit = {
|
|
'r18': r18
|
|
}
|
|
s_time = time.time()
|
|
params = get_params(keyword, args)
|
|
async with aiohttp.ClientSession(headers=get_user_agent(), cookies=cookiesDit) as session:
|
|
async with session.get(url, proxy=get_local_proxy(), params=params, timeout=30) as response:
|
|
html = etree.HTML(await response.text())
|
|
print(response.url)
|
|
num = html.xpath('//div[@id="wall"]//span/b/text()')[0]
|
|
print(num)
|
|
if num.find(",") != -1:
|
|
num = num.split(',')[0]
|
|
if num == '0':
|
|
return "没有找到记录"
|
|
|
|
div_all = html.xpath('//div[@class="search-item"]')[1:]
|
|
div_all = div_all[:MAXINFO_BT] if len(div_all) > MAXINFO_BT else div_all
|
|
line_list = [await get_item_line(div) for div in div_all]
|
|
clist = []
|
|
for line in line_list:
|
|
if line.strip() != '':
|
|
clist.append(line)
|
|
|
|
return f"搜索 {keyword} 结果（共 {int(int(num.text) / 10) if int(num) % 10 == 0 else int(int(num) / 10) + 1} " \
|
|
f"页）（耗时 {int(time.time() - s_time)} 秒）：\n" + "\n\n".join(clist)
|
|
|
|
|
|
async def get_item_line(div: Element) -> str:
|
|
try:
|
|
magent = div.xpath('./div[2]/a/@href')[0]
|
|
size = div.xpath('./div[@class="f_left"]/div[@class="item-bar"]/span/b/font/text()')[0]
|
|
type = div.xpath('./div[@class="f_left"]/div[@class="item-bar"]/span[@class="cpill blue-pill"]/text()')[0].strip()
|
|
|
|
title_doc = div.xpath('.//a[@class="smashTitle"]//text()')[0]
|
|
title_code = title_doc[title_doc.find('("') + 2: title_doc.find('")')]
|
|
title_xml_code = parse.unquote(title_code)
|
|
title_xml = etree.HTML(unescape(title_xml_code))
|
|
title = title_xml.xpath('string(.)')
|
|
except Exception:
|
|
return ''
|
|
return "【{}】| {}\n【{}】| {}".format(type, title, size, magent)
|
|
|
|
|
|
# https://www.btmet.com/search.php?q=%E9%92%A2%E9%93%81%E4%BE%A0&c=5&o=0&l=&p=2
|
|
def get_params(keyword: str, args: str) -> dict:
|
|
params = {
|
|
'q': keyword,
|
|
'c': '',
|
|
'l': '',
|
|
'o': 0,
|
|
'p': ''
|
|
}
|
|
if not args:
|
|
return params
|
|
args = args.split(" ")
|
|
for arg in args:
|
|
if '-U' == arg.upper():
|
|
params['o'] = 1
|
|
if '-S' == arg.upper():
|
|
params['o'] = 2
|
|
if '-H' == arg.upper():
|
|
params['o'] = 3
|
|
if '-V' == arg.upper():
|
|
params['c'] = 1
|
|
if '-P' == arg.upper():
|
|
params['c'] = 2
|
|
if '-A' == arg.upper():
|
|
params['c'] = 5
|
|
if is_number(arg):
|
|
params['p'] = arg
|
|
return params
|
|
|
|
|
|
# print(asyncio.get_event_loop().run_until_complete(get_bt_info('钢铁侠', '')))
|
|
|
|
|