Jeason
400569ad03
cart_service: 对没有itemID的预售商品,点击图片触发跳转 从跳转URL中提取真实itemID(支持多种URL格式) 点击后自动关闭新tab并回到购物车继续处理下一个商品
239 lines
8.0 KiB
Python
239 lines
8.0 KiB
Python
"""
|
||
购物车预售商品抓取服务
|
||
通过 Playwright 打开购物车页面，从 DOM 提取商品信息，
|
||
并通过点击商品图片获取跳转 URL 来提取真实 itemID。
|
||
"""
|
||
import asyncio
|
||
import re
|
||
from playwright.async_api import async_playwright
|
||
from utils.stealth import stealth_async
|
||
from server.services.auth_service import get_browser_context, has_auth
|
||
|
||
CART_URL = "https://weidian.com/new-cart/index.php"
|
||
|
||
# 从 DOM 提取购物车商品基本信息
|
||
EXTRACT_JS = """() => {
|
||
const R = [];
|
||
const sws = document.querySelectorAll(
|
||
'div.shop_info.cart_content div.shop_warp'
|
||
);
|
||
for (const sw of sws) {
|
||
const sn = (sw.querySelector('.shop_name') || {}).textContent || '';
|
||
const iws = sw.querySelectorAll('.item_warp');
|
||
for (const iw of iws) {
|
||
const o = {
|
||
shop_name: sn.trim(),
|
||
cart_item_id: iw.id,
|
||
item_id: '',
|
||
title: '', sku_name: '', price: '',
|
||
is_presale: false, countdown_text: '',
|
||
sale_time: '', presale_type: ''
|
||
};
|
||
|
||
// 尝试多种方式提取 itemID
|
||
// 1. Vue 组件数据
|
||
try {
|
||
const vue = iw.__vue__;
|
||
if (vue) {
|
||
const d = vue.$data || vue._data || vue;
|
||
o.item_id = String(d.itemID || d.itemId || d.item_id
|
||
|| d.goodsId || d.goods_id || '');
|
||
// 也检查 props
|
||
if (!o.item_id && vue.$props) {
|
||
const pp = vue.$props;
|
||
o.item_id = String(pp.itemID || pp.itemId
|
||
|| pp.item_id || pp.goodsId || '');
|
||
}
|
||
// 检查 item 对象
|
||
if (!o.item_id && d.item) {
|
||
o.item_id = String(d.item.itemID || d.item.itemId
|
||
|| d.item.item_id || '');
|
||
}
|
||
}
|
||
} catch(e) {}
|
||
|
||
// 2. data-* 属性
|
||
if (!o.item_id) {
|
||
o.item_id = iw.dataset.itemId || iw.dataset.itemid
|
||
|| iw.dataset.goodsId || iw.dataset.id || '';
|
||
}
|
||
|
||
// 3. 内部链接
|
||
if (!o.item_id) {
|
||
const a = iw.querySelector('a[href*="itemID"]');
|
||
if (a) {
|
||
const m = a.href.match(/itemID=(\\d+)/);
|
||
if (m) o.item_id = m[1];
|
||
}
|
||
}
|
||
|
||
// 4. 图片 URL 中可能有商品信息（备用）
|
||
const img = iw.querySelector('.item_img img');
|
||
o.img_src = img ? (img.src || img.dataset.src || '') : '';
|
||
|
||
const te = iw.querySelector('.item_title');
|
||
if (te) o.title = te.textContent.trim();
|
||
const sk = iw.querySelector('.item_sku');
|
||
if (sk) o.sku_name = sk.textContent.trim();
|
||
const pr = iw.querySelector('.item_prices');
|
||
if (pr) o.price = pr.textContent.replace(/[^\\d.]/g, '');
|
||
const de = iw.querySelector('.item_desc');
|
||
if (de) {
|
||
const dt = de.querySelector('.title');
|
||
const dd = de.querySelector('.desc');
|
||
const wm = de.querySelector('.warn_msg');
|
||
if (dt && /\\u5b9a\\u65f6\\s*\\u5f00\\u552e/.test(dt.textContent)) {
|
||
o.is_presale = true;
|
||
const d = dd ? dd.textContent.trim() : '';
|
||
const w = wm ? wm.textContent.trim() : '';
|
||
if (d.includes('\\u8ddd\\u79bb\\u5f00\\u552e\\u8fd8\\u5269')) {
|
||
o.presale_type = 'countdown';
|
||
o.countdown_text = w;
|
||
} else if (d.includes('\\u5f00\\u552e\\u65f6\\u95f4')) {
|
||
o.presale_type = 'scheduled';
|
||
o.sale_time = w;
|
||
} else {
|
||
o.presale_type = 'unknown';
|
||
o.countdown_text = w;
|
||
}
|
||
}
|
||
}
|
||
R.push(o);
|
||
}
|
||
}
|
||
return R;
|
||
}"""
|
||
|
||
|
||
async def fetch_cart_presale_items(account_id):
|
||
"""
|
||
获取指定账号购物车中的预售商品列表。
|
||
对于没有 itemID 的商品，通过点击图片获取跳转 URL 来提取。
|
||
返回: (success, items_or_msg)
|
||
"""
|
||
if not has_auth(account_id):
|
||
return False, "账号未登录"
|
||
|
||
async with async_playwright() as p:
|
||
browser, context = await get_browser_context(
|
||
p, account_id, headless=True
|
||
)
|
||
page = await context.new_page()
|
||
await stealth_async(page)
|
||
|
||
try:
|
||
await page.goto(
|
||
CART_URL, wait_until="networkidle", timeout=20000
|
||
)
|
||
await asyncio.sleep(3)
|
||
|
||
if "login" in page.url.lower():
|
||
await browser.close()
|
||
return False, "登录态已过期，请重新登录"
|
||
|
||
if "error" in page.url.lower():
|
||
await browser.close()
|
||
return False, "购物车页面加载失败"
|
||
|
||
except Exception as e:
|
||
await browser.close()
|
||
return False, f"打开购物车失败: {e}"
|
||
|
||
# 提取基本信息
|
||
raw_items = await page.evaluate(EXTRACT_JS)
|
||
|
||
# 筛选预售商品
|
||
presale = [it for it in raw_items if it.get("is_presale")]
|
||
|
||
# 对没有 itemID 的预售商品，通过点击图片获取跳转 URL
|
||
for item in presale:
|
||
if item.get('item_id'):
|
||
continue
|
||
cid = item.get('cart_item_id', '')
|
||
if not cid:
|
||
continue
|
||
|
||
item_id = await _get_item_id_by_click(page, context, cid)
|
||
if item_id:
|
||
item['item_id'] = item_id
|
||
item['url'] = f'https://weidian.com/item.html?itemID={item_id}'
|
||
|
||
await browser.close()
|
||
|
||
return True, presale
|
||
|
||
|
||
async def _get_item_id_by_click(page, context, cart_item_id):
|
||
"""
|
||
通过点击购物车中商品的图片，拦截跳转 URL 来提取 itemID。
|
||
点击后会打开新 tab，从新 tab 的 URL 中提取 itemID，然后关闭。
|
||
"""
|
||
try:
|
||
# 定位商品图片
|
||
img_locator = page.locator(
|
||
f'.item_warp[id="{cart_item_id}"] .item_img')
|
||
if await img_locator.count() == 0:
|
||
return None
|
||
|
||
# 监听新页面打开事件
|
||
async with context.expect_page(timeout=5000) as new_page_info:
|
||
await img_locator.first.click()
|
||
|
||
new_page = await new_page_info.value
|
||
# 等待 URL 加载
|
||
await asyncio.sleep(1)
|
||
url = new_page.url
|
||
|
||
# 从 URL 提取 itemID
|
||
item_id = _extract_item_id_from_url(url)
|
||
|
||
# 关闭新 tab
|
||
await new_page.close()
|
||
|
||
return item_id
|
||
|
||
except Exception:
|
||
# 如果 expect_page 超时，可能是在当前页面跳转了
|
||
# 检查当前页面 URL
|
||
try:
|
||
current_url = page.url
|
||
item_id = _extract_item_id_from_url(current_url)
|
||
if item_id:
|
||
# 跳回购物车
|
||
await page.goto(CART_URL, wait_until="networkidle",
|
||
timeout=15000)
|
||
await asyncio.sleep(2)
|
||
return item_id
|
||
except Exception:
|
||
pass
|
||
|
||
# 确保回到购物车页面
|
||
try:
|
||
if 'new-cart' not in page.url:
|
||
await page.goto(CART_URL, wait_until="networkidle",
|
||
timeout=15000)
|
||
await asyncio.sleep(2)
|
||
except Exception:
|
||
pass
|
||
|
||
return None
|
||
|
||
|
||
def _extract_item_id_from_url(url):
|
||
"""从 URL 中提取 itemID"""
|
||
if not url:
|
||
return None
|
||
# https://weidian.com/item.html?itemID=123456
|
||
m = re.search(r'itemID=(\d+)', url, re.IGNORECASE)
|
||
if m:
|
||
return m.group(1)
|
||
# https://shop.weidian.com/item/123456
|
||
m = re.search(r'/item/(\d+)', url)
|
||
if m:
|
||
return m.group(1)
|
||
# https://weidian.com/...?id=123456
|
||
m = re.search(r'[?&]id=(\d+)', url)
|
||
if m:
|
||
return m.group(1)
|
||
return None
|