mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
345 lines
14 KiB
Python
Executable File
345 lines
14 KiB
Python
Executable File
from asyncio.exceptions import TimeoutError
|
||
from datetime import datetime
|
||
from typing import Optional
|
||
|
||
import nonebot
|
||
|
||
from configs.config import Config
|
||
from configs.path_config import IMAGE_PATH
|
||
from services.log import logger
|
||
from utils.http_utils import AsyncHttpx
|
||
from utils.utils import broadcast_group, cn2py
|
||
|
||
from .config import *
|
||
from .models.buff_prices import BuffPrice
|
||
from .models.buff_skin import BuffSkin
|
||
from .models.open_cases_user import OpenCasesUser
|
||
|
||
url = "https://buff.163.com/api/market/goods"
|
||
# proxies = 'http://49.75.59.242:3128'
|
||
|
||
driver = nonebot.get_driver()
|
||
|
||
|
||
async def util_get_buff_price(case_name: str = "狂牙大行动") -> str:
|
||
cookie = {"session": Config.get_config("open_cases", "COOKIE")}
|
||
failed_list = []
|
||
case = cn2py(case_name)
|
||
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")
|
||
for total_list in [CASE_KNIFE, CASE_RED, CASE_PINK, CASE_PURPLE, CASE_BLUE]:
|
||
for skin in total_list:
|
||
if skin in [
|
||
"蝴蝶刀 | 无涂装",
|
||
"求生匕首 | 无涂装",
|
||
"流浪者匕首 | 无涂装",
|
||
"系绳匕首 | 无涂装",
|
||
"骷髅匕首 | 无涂装",
|
||
]:
|
||
skin = skin.split("|")[0].strip()
|
||
name_list = []
|
||
price_list = []
|
||
parameter = {"game": "csgo", "page_num": "1", "search": skin}
|
||
try:
|
||
response = await AsyncHttpx.get(
|
||
url,
|
||
proxy=Config.get_config("open_cases", "BUFF_PROXY"),
|
||
params=parameter,
|
||
cookies=cookie,
|
||
)
|
||
if response.status_code == 200:
|
||
data = 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,
|
||
}
|
||
res = await AsyncHttpx.get(
|
||
url, params=parameter, cookies=cookie
|
||
)
|
||
data = res.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)
|
||
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)
|
||
logger.warning(f"{skin}更新失败")
|
||
else:
|
||
failed_list.append(skin)
|
||
logger.warning(f"{skin}更新失败")
|
||
except Exception:
|
||
failed_list.append(skin)
|
||
logger.warning(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()
|
||
)
|
||
if dbskin := await BuffPrice.get_or_none(skin_name=name):
|
||
if dbskin.update_date.date() == datetime.now().date():
|
||
continue
|
||
dbskin.case_id = case_id
|
||
dbskin.skin_price = price
|
||
dbskin.update_date = datetime.now()
|
||
await dbskin.save(
|
||
update_fields=["case_id", "skin_price", "update_date"]
|
||
)
|
||
logger.info(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": Config.get_config("open_cases", "COOKIE")}
|
||
error_list = []
|
||
case = cn2py(case_name)
|
||
path = IMAGE_PATH / "cases/" / case
|
||
path.mkdir(exist_ok=True, parents=True)
|
||
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")
|
||
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()
|
||
logger.info(f"开始更新----->{skin}")
|
||
skin_name = ""
|
||
# try:
|
||
response = await AsyncHttpx.get(
|
||
url,
|
||
proxy=Config.get_config("open_cases", "BUFF_PROXY"),
|
||
params=parameter,
|
||
)
|
||
if response.status_code == 200:
|
||
data = response.json()["data"]
|
||
total_page = data["total_page"]
|
||
flag = False
|
||
if skin.find("|") == -1: # in ['蝴蝶刀', '求生匕首', '流浪者匕首', '系绳匕首', '骷髅匕首']:
|
||
for i in range(1, total_page + 1):
|
||
res = await AsyncHttpx.get(url, params=parameter)
|
||
data = res.json()["data"]["items"]
|
||
for j in range(len(data)):
|
||
if data[j]["name"] in [f"{skin}（★）"]:
|
||
img_url = data[j]["goods_info"]["icon_url"]
|
||
skin_name = cn2py(skin + "无涂装")
|
||
await AsyncHttpx.download_file(
|
||
img_url, path / f"{skin_name}.png"
|
||
)
|
||
flag = True
|
||
break
|
||
if flag:
|
||
break
|
||
else:
|
||
img_url = (await response.json())["data"]["items"][0]["goods_info"][
|
||
"icon_url"
|
||
]
|
||
skin_name += cn2py(skin.replace("|", "-").strip())
|
||
if await AsyncHttpx.download_file(
|
||
img_url, path / f"{skin_name}.png"
|
||
):
|
||
logger.info(f"------->写入 {skin} 成功")
|
||
else:
|
||
logger.info(f"------->写入 {skin} 失败")
|
||
result = None
|
||
if error_list:
|
||
result = ""
|
||
for err_skin in error_list:
|
||
result += err_skin + "\n"
|
||
return result[:-1] if result else "更新图片成功"
|
||
|
||
|
||
async def get_price(d_name):
|
||
cookie = {"session": Config.get_config("open_cases", "COOKIE")}
|
||
name_list = []
|
||
price_list = []
|
||
parameter = {"game": "csgo", "page_num": "1", "search": d_name}
|
||
try:
|
||
response = await AsyncHttpx.get(url, cookies=cookie, params=parameter)
|
||
if response.status_code == 200:
|
||
try:
|
||
data = 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_code
|
||
except TimeoutError as e:
|
||
return "访问超时! 请重试或稍后再试!", 997
|
||
result = f"皮肤: {d_name}({len(name_list)})\n"
|
||
for i in range(len(name_list)):
|
||
result += name_list[i] + ": " + price_list[i] + "\n"
|
||
return result[:-1], 999
|
||
|
||
|
||
async def reset_count_daily():
|
||
"""
|
||
重置每日开箱
|
||
"""
|
||
try:
|
||
await OpenCasesUser.all().update(today_open_total=0)
|
||
await broadcast_group(
|
||
"[[_task|open_case_reset_remind]]今日开箱次数重置成功", log_cmd="开箱重置提醒"
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"开箱重置错误", e=e)
|
||
|
||
|
||
def get_color(case_name: str, name: str, skin_name: str) -> Optional[str]:
|
||
case_py = cn2py(case_name).upper()
|
||
color_map = {}
|
||
color_map["KNIFE"] = eval(case_py + "_CASE_KNIFE")
|
||
color_map["RED"] = eval(case_py + "_CASE_RED")
|
||
color_map["PINK"] = eval(case_py + "_CASE_PINK")
|
||
color_map["PURPLE"] = eval(case_py + "_CASE_PURPLE")
|
||
color_map["BLUE"] = eval(case_py + "_CASE_BLUE")
|
||
for key in color_map:
|
||
for skin in color_map[key]:
|
||
if name in skin and skin_name in skin:
|
||
return key
|
||
return None
|
||
|
||
|
||
@driver.on_startup
|
||
async def _():
|
||
"""
|
||
将旧表数据移动到新表
|
||
"""
|
||
if not await BuffSkin.first() and await BuffPrice.first():
|
||
logger.debug("开始移动旧表数据 BuffPrice -> BuffSkin")
|
||
id2name = {1: "狂牙大行动", 2: "突围大行动", 3: "命悬一线", 4: "裂空", 5: "光谱"}
|
||
data_list: List[BuffSkin] = []
|
||
for data in await BuffPrice.all():
|
||
logger.debug(f"移动旧表数据: {data.skin_name}")
|
||
case_name = id2name[data.case_id]
|
||
name = data.skin_name
|
||
is_stattrak = "StatTrak" in name
|
||
name = name.replace("（★ StatTrak™）", "").replace("（StatTrak™）", "").strip()
|
||
name, skin_name = name.split("|")
|
||
abrasion = "无涂装"
|
||
if "(" in skin_name:
|
||
skin_name, abrasion = skin_name.split("(")
|
||
if abrasion.endswith(")"):
|
||
abrasion = abrasion[:-1]
|
||
color = get_color(case_name, name.strip(), skin_name.strip())
|
||
if not color:
|
||
search_list = [
|
||
x
|
||
for x in data_list
|
||
if x.skin_name == skin_name.strip() and x.name == name.strip()
|
||
]
|
||
if search_list:
|
||
color = get_color(
|
||
case_name, search_list[0].name, search_list[0].skin_name
|
||
)
|
||
if not color:
|
||
logger.debug(
|
||
f"箱子: [{case_name}] 皮肤: [{name}|{skin_name}] 未获取到皮肤品质，跳过..."
|
||
)
|
||
continue
|
||
data_list.append(
|
||
BuffSkin(
|
||
case_name=case_name,
|
||
name=name.strip(),
|
||
skin_name=skin_name.strip(),
|
||
is_stattrak=is_stattrak,
|
||
abrasion=abrasion.strip(),
|
||
skin_price=data.skin_price,
|
||
color=color,
|
||
create_time=datetime.now(),
|
||
update_time=datetime.now(),
|
||
)
|
||
)
|
||
await BuffSkin.bulk_create(data_list, batch_size=10)
|
||
logger.debug("完成移动旧表数据 BuffPrice -> BuffSkin")
|
||
|
||
|
||
# 蝴蝶刀（★） | 噩梦之夜 (久经沙场)
|
||
if __name__ == "__main__":
|
||
print(util_get_buff_img("xxxx/"))
|