mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
295 lines
14 KiB
Python
295 lines
14 KiB
Python
from models.buff_price import BuffPrice
|
||
from services.db_context import db
|
||
from datetime import datetime, timedelta
|
||
from util.user_agent import get_user_agent
|
||
from configs.path_config import IMAGE_PATH
|
||
import aiohttp
|
||
import aiofiles
|
||
from models.open_cases_user import OpenCasesUser
|
||
import os
|
||
from services.log import logger
|
||
from util.utils import get_bot
|
||
from models.group_remind import GroupRemind
|
||
from util.utils import get_cookie_text
|
||
from asyncio.exceptions import TimeoutError
|
||
import pypinyin
|
||
from nonebot.adapters.cqhttp.exception import ActionFailed
|
||
<<<<<<< HEAD
|
||
from configs.config import buff_proxy
|
||
=======
|
||
>>>>>>> master
|
||
|
||
url = "https://buff.163.com/api/market/goods"
|
||
# proxies = 'http://49.75.59.242:3128'
|
||
|
||
|
||
async def util_get_buff_price(case_name: str = "狂牙大行动") -> str:
|
||
cookie = {'session': get_cookie_text('buff')}
|
||
failed_list = []
|
||
case = ""
|
||
for i in pypinyin.pinyin(case_name, style=pypinyin.NORMAL):
|
||
case += ''.join(i)
|
||
if case_name == "狂牙大行动":
|
||
case_id = 1
|
||
elif case_name == "突围大行动":
|
||
case_id = 2
|
||
elif case_name == "命悬一线":
|
||
case_id = 3
|
||
elif case_name == '裂空':
|
||
case_id = 4
|
||
elif case_name == '光谱':
|
||
case_id = 5
|
||
else:
|
||
return "未查询到武器箱"
|
||
case = case.upper()
|
||
CASE_KNIFE = eval(case + "_CASE_KNIFE")
|
||
CASE_RED = eval(case + "_CASE_RED")
|
||
CASE_PINK = eval(case + "_CASE_PINK")
|
||
CASE_PURPLE = eval(case + "_CASE_PURPLE")
|
||
CASE_BLUE = eval(case + "_CASE_BLUE")
|
||
async with aiohttp.ClientSession(cookies=cookie, headers=get_user_agent()) as session:
|
||
for total_list in [CASE_KNIFE, CASE_RED, CASE_PINK, CASE_PURPLE, CASE_BLUE]:
|
||
print("----------------------------------")
|
||
for skin in total_list:
|
||
print(skin)
|
||
if skin in ["蝴蝶刀 | 无涂装", '求生匕首 | 无涂装', '流浪者匕首 | 无涂装', '系绳匕首 | 无涂装', '骷髅匕首 | 无涂装']:
|
||
skin = skin.split('|')[0].strip()
|
||
async with db.transaction():
|
||
name_list = []
|
||
price_list = []
|
||
parameter = {
|
||
"game": "csgo",
|
||
"page_num": "1",
|
||
"search": skin
|
||
}
|
||
try:
|
||
<<<<<<< HEAD
|
||
async with session.get(url, proxy=buff_proxy, params=parameter, timeout=20) as response:
|
||
=======
|
||
async with session.get(url, params=parameter, timeout=20) as response:
|
||
>>>>>>> master
|
||
if response.status == 200:
|
||
data = (await response.json())["data"]
|
||
total_page = data["total_page"]
|
||
data = data["items"]
|
||
flag = False
|
||
if skin.find('|') == -1: # in ['蝴蝶刀', '求生匕首', '流浪者匕首', '系绳匕首', '骷髅匕首']:
|
||
for i in range(1, total_page + 1):
|
||
name_list = []
|
||
price_list = []
|
||
parameter = {
|
||
"game": "csgo",
|
||
"page_num": f"{i}",
|
||
"search": skin
|
||
}
|
||
async with session.get(url, params=parameter, timeout=20) as res:
|
||
data = (await response.json())["data"]["items"]
|
||
for j in range(len(data)):
|
||
if data[j]['name'] in [f'{skin}（★）']:
|
||
name = data[j]["name"]
|
||
price = data[j]["sell_reference_price"]
|
||
name_list.append(name.split('（')[0].strip() + ' | 无涂装')
|
||
price_list.append(price)
|
||
print(name_list[-1])
|
||
print(price_list[-1])
|
||
flag = True
|
||
break
|
||
if flag:
|
||
break
|
||
else:
|
||
try:
|
||
for _ in range(total_page):
|
||
for i in range(len(data)):
|
||
name = data[i]["name"]
|
||
price = data[i]["sell_reference_price"]
|
||
name_list.append(name)
|
||
price_list.append(price)
|
||
except Exception as e:
|
||
failed_list.append(skin)
|
||
print(f"{skin}更新失败")
|
||
else:
|
||
failed_list.append(skin)
|
||
print(f"{skin}更新失败")
|
||
except Exception:
|
||
failed_list.append(skin)
|
||
print(f"{skin}更新失败")
|
||
continue
|
||
for i in range(len(name_list)):
|
||
name = name_list[i].strip()
|
||
price = float(price_list[i])
|
||
if name.find("（★）") != -1:
|
||
name = name[: name.find("（")] + name[name.find("）") + 1:]
|
||
if name.find("消音") != -1 and name.find("（S") != -1:
|
||
name = name.split("（")[0][:-4] + "（" + name.split("（")[1]
|
||
name = name.split("|")[0].strip() + " | " + name.split("|")[1].strip()
|
||
elif name.find("消音") != -1:
|
||
name = name.split("|")[0][:-5].strip() + " | " + name.split("|")[1].strip()
|
||
if name.find(" 18 ") != -1 and name.find("（S") != -1:
|
||
name = name.split("（")[0][:-5] + "（" + name.split("（")[1]
|
||
name = name.split("|")[0].strip() + " | " + name.split("|")[1].strip()
|
||
elif name.find(" 18 ") != -1:
|
||
name = name.split("|")[0][:-6].strip() + " | " + name.split("|")[1].strip()
|
||
dbskin = await BuffPrice.ensure(name, True)
|
||
if (dbskin.update_date + timedelta(8)).date() == datetime.now().date():
|
||
continue
|
||
await dbskin.update(
|
||
case_id=case_id,
|
||
skin_price=price,
|
||
update_date=datetime.now(),
|
||
).apply()
|
||
print(f"{name_list[i]}---------->成功更新")
|
||
result = None
|
||
if failed_list:
|
||
result = ""
|
||
for fail_skin in failed_list:
|
||
result += fail_skin + "\n"
|
||
return result[:-1] if result else "更新价格成功"
|
||
|
||
|
||
async def util_get_buff_img(case_name: str = "狂牙大行动") -> str:
|
||
cookie = {'session': get_cookie_text('buff')}
|
||
error_list = []
|
||
case = ""
|
||
for i in pypinyin.pinyin(case_name, style=pypinyin.NORMAL):
|
||
case += ''.join(i)
|
||
path = "cases/" + case + "/"
|
||
if not os.path.exists(IMAGE_PATH + path):
|
||
os.mkdir(IMAGE_PATH + path)
|
||
case = case.upper()
|
||
CASE_KNIFE = eval(case + "_CASE_KNIFE")
|
||
CASE_RED = eval(case + "_CASE_RED")
|
||
CASE_PINK = eval(case + "_CASE_PINK")
|
||
CASE_PURPLE = eval(case + "_CASE_PURPLE")
|
||
CASE_BLUE = eval(case + "_CASE_BLUE")
|
||
async with aiohttp.ClientSession(cookies=cookie, headers=get_user_agent()) as session:
|
||
for total_list in [CASE_KNIFE, CASE_RED, CASE_PINK, CASE_PURPLE, CASE_BLUE]:
|
||
for skin in total_list:
|
||
parameter = {
|
||
"game": "csgo",
|
||
"page_num": "1",
|
||
"search": skin
|
||
}
|
||
if skin in ["蝴蝶刀 | 无涂装", '求生匕首 | 无涂装', '流浪者匕首 | 无涂装', '系绳匕首 | 无涂装', '骷髅匕首 | 无涂装']:
|
||
skin = skin.split('|')[0].strip()
|
||
print("开始更新----->", skin)
|
||
print(skin)
|
||
skin_name = ''
|
||
# try:
|
||
<<<<<<< HEAD
|
||
async with session.get(url, proxy=buff_proxy, params=parameter, timeout=20) as response:
|
||
=======
|
||
async with session.get(url, params=parameter, timeout=20) as response:
|
||
>>>>>>> master
|
||
if response.status == 200:
|
||
data = (await response.json())["data"]
|
||
total_page = data["total_page"]
|
||
flag = False
|
||
if skin.find('|') == -1: # in ['蝴蝶刀', '求生匕首', '流浪者匕首', '系绳匕首', '骷髅匕首']:
|
||
for i in range(1, total_page + 1):
|
||
async with session.get(url, params=parameter, timeout=20) as res:
|
||
data = (await response.json())["data"]["items"]
|
||
for j in range(len(data)):
|
||
if data[j]['name'] in [f'{skin}（★）']:
|
||
img_url = data[j]['goods_info']['icon_url']
|
||
for k in pypinyin.pinyin(skin + '无涂装', style=pypinyin.NORMAL):
|
||
skin_name += ''.join(k)
|
||
async with aiofiles.open(IMAGE_PATH + path + skin_name + ".png", 'wb') as f:
|
||
print("------->开始写入 ", skin)
|
||
async with session.get(img_url, timeout=7) as res:
|
||
await f.write(await res.read())
|
||
flag = True
|
||
break
|
||
if flag:
|
||
break
|
||
else:
|
||
img_url = (await response.json())["data"]['items'][0]['goods_info']['icon_url']
|
||
for i in pypinyin.pinyin(skin.replace('|', '-').strip(), style=pypinyin.NORMAL):
|
||
skin_name += ''.join(i)
|
||
async with aiofiles.open(IMAGE_PATH + path + skin_name + ".png", 'wb') as f:
|
||
print("------->开始写入 ", skin)
|
||
async with session.get(img_url, timeout=7) as res:
|
||
await f.write(await res.read())
|
||
# async with session.get(url, params=parameter, timeout=7) as response:
|
||
# if response.status == 200:
|
||
# img_url = (await response.json())["data"]['items'][0]['goods_info']['icon_url']
|
||
# skin_name = ''
|
||
# for i in pypinyin.pinyin(skin.split("|")[1].strip(), style=pypinyin.NORMAL):
|
||
# skin_name += ''.join(i)
|
||
# async with aiofiles.open(IMAGE_PATH + path + skin_name + ".png", 'wb') as f:
|
||
# print("------->开始写入 ", skin)
|
||
# async with session.get(img_url, timeout=7) as res:
|
||
# await f.write(await res.read())
|
||
# except Exception:
|
||
# print("图片更新失败 ---->", skin)
|
||
# error_list.append(skin)
|
||
result = None
|
||
if error_list:
|
||
result = ""
|
||
for errskin in error_list:
|
||
result += errskin + "\n"
|
||
return result[:-1] if result else "更新图片成功"
|
||
|
||
|
||
async def get_price(dname):
|
||
cookie = {'session': get_cookie_text('buff')}
|
||
name_list = []
|
||
price_list = []
|
||
parameter = {
|
||
"game": "csgo",
|
||
"page_num": "1",
|
||
"search": dname
|
||
}
|
||
try:
|
||
async with aiohttp.ClientSession(cookies=cookie, headers=get_user_agent()) as session:
|
||
async with session.get(url, params=parameter, timeout=7) as response:
|
||
if response.status == 200:
|
||
try:
|
||
data = (await response.json())["data"]
|
||
total_page = data["total_page"]
|
||
data = data["items"]
|
||
for _ in range(total_page):
|
||
for i in range(len(data)):
|
||
name = data[i]["name"]
|
||
price = data[i]["sell_reference_price"]
|
||
name_list.append(name)
|
||
price_list.append(price)
|
||
except Exception as e:
|
||
return "没有查询到...", 998
|
||
else:
|
||
return "访问失败！", response.status
|
||
except TimeoutError as e:
|
||
return "访问超时! 请重试或稍后再试!", 997
|
||
result = f"皮肤: {dname}({len(name_list)})\n"
|
||
# result = "皮肤: " + dname + "\n"
|
||
for i in range(len(name_list)):
|
||
result += name_list[i] + ": " + price_list[i] + "\n"
|
||
return result[:-1], 999
|
||
|
||
|
||
async def update_count_daily():
|
||
try:
|
||
users = await OpenCasesUser.get_user_all()
|
||
if users:
|
||
for user in users:
|
||
await user.update(
|
||
today_open_total=0,
|
||
).apply()
|
||
bot = get_bot()
|
||
gl = await bot.get_group_list(self_id=bot.self_id)
|
||
gl = [g['group_id'] for g in gl]
|
||
for g in gl:
|
||
if await GroupRemind.get_status(g, 'kxcz'):
|
||
try:
|
||
await bot.send_group_msg(group_id=g, message="今日开箱次数重置成功")
|
||
except ActionFailed:
|
||
logger.warning(f'{g} 群被禁言，无法发送 开箱重置提醒')
|
||
logger.info("今日开箱次数重置成功")
|
||
except Exception as e:
|
||
logger.error(f'开箱重置错误 e:{e}')
|
||
|
||
|
||
|
||
# 蝴蝶刀（★） | 噩梦之夜 (久经沙场)
|
||
if __name__ == '__main__':
|
||
print(util_get_buff_img("xxxx/"))
|