first_commit .gitignore

plugins/search_buff_skin_price/__init__.py

@@ -0,0 +1,75 @@
+from nonebot import on_command
+from .data_source import get_price, update_buff_cookie
+from services.log import logger
+from nonebot.typing import T_State
+from nonebot.adapters import Bot, Event
+from nonebot.rule import to_me
+from nonebot.permission import SUPERUSER
+from util.utils import UserExistLimiter, get_message_text
+
+
+__plugin_name__ = '查询皮肤'
+__plugin_usage__ = '查询皮肤帮助:\n\t' \
+                   '对我说 "查询皮肤 xxx yyyy"，我会回复xxx的底价哦\n\t' \
+                   '示例: 查询皮肤 awp 二西莫夫'
+
+
+_ulmt = UserExistLimiter()
+
+
+search_skin = on_command('查询皮肤', aliases={'皮肤查询'}, priority=5, block=True)
+
+
+@search_skin.args_parser
+async def parse(bot: Bot, event: Event, state: T_State):
+    if get_message_text(event.json()) in ['取消', '算了']:
+        await search_skin.finish("已取消操作..", at_sender=True)
+    state[state["_current_key"]] = str(event.get_message())
+
+
+@search_skin.handle()
+async def _(bot: Bot, event: Event, state: T_State):
+    if str(event.get_message()) in ['帮助']:
+        await search_skin.finish(__plugin_usage__)
+    raw_arg = get_message_text(event.json())
+    if _ulmt.check(event.user_id):
+        await search_skin.finish('您有皮肤正在搜索，请稍等...', at_sender=True)
+    if raw_arg:
+        args = raw_arg.split(" ")
+        if len(args) >= 2:
+            state['name'] = args[0]
+            state['skin'] = args[1]
+
+
+@search_skin.got('name', prompt="要查询什么武器呢？")
+@search_skin.got('skin', prompt="要查询该武器的什么皮肤呢？")
+async def arg_handle(bot: Bot, event: Event, state: T_State):
+    _ulmt.set_True(event.user_id)
+    if state['name'] in ['ak', 'ak47']:
+        state['name'] = 'ak-47'
+    name = state['name'] + " | " + state['skin']
+    try:
+        result, status_code = await get_price(name)
+    except FileNotFoundError:
+        await search_skin.finish('请先对真寻说"设置cookie"来设置cookie！')
+    if status_code in [996, 997, 998]:
+        _ulmt.set_False(event.user_id)
+        await search_skin.finish(result)
+    if result:
+        logger.info(
+            f"(USER {event.user_id}, GROUP {event.group_id if event.message_type != 'private' else 'private'}) 查询皮肤:" + name)
+        _ulmt.set_False(event.user_id)
+        await search_skin.finish(result)
+    else:
+        logger.info(f"USER {event.user_id}, GROUP {event.group_id if event.message_type != 'private' else 'private'}"
+                    f" 查询皮肤：{name} 没有查询到")
+        _ulmt.set_False(event.user_id)
+        await search_skin.finish("没有查询到哦，请检查格式吧")
+
+
+update_buff_session = on_command("更新cookie", rule=to_me(), permission=SUPERUSER, priority=1)
+
+
+@update_buff_session.handle()
+async def _(bot: Bot, event: Event, state: T_State):
+    await update_buff_session.finish(update_buff_cookie(str(event.get_message())), at_sender=True)

plugins/search_buff_skin_price/data_source.py

@@ -0,0 +1,66 @@
+from util.user_agent import get_user_agent
+import aiohttp
+from util.utils import get_cookie_text
+from configs.path_config import TXT_PATH
+from asyncio.exceptions import TimeoutError
+from configs.config import buff_proxy
+from pathlib import Path
+from services.log import logger
+
+
+url = "https://buff.163.com/api/market/goods"
+
+
+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, proxy=buff_proxy, params=parameter, timeout=5) as response:
+                if response.status == 200:
+                    try:
+                        if str(await response.text()).find("Login Required") != -1:
+                            return "BUFF登录被重置，请联系管理员重新登入", 996
+                        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
+
+
+def update_buff_cookie(cookie: str):
+    _cookie = Path(TXT_PATH + "cookie/buff.txt")
+    try:
+        _cookie.parent.mkdir(parents=True, exist_ok=True)
+        with open(_cookie, 'w') as f:
+            f.write(cookie)
+        return "更新cookie成功"
+    except Exception as e:
+        logger.error(f'更新cookie失败 e:{e}')
+        return "更新cookie失败"
+
+
+if __name__ == '__main__':
+    print(get_price("awp 二西莫夫"))
+