#!/usr/bin/env python3
import sys
sys.path.append('backend')

from app.utils.text_parser import BrandExtractor

def test_brand_extraction():
    """测试修复后的品牌提取逻辑"""
    extractor = BrandExtractor()
    
    test_cases = [
        "Tods女款蓝色鳄鱼压纹麻花扣乐福鞋XXW79A0DD00WENU824代购8.15LA",
        "小王国S MaxMara女款深灰色羊毛双排扣短大衣 CAPE 002代购8.18",
        "SportMax女款黑色羊毛大衣 CANOA 代购8.20",
        "ST JOHN女款白色针织上衣 STJ001代购8.22",
        "Stuart Weitzman女款黑色高跟鞋 SW123代购8.24"
    ]
    
    print("=== 品牌提取测试 ===")
    for product_name in test_cases:
        candidates = extractor._extract_brand_candidates(product_name)
        extracted_brand = extractor.extract_brand(product_name)
        
        print(f"\n产品名称: {product_name}")
        print(f"候选品牌: {candidates}")
        print(f"提取品牌: {extracted_brand}")

if __name__ == "__main__":
    test_brand_extraction()