zhenxun_bot/plugins/open_cases/utils.py

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