316 lines
10 KiB
Python
316 lines
10 KiB
Python
# core/jx3jiaoyihang.py
|
|
from astrbot.api import logger
|
|
from urllib.parse import quote
|
|
from datetime import datetime
|
|
|
|
from .api_data import api_data_get, api_data_post
|
|
from .sql_data import sql_data_searchdata,sql_data_select
|
|
|
|
|
|
def fetch_all_pages(base_url, initial_params, max_pages=None):
|
|
"""
|
|
获取所有分页数据
|
|
|
|
Args:
|
|
base_url: API基础URL
|
|
initial_params: 初始请求参数
|
|
max_pages: 最大页数限制（可选）
|
|
|
|
Returns:
|
|
list: 所有页面的数据列表
|
|
"""
|
|
all_data = []
|
|
current_page = 1
|
|
|
|
while True:
|
|
# 设置当前页码
|
|
params = initial_params.copy()
|
|
params["page"] = str(current_page)
|
|
|
|
# 请求数据
|
|
data = api_data_get(base_url, params)
|
|
|
|
if not data or "list" not in data:
|
|
break
|
|
|
|
# 添加当前页数据到总列表
|
|
all_data.extend(data["list"])
|
|
|
|
# 检查是否还有更多页面
|
|
total_pages = data.get("pages", 1)
|
|
if current_page >= total_pages:
|
|
break
|
|
|
|
# 检查是否达到最大页数限制
|
|
if max_pages and current_page >= max_pages:
|
|
break
|
|
|
|
current_page += 1
|
|
|
|
return all_data
|
|
|
|
|
|
def merge_item_data_by_itemid(price_data, item_list_data):
|
|
"""
|
|
根据ItemId和id字段将物品名称信息合并到价格数据中
|
|
|
|
Args:
|
|
price_data: 第一组数据，包含价格信息的字典
|
|
item_list_data: 第二组数据，可以是字典（包含"list"键）或直接是物品列表
|
|
|
|
Returns:
|
|
dict: 合并后的数据，包含价格和名称信息
|
|
"""
|
|
# 提取价格数据中的物品信息
|
|
price_items = price_data.get("data", {})
|
|
|
|
# 处理不同类型的item_list_data输入
|
|
if isinstance(item_list_data, dict) and "list" in item_list_data:
|
|
# 如果是字典且包含"list"键
|
|
item_list = item_list_data.get("list", [])
|
|
elif isinstance(item_list_data, list):
|
|
# 如果直接是列表
|
|
item_list = item_list_data
|
|
else:
|
|
# 其他情况，使用空列表
|
|
item_list = []
|
|
|
|
# 创建一个字典用于快速查找物品名称，使用id作为键
|
|
item_name_map = {}
|
|
for item in item_list:
|
|
item_id = item.get("id")
|
|
item_name = item.get("Name")
|
|
if item_id and item_name:
|
|
item_name_map[item_id] = item_name
|
|
|
|
# 创建一个新的结果字典
|
|
merged_data = {"code": price_data.get("code", 0), "msg": price_data.get("msg", ""), "data": {}}
|
|
|
|
# 遍历价格数据，添加名称信息
|
|
for item_id, price_info in price_items.items():
|
|
# 复制价格信息
|
|
merged_item = price_info.copy()
|
|
|
|
# 使用ItemId字段查找对应的名称
|
|
item_id_to_match = merged_item.get("ItemId")
|
|
if item_id_to_match and item_id_to_match in item_name_map:
|
|
merged_item["Name"] = item_name_map[item_id_to_match]
|
|
else:
|
|
merged_item["Name"] = "未知物品"
|
|
|
|
# 添加到结果中
|
|
merged_data["data"][item_id] = merged_item
|
|
|
|
return merged_data
|
|
|
|
|
|
#交易行数据查询函数
|
|
def jx3_data_jiaoyihang(inserver="眉间雪", inname="武技殊影图"):
|
|
"""
|
|
获取剑三交易行某区某物品价格数据
|
|
|
|
Args:
|
|
inserver: 第一组数据，服务器名称
|
|
inname: 物品名称
|
|
|
|
Returns:
|
|
list: 合并后的数据，包含价格和名称信息
|
|
"""
|
|
# 第一步：获取所有物品列表数据（处理分页）
|
|
custom_url = f"https://node.jx3box.com/item_merged/name/{quote(inname)}"
|
|
initial_params = {
|
|
"client": "std",
|
|
"strict": "0",
|
|
"per": "50" # 每页数量，可以根据需要调整
|
|
}
|
|
|
|
# 获取所有页面的物品数据
|
|
all_items = fetch_all_pages(custom_url, initial_params)
|
|
|
|
if not all_items:
|
|
#logger.error(f"未找到物品: {inname}")
|
|
return "未找到改物品"
|
|
|
|
# 提取所有ID
|
|
ids = [item.get("id") for item in all_items if item.get("id")]
|
|
|
|
if not ids:
|
|
#logger.error(f"物品 {inname} 没有有效的ID")
|
|
return "未找到改物品"
|
|
|
|
# 返回格式化结果
|
|
stringids = ",".join(ids)
|
|
logger.info(f"搜索到物品: {inname}, 共找到 {len(ids)} 个物品\nIDs: {stringids}")
|
|
|
|
# 第二步：获取价格数据
|
|
price_url = "https://next2.jx3box.com/api/item-price/list"
|
|
price_params = {
|
|
"server": inserver,
|
|
"itemIds": stringids
|
|
}
|
|
|
|
price_data = api_data_get(price_url, price_params)
|
|
|
|
if not price_data or "data" not in price_data:
|
|
return "无交易行数据"
|
|
|
|
if not price_data["data"] or (isinstance(price_data["data"], dict) and not price_data["data"]) or (isinstance(price_data["data"], list) and not price_data["data"]):
|
|
return "无交易行数据"
|
|
|
|
# 第三步：合并数据
|
|
merged_data = merge_item_data_by_itemid(price_data, {"list": all_items})
|
|
|
|
# 处理合并后的数据
|
|
result_items = []
|
|
for item_id, item_info in merged_data.get("data", {}).items():
|
|
# 格式化价格（假设价格是以铜钱为单位，转换为金）
|
|
lowest_price = item_info.get("LowestPrice", 0)
|
|
bricks = lowest_price // 100000000 if lowest_price else 0
|
|
gold = (lowest_price % 100000000) // 10000 if lowest_price else 0
|
|
silver = (lowest_price % 10000) // 100 if lowest_price else 0
|
|
copper = lowest_price % 100 if lowest_price else 0
|
|
|
|
price_str = f"{bricks}砖{gold}金{silver}银{copper}铜" if lowest_price else "无价格"
|
|
|
|
result_items.append({
|
|
"name": item_info.get("Name", "未知物品"),
|
|
"price": price_str,
|
|
"avg_price": item_info.get("AvgPrice", 0),
|
|
"sample_size": item_info.get("SampleSize", 0)
|
|
})
|
|
|
|
# 按价格排序
|
|
result_items.sort(key=lambda x: x["avg_price"])
|
|
|
|
return result_items
|
|
|
|
|
|
def jx3_data_wujia(inname="秃盒"):
|
|
"""
|
|
获取剑三外观物品价格数据
|
|
|
|
Args:
|
|
inname: 物品名称
|
|
|
|
Returns:
|
|
dict: 包含价格和名称信息的字典
|
|
"""
|
|
# 初始化返回数据结构
|
|
datas = {
|
|
"code": 0,
|
|
"msg": "未获取数据",
|
|
"data": {}
|
|
}
|
|
|
|
try:
|
|
# 1. 获取物品ID和名称
|
|
idname = sql_data_select(inname)
|
|
if not idname:
|
|
datas.update({"code": 201, "msg": "未找到该外观信息"})
|
|
return datas
|
|
|
|
# 更新基础信息
|
|
datas.update({
|
|
"code": 1,
|
|
"msg": "获取外观名称ID完成",
|
|
"data": {
|
|
"showName": idname.get("showName", "未知物品"),
|
|
"searchId": idname.get("searchId", "未知ID"),
|
|
"Name": inname
|
|
}
|
|
})
|
|
|
|
# 2. 查询爱剑三获取外观数据
|
|
aj3_data = get_aj3_goods_data(datas["data"]["showName"])
|
|
if aj3_data:
|
|
datas["data"].update(aj3_data)
|
|
datas.update({"code": 2, "msg": "获取外观数据完成"})
|
|
|
|
# 3. 查询万宝楼数据（公示和在售）
|
|
wbl_data = get_wbl_data(idname['searchId'])
|
|
if wbl_data:
|
|
datas["data"].update(wbl_data)
|
|
datas.update({"code": 200, "msg": "获取万宝楼数据完成"})
|
|
|
|
except Exception as e:
|
|
logger.error(f"物价查询出错: {e}")
|
|
datas.update({"code": 500, "msg": f"查询过程中出错: {e}"})
|
|
|
|
return datas
|
|
|
|
|
|
def get_aj3_goods_data(goods_name):
|
|
"""获取爱剑三商品数据"""
|
|
try:
|
|
aj3_url = "https://www.aijx3.cn/api2/aijx3-wj/goods/getGoodsDetail"
|
|
aj3_params = {"goodsName": goods_name}
|
|
|
|
dataaj3 = api_data_post(aj3_url, aj3_params, "data")
|
|
if not dataaj3:
|
|
return None
|
|
|
|
imgs = dataaj3.get("imgs", [])
|
|
return {
|
|
"goodsDesc": dataaj3.get("goodsDesc", "无描述"),
|
|
"publishTime": dataaj3.get("publishTime", "无数据"),
|
|
"priceNum": dataaj3.get("priceNum", 0),
|
|
"imgs": imgs[0] if imgs else "",
|
|
"goodsId": dataaj3.get("goodsId", "无数据")
|
|
}
|
|
except Exception as e:
|
|
logger.error(f"获取爱剑三数据出错: {e}")
|
|
return None
|
|
|
|
|
|
def get_wbl_data(search_id):
|
|
"""获取万宝楼数据（公示和在售）"""
|
|
try:
|
|
wbl_url = "https://www.aijx3.cn/api2/aijx3-wblwg/record/queryByCondition"
|
|
base_params = {
|
|
"accoSeq": "",
|
|
"orderMode": 1,
|
|
"orderBy": "price_num",
|
|
"searchId": [search_id],
|
|
"current": 1,
|
|
"size": 10
|
|
}
|
|
|
|
# 获取公示数据
|
|
gs_params = base_params.copy()
|
|
gs_params["tradeStatus"] = "3"
|
|
datawblgs = api_data_post(wbl_url, gs_params, "data")
|
|
|
|
# 获取在售数据
|
|
zs_params = base_params.copy()
|
|
zs_params["tradeStatus"] = "5"
|
|
datawblzs = api_data_post(wbl_url, zs_params, "data")
|
|
|
|
return {
|
|
"wblgs": process_wbl_records(datawblgs.get("records", [])),
|
|
"wblzs": process_wbl_records(datawblzs.get("records", []))
|
|
}
|
|
except Exception as e:
|
|
logger.error(f"获取万宝楼数据出错: {e}")
|
|
return None
|
|
|
|
|
|
def process_wbl_records(records):
|
|
"""处理万宝楼记录数据"""
|
|
processed = []
|
|
for record in records:
|
|
try:
|
|
# 转换时间戳为可读格式
|
|
timestamp = record.get("replyTime", 0)
|
|
dt = datetime.fromtimestamp(timestamp / 1000) if timestamp else datetime.now()
|
|
|
|
processed.append({
|
|
"priceNum": record.get("priceNum", 0),
|
|
"belongQf2": record.get("belongQf2", "无数据"),
|
|
"replyTime": dt.strftime("%Y-%m-%d %H:%M:%S"),
|
|
"discountRate": record.get("discountRate", 0.0),
|
|
})
|
|
except Exception as e:
|
|
logger.error(f"处理万宝楼记录出错: {e}")
|
|
continue
|
|
|
|
return processed |