zhenxun_bot/plugins/open_cases/utils.py

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/"))