zhenxun_bot/plugins/open_cases/utils.py

from .models.buff_prices import BuffPrice
from services.db_context import db
from datetime import datetime, timedelta
from configs.path_config import IMAGE_PATH
from utils.http_utils import AsyncHttpx
from .models.open_cases_user import OpenCasesUser
import os
from services.log import logger
from utils.utils import get_bot
from asyncio.exceptions import TimeoutError
from nonebot.adapters.cqhttp.exception import ActionFailed
from configs.config import Config
from utils.manager import group_manager
from .config import *

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": Config.get_config("open_cases", "COOKIE")}
    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")
    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()
            async with db.transaction():
                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)
                                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()
                        )
                    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()
                    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 = ""
    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")
    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"]
                                for k in pypinyin.pinyin(
                                    skin + "无涂装", style=pypinyin.NORMAL
                                ):
                                    skin_name += "".join(k)
                                await AsyncHttpx.download_file(img_url, IMAGE_PATH + path + skin_name + ".png")
                                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)
                    if await AsyncHttpx.download_file(img_url, IMAGE_PATH + path + 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 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()
        gl = [g["group_id"] for g in gl]
        for g in gl:
            if await group_manager.check_group_task_status(g, "open_case_reset_remind"):
                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/"))