qrpa 1.0.65__py3-none-any.whl → 1.0.68__py3-none-any.whl

qrpa/mysql_module/shein_product_model.py

@@ -0,0 +1,476 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+SHEIN商品数据模型
+使用SQLAlchemy定义SKC表、SKU表和详情表结构
+"""
+
+from sqlalchemy import create_engine, Column, Integer, String, Text, DateTime, JSON, Date, DECIMAL, Index
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+from datetime import datetime
+import json
+import os
+
+# 创建基类
+Base = declarative_base()
+
+class SheinProductSkc(Base):
+    """
+    SHEIN商品SKC表
+    存储SKC维度的商品信息
+    """
+    __tablename__ = 'shein_product_skc'
+
+    # 主键ID
+    id = Column(Integer, primary_key=True, autoincrement=True, comment='主键ID')
+
+    # 原始数据中的ID作为skc_id
+    skc_id = Column(Integer, nullable=False, unique=True, comment='原始SKC ID')
+
+    # 店铺信息
+    store_username = Column(String(50), nullable=True, comment='店铺账号')
+    store_name = Column(String(100), nullable=True, comment='店铺名称')
+    store_manager = Column(String(50), nullable=True, comment='店长')
+
+    # 商品基本信息
+    pic_url = Column(Text, nullable=True, comment='SKC主图')
+    supplier_code = Column(String(100), nullable=True, comment='商家SKC')
+    skc = Column(String(50), nullable=True, comment='平台SKC')
+    spu = Column(String(50), nullable=True, comment='SPU编码')
+    category_name = Column(String(100), nullable=True, comment='叶子类目')
+
+    # 上架信息
+    shelf_days = Column(Integer, nullable=True, comment='上架天数')
+    shelf_date = Column(Date, nullable=True, comment='上架日期')
+
+    # 商品标签和状态
+    goods_label_list = Column(JSON, nullable=True, comment='商品标签列表')
+    supply_status = Column(String(50), nullable=True, comment='供货状态')
+    shelf_status = Column(Integer, nullable=True, comment='上架状态')
+    goods_level = Column(String(50), nullable=True, comment='商品等级')
+
+    # 其他字段
+    group_flag = Column(Integer, nullable=True, comment='组合标志')
+    goods_level_can_order_flag = Column(Integer, nullable=True, comment='商品等级可下单标志')
+    stock_standard = Column(Integer, nullable=True, comment='备货标准')
+    stock_warn_status = Column(Integer, nullable=True, comment='库存预警状态')
+
+    # 销量汇总
+    c7d_sale_cnt_sum = Column(Integer, nullable=True, comment='7日销量汇总')
+    shein_sale_by_inventory = Column(Integer, nullable=True, comment='希音库存销售')
+
+    # 时间戳
+    created_at = Column(DateTime, default=datetime.now, comment='创建时间')
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now, comment='更新时间')
+
+    # 定义索引
+    __table_args__ = (
+        Index('ix_skc_id', 'skc_id'),
+        Index('ix_skc', 'skc'),
+        Index('ix_spu', 'spu'),
+        Index('ix_supplier_code', 'supplier_code'),
+        Index('ix_store_username', 'store_username'),
+    )
+
+    def __repr__(self):
+        return f"<SheinProductSkc(id={self.id}, skc='{self.skc}', spu='{self.spu}')>"
+
+class SheinProductSku(Base):
+    """
+    SHEIN商品SKU表
+    存储SKU维度的商品信息
+    """
+    __tablename__ = 'shein_product_sku'
+
+    # 主键ID
+    id = Column(Integer, primary_key=True, autoincrement=True, comment='主键ID')
+
+    # 原始数据中的ID作为sku_id
+    sku_id = Column(Integer, nullable=False, unique=True, comment='原始SKU ID')
+
+    # 关联SKC本地主键ID
+    local_skc_id = Column(Integer, nullable=False, comment='关联的SKC本地主键ID')
+
+    # SKU基本信息
+    sort_value = Column(Integer, nullable=True, comment='排序值')
+    sku_code = Column(String(50), nullable=True, comment='SKU编码')
+    attr = Column(String(100), nullable=True, comment='SKU属性')
+    supplier_sku = Column(String(100), nullable=True, comment='供应商SKU')
+
+    # 销量数据
+    order_cnt = Column(Integer, nullable=True, comment='今日订单数')
+    total_sale_volume = Column(Integer, nullable=True, comment='今日总销量')
+    c7d_sale_cnt = Column(Integer, nullable=True, comment='7日销量')
+    c30d_sale_cnt = Column(Integer, nullable=True, comment='30日销量')
+
+    # 价格成本信息
+    price = Column(DECIMAL(10, 2), nullable=True, comment='价格')
+    erp_cost_price = Column(DECIMAL(10, 2), nullable=True, comment='ERP成本价')
+    erp_supplier_name = Column(String(100), nullable=True, comment='ERP默认供货商')
+
+    # 时间戳
+    created_at = Column(DateTime, default=datetime.now, comment='创建时间')
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now, comment='更新时间')
+
+    # 定义索引
+    __table_args__ = (
+        Index('ix_sku_id', 'sku_id'),
+        Index('ix_local_skc_id', 'local_skc_id'),
+        Index('ix_sku_code', 'sku_code'),
+        Index('ix_supplier_sku', 'supplier_sku'),
+    )
+
+    def __repr__(self):
+        return f"<SheinProductSku(id={self.id}, sku_code='{self.sku_code}', attr='{self.attr}')>"
+
+class SheinProductDetail(Base):
+    """
+    SHEIN商品详情信息表
+    以KV形式保存SPU维度的JSON数据
+    """
+    __tablename__ = 'shein_product_detail'
+
+    # 主键ID
+    id = Column(Integer, primary_key=True, autoincrement=True, comment='主键ID')
+
+    # SPU编码
+    spu = Column(String(50), nullable=False, comment='SPU编码')
+
+    # 字段名称
+    field_name = Column(String(100), nullable=False, comment='字段名称')
+
+    # JSON数据
+    json_data = Column(JSON, nullable=True, comment='JSON数据')
+
+    # 时间戳
+    created_at = Column(DateTime, default=datetime.now, comment='创建时间')
+    updated_at = Column(DateTime, default=datetime.now, onupdate=datetime.now, comment='更新时间')
+
+    # 定义索引
+    __table_args__ = (
+        Index('ix_spu', 'spu'),
+        Index('ix_field_name', 'field_name'),
+        Index('ix_spu_field', 'spu', 'field_name'),
+    )
+
+    def __repr__(self):
+        return f"<SheinProductDetail(id={self.id}, spu='{self.spu}', field_name='{self.field_name}')>"
+
+class SheinProductManager:
+    """
+    SHEIN商品数据管理器
+    提供数据库操作相关方法
+    """
+
+    def __init__(self, database_url):
+        """
+        初始化数据库连接
+        
+        Args:
+            database_url (str): 数据库连接URL，例如：
+                mysql+pymysql://username:password@localhost:3306/database_name
+        """
+        self.engine = create_engine(database_url, echo=False)
+        self.Session = sessionmaker(bind=self.engine)
+
+    def create_tables(self):
+        """
+        创建数据表
+        """
+        Base.metadata.create_all(self.engine)
+        print("数据表创建成功！")
+
+    def drop_tables(self):
+        """
+        删除数据表
+        """
+        Base.metadata.drop_all(self.engine)
+        print("数据表删除成功！")
+
+    def _parse_date(self, date_str):
+        """
+        解析日期字符串
+        """
+        if not date_str:
+            return None
+        try:
+            return datetime.strptime(date_str, '%Y-%m-%d').date()
+        except:
+            return None
+
+    def _extract_goods_labels(self, goods_label_list):
+        """
+        提取商品标签名称列表
+        """
+        if not goods_label_list:
+            return []
+        return [label.get('name', '') for label in goods_label_list if isinstance(label, dict)]
+
+    def upsert_product_data(self, data_list):
+        """
+        从JSON数据中执行upsert操作（插入或更新）
+        
+        Args:
+            store_username (str): 店铺账号
+            data_list (list): 商品数据列表
+        """
+        session = self.Session()
+        try:
+            for data in data_list:
+                
+                # 处理SKC数据
+                skc_record = self._upsert_skc_data(session, data)
+                
+                # 处理SKU数据
+                for sku_data in data.get('skuList', []):
+                    sku_data['local_skc_id'] = skc_record.id
+                    self._upsert_sku_data(session, sku_data)
+
+            session.commit()
+            print(f"成功处理 {len(data_list)} 条商品数据")
+
+        except Exception as e:
+            session.rollback()
+            print(f"处理数据失败: {e}")
+            raise
+        finally:
+            session.close()
+
+    def _upsert_skc_data(self, session, data):
+        """
+        插入或更新SKC数据
+        返回SKC记录对象
+        """
+        skc_id = data.get('id')
+        existing_skc = session.query(SheinProductSkc).filter_by(skc_id=skc_id).first()
+
+        if existing_skc:
+            # 更新现有记录（保留用户备注）
+            existing_skc.store_username = data.get('store_username')
+            existing_skc.store_name = data.get('store_name')
+            existing_skc.store_manager = data.get('store_manager')
+            existing_skc.pic_url = data.get('picUrl')
+            existing_skc.supplier_code = data.get('supplierCode')
+            existing_skc.skc = data.get('skc')
+            existing_skc.spu = data.get('spu')
+            existing_skc.category_name = data.get('categoryName')
+            existing_skc.shelf_days = data.get('shelfDays')
+            existing_skc.shelf_date = self._parse_date(data.get('shelfDate'))
+            existing_skc.goods_label_list = self._extract_goods_labels(data.get('goodsLabelList'))
+            existing_skc.supply_status = data.get('supplyStatus', {}).get('name') if data.get('supplyStatus') else None
+            existing_skc.shelf_status = data.get('shelfStatus', {}).get('value') if data.get('shelfStatus') else None
+            existing_skc.goods_level = data.get('goodsLevel', {}).get('name') if data.get('goodsLevel') else None
+            existing_skc.group_flag = data.get('groupFlag')
+            existing_skc.goods_level_can_order_flag = 1 if data.get('goodsLevelCanOrderFlag') else 0
+            existing_skc.stock_standard = data.get('stockStandard', {}).get('value') if data.get('stockStandard') else None
+            existing_skc.stock_warn_status = data.get('stockWarnStatus', {}).get('value') if data.get('stockWarnStatus') else None
+            existing_skc.c7d_sale_cnt_sum = data.get('c7dSaleCntSum')
+            existing_skc.shein_sale_by_inventory = data.get('sheinSaleByInventory')
+            existing_skc.updated_at = datetime.now()
+            return existing_skc
+        else:
+            # 插入新记录
+            new_skc = SheinProductSkc(
+                skc_id=skc_id,
+                store_username=data.get('store_username'),
+                store_name=data.get('store_name'),
+                store_manager=data.get('store_manager'),
+                pic_url=data.get('picUrl'),
+                supplier_code=data.get('supplierCode'),
+                skc=data.get('skc'),
+                spu=data.get('spu'),
+                category_name=data.get('categoryName'),
+                shelf_days=data.get('shelfDays'),
+                shelf_date=self._parse_date(data.get('shelfDate')),
+                goods_label_list=self._extract_goods_labels(data.get('goodsLabelList')),
+                supply_status=data.get('supplyStatus', {}).get('name') if data.get('supplyStatus') else None,
+                shelf_status=data.get('shelfStatus', {}).get('value') if data.get('shelfStatus') else None,
+                goods_level=data.get('goodsLevel', {}).get('name') if data.get('goodsLevel') else None,
+                group_flag=data.get('groupFlag'),
+                goods_level_can_order_flag=1 if data.get('goodsLevelCanOrderFlag') else 0,
+                stock_standard=data.get('stockStandard', {}).get('value') if data.get('stockStandard') else None,
+                stock_warn_status=data.get('stockWarnStatus', {}).get('value') if data.get('stockWarnStatus') else None,
+                c7d_sale_cnt_sum=data.get('c7dSaleCntSum'),
+                shein_sale_by_inventory=data.get('sheinSaleByInventory')
+            )
+            session.add(new_skc)
+            session.flush()  # 获取主键ID
+            return new_skc
+
+    def _upsert_sku_data(self, session, sku_data):
+        """
+        插入或更新SKU数据
+        """
+        sku_id = sku_data.get('id')
+        existing_sku = session.query(SheinProductSku).filter_by(sku_id=sku_id).first()
+
+        if existing_sku:
+            # 更新现有记录（保留用户备注）
+            existing_sku.local_skc_id = sku_data.get('local_skc_id')
+            existing_sku.sort_value = sku_data.get('sortValue')
+            existing_sku.sku_code = sku_data.get('skuCode')
+            existing_sku.attr = sku_data.get('attr')
+            existing_sku.supplier_sku = sku_data.get('supplierSku')
+            existing_sku.order_cnt = sku_data.get('orderCnt')
+            existing_sku.total_sale_volume = sku_data.get('totalSaleVolume')
+            existing_sku.c7d_sale_cnt = sku_data.get('c7dSaleCnt')
+            existing_sku.c30d_sale_cnt = sku_data.get('c30dSaleCnt')
+            existing_sku.price = sku_data.get('price')
+            existing_sku.erp_cost_price = sku_data.get('erp_cost_price')
+            existing_sku.erp_supplier_name = sku_data.get('erp_supplier_name')
+            existing_sku.updated_at = datetime.now()
+        else:
+            # 插入新记录
+            new_sku = SheinProductSku(
+                sku_id=sku_id,
+                local_skc_id=sku_data.get('local_skc_id'),
+                sort_value=sku_data.get('sortValue'),
+                sku_code=sku_data.get('skuCode'),
+                attr=sku_data.get('attr'),
+                supplier_sku=sku_data.get('supplierSku'),
+                order_cnt=sku_data.get('orderCnt'),
+                total_sale_volume=sku_data.get('totalSaleVolume'),
+                c7d_sale_cnt=sku_data.get('c7dSaleCnt'),
+                c30d_sale_cnt=sku_data.get('c30dSaleCnt'),
+                price=sku_data.get('price'),
+                erp_cost_price=sku_data.get('erp_cost_price'),
+                erp_supplier_name=sku_data.get('erp_supplier_name')
+            )
+            session.add(new_sku)
+
+    def upsert_product_detail(self, spu, field_name, json_data):
+        """
+        插入或更新商品详情数据
+        
+        Args:
+            spu (str): SPU编码
+            field_name (str): 字段名称
+            json_data (dict): JSON数据
+        """
+        session = self.Session()
+        try:
+            existing_detail = session.query(SheinProductDetail).filter_by(
+                spu=spu, field_name=field_name
+            ).first()
+
+            if existing_detail:
+                # 更新现有记录
+                existing_detail.json_data = json_data
+                existing_detail.updated_at = datetime.now()
+            else:
+                # 插入新记录
+                new_detail = SheinProductDetail(
+                    spu=spu,
+                    field_name=field_name,
+                    json_data=json_data
+                )
+                session.add(new_detail)
+
+            session.commit()
+            print(f"成功处理详情数据: spu={spu}, field_name={field_name}")
+
+        except Exception as e:
+            session.rollback()
+            print(f"处理详情数据失败: {e}")
+            raise
+        finally:
+            session.close()
+
+    def get_skc_products(self, limit=None, offset=None):
+        """
+        查询SKC商品列表
+        
+        Args:
+            limit (int): 限制返回数量
+            offset (int): 偏移量
+            
+        Returns:
+            list: SKC商品列表
+        """
+        session = self.Session()
+        try:
+            query = session.query(SheinProductSkc)
+            if offset:
+                query = query.offset(offset)
+            if limit:
+                query = query.limit(limit)
+            return query.all()
+        finally:
+            session.close()
+
+    def search_products(self, **kwargs):
+        """
+        根据条件搜索商品
+        
+        Args:
+            **kwargs: 搜索条件，如skc, spu, store_username等
+            
+        Returns:
+            list: 符合条件的商品列表
+        """
+        session = self.Session()
+        try:
+            query = session.query(SheinProductSkc)
+
+            # 根据传入的条件进行过滤
+            for key, value in kwargs.items():
+                if hasattr(SheinProductSkc, key) and value is not None:
+                    if isinstance(value, str) and key in ['skc', 'spu', 'supplier_code', 'category_name']:
+                        # 字符串字段支持模糊搜索
+                        query = query.filter(getattr(SheinProductSkc, key).like(f'%{value}%'))
+                    else:
+                        query = query.filter(getattr(SheinProductSkc, key) == value)
+
+            return query.all()
+        finally:
+            session.close()
+
+def example_usage():
+    """
+    使用示例
+    """
+    # 数据库连接URL（请根据实际情况修改）
+    database_url = "mysql+pymysql://root:123wyk@localhost:3306/lz"
+
+    # 创建管理器实例
+    manager = SheinProductManager(database_url)
+
+    # 创建数据表
+    manager.create_tables()
+
+    # 从JSON文件导入数据
+    json_file = "skc_list_GS4355889.json"
+
+    # 读取JSON文件
+    with open(json_file, 'r', encoding='utf-8') as f:
+        data_dict = json.load(f)
+        for store_username, data_list in data_dict.items():
+            manager.upsert_product_data(data_list)
+
+    # 查询示例
+    products = manager.get_skc_products(limit=10)
+    for product in products:
+        print(f"SKC: {product.skc}, SPU: {product.spu}, 类目: {product.category_name}")
+
+    # 搜索示例
+    search_results = manager.search_products(store_username="GS4355889", shelf_status=1)
+    print(f"已上架商品数量: {len(search_results)}")
+
+    # 详情数据示例
+    # manager.upsert_product_detail("i250319906447", "attribute_template", {"template_id": "t23082580827"})
+
+def example_usage2():
+    database_url = "mysql+pymysql://root:123wyk@localhost:3306/lz"
+    manager = SheinProductManager(database_url)
+    manager.create_tables()
+    # 从JSON文件导入数据
+    json_file = "attribute_template_t23082580827.json"
+    # 读取JSON文件
+    with open(json_file, 'r', encoding='utf-8') as f:
+        data_dict = json.load(f)
+        manager.upsert_product_detail("t23082580827", "attribute_template", data_dict)
+
+if __name__ == "__main__":
+    pass
+    # example_usage()

qrpa/shein_excel.py

@@ -281,11 +281,11 @@ class SheinExcel:
         cache_file_excel = f'{self.config.auto_dir}/shein/cache/shein_return_order_list_excel_{start_date}_{end_date}.json'
         write_dict_to_file(cache_file_excel, excel_data)
 
-        sheet_name = '希音退货列表'
-        batch_excel_operations(self.config.excel_return_list, [
-            (sheet_name, 'write', excel_data, ['W', 'Z']),
-            (sheet_name, 'format', self.format_return_list)
-        ])
+        # sheet_name = '希音退货列表'
+        # batch_excel_operations(self.config.excel_return_list, [
+        #     (sheet_name, 'write', excel_data, ['W', 'Z']),
+        #     (sheet_name, 'format', self.format_return_list)
+        # ])
 
         excel_data = [header]
         cache_file = f'{self.config.auto_dir}/shein/cache/shein_return_order_list_{TimeUtils.today_date()}.json'
@@ -381,7 +381,8 @@ class SheinExcel:
 
         batch_excel_operations(self.config.excel_return_list, [
             (sheet_name, 'write', excel_data, ['W', 'Z']),
-            (sheet_name, 'format', self.format_return_list)
+            (sheet_name, 'format', self.format_return_list),
+            ('Sheet1', 'delete')
         ])
 
     def format_return_list(self, sheet):

qrpa/shein_lib.py

@@ -48,7 +48,7 @@ class SheinLib:
 
                     if web_page.locator('xpath=//div[@id="container" and @alita-name="gmpsso"]//button[@type="button" and @id]').nth(0).is_visible():
                         if 'https://sso.geiwohuo.com/#/home' not in web_page.url:
-                            log("鉴权确定按钮可见 点击'确定'按钮", self.store_username, self.store_name)
+                            log("鉴权确定按钮可见 点击'确定'按钮", web_page.title(), web_page.url, self.store_username, self.store_name)
                             web_page.locator('xpath=//div[@id="container" and @alita-name="gmpsso"]//button[@type="button" and @id]').nth(0).click()
                             web_page.wait_for_load_state("load")
                             web_page.wait_for_timeout(1000)
@@ -91,6 +91,10 @@ class SheinLib:
                         return
 
                     log('商家后台不可见', web_page.title(), web_page.url, self.store_username, self.store_name)
+                    if 'https://sso.geiwohuo.com/#/home' in web_page.url:
+                        web_page.wait_for_timeout(3000)
+                        web_page.reload()
+                        web_page.wait_for_load_state("load")
 
                     while r'=/CN' in web_page.url:
                         web_page.goto('https://sso.geiwohuo.com')
@@ -105,7 +109,6 @@ class SheinLib:
                             log('再延时5秒', self.store_username, self.store_name)
                             web_page.wait_for_timeout(5000)
 
-                    web_page.wait_for_load_state("load")
                     web_page.wait_for_timeout(1000)
 
                     if 'https://sso.geiwohuo.com/#/home' in web_page.url:
@@ -231,10 +234,11 @@ class SheinLib:
 
         url = f"https://sso.geiwohuo.com/pfmp/returnOrder/page"
         payload = {
-            "addTimeStart": f"{start_date} 00:00:00",
-            "addTimeEnd"  : f"{end_date} 23:59:59",
-            "page"        : page_num,
-            "perPage"     : page_size
+            "returnOrderType": 1,  # 只查询退货
+            "addTimeStart"   : f"{start_date} 00:00:00",
+            "addTimeEnd"     : f"{end_date} 23:59:59",
+            "page"           : page_num,
+            "perPage"        : page_size
         }
         response_text = fetch(self.web_page, url, payload)
         error_code = response_text.get('code')
@@ -256,30 +260,31 @@ class SheinLib:
         today_list_item = []
         # 过滤 退货出库时间 是昨天的
         for item in list_item:
-            has_valid_package = item.get('hasPackage') == 1
-            is_valid_yesterday = TimeUtils.is_yesterday(item['completeTime'], None) if item.get('completeTime') else False
             returnOrderId = item['id']
-            item['return_box_detail'] = []
-            item['qc_report_url'] = ''
-            item['report_url'] = ''
             item['store_username'] = self.store_username
             item['store_name'] = self.store_name
             item['store_manager'] = self.config.shein_store_manager.get(str(self.store_username).lower())
-            if has_valid_package:
-                return_box_detail = self.get_return_order_box_detail(returnOrderId)
-                if len(return_box_detail) > 0:
 
-                    if int(item['returnScrapType']) == 1:
-                        purchaseCode = item['sellerOrderNo']
-                        delivery_code = item['sellerDeliveryNo']
-                        item['qc_report_url'] = self.get_qc_report_url(delivery_code, purchaseCode)
+            item['qc_report_url'] = ''
+            if int(item['returnScrapType']) == 1:
+                purchaseCode = item['sellerOrderNo']
+                delivery_code = item['sellerDeliveryNo']
+                item['qc_report_url'] = self.get_qc_report_url(delivery_code, purchaseCode)
+
+            item['report_url'] = ''
+            if int(item['returnScrapType']) == 2:
+                item['report_url'] = self.get_inspect_report_url(returnOrderId)
 
-                    if int(item['returnScrapType']) == 2:
-                        item['report_url'] = self.get_inspect_report_url(returnOrderId)
+            item['return_box_detail'] = []
 
+            all_list_item.append(item)
+            has_valid_package = item.get('hasPackage') == 1
+            if has_valid_package:
+                return_box_detail = self.get_return_order_box_detail(returnOrderId)
+                if len(return_box_detail) > 0:
                     item['return_box_detail'] = return_box_detail
 
-                    all_list_item.append(item)
+                    is_valid_yesterday = TimeUtils.is_yesterday(item['completeTime'], None) if item.get('completeTime') else False
                     if is_valid_yesterday:
                         today_list_item.append(item)
 
@@ -631,17 +636,102 @@ class SheinLib:
         log(f'正在获取 {self.store_name} 最近一个月出库金额: {last_item["totalCustomerAmount"]}')
         return last_item['totalCustomerAmount']
 
+    def get_product_attr(self, spu, attr_name):
+        try:
+            product_detail = self.get_product_detail(spu)
+            product_type_id = product_detail.get('product_type_id')
+            category_id = product_detail.get('category_id')
+
+            if not product_type_id or not category_id:
+                return None  # 或者根据需要返回一个默认值
+
+            attribute_template = self.get_attribute_templates(spu, category_id, [product_type_id])
+            attr_info = attribute_template.get('attribute_infos', [])
+
+            # 查找材质属性映射，防止没有匹配项
+            attr_item = next((item for item in attr_info if item.get('attribute_name') == attr_name), None)
+            if not attr_item:
+                return None  # 或者返回一个默认值
+
+            attr_id = attr_item.get('attribute_id')
+
+            # 拿到产品材质的属性值ID
+            product_attribute_list = product_detail.get('product_attribute_list', [])
+            attribute_value_id = next((item['attribute_value_id'] for item in product_attribute_list if item.get('attribute_id') == attr_id), None)
+            if not attribute_value_id:
+                return None  # 或者返回一个默认值
+
+            # 获取属性值名称
+            attr_value = next((item['attribute_value'] for item in attr_item.get('attribute_value_info_list', []) if item.get('attribute_value_id') == attribute_value_id), None)
+            return attr_value  # 返回找到的属性值
+        except Exception as e:
+            log(f"Error occurred: {e}")
+            send_exception()
+            return None  # 或者返回一个默认值
+
+    def get_attribute_templates(self, spu_name, category_id, product_type_id_list):
+        log(f'正在获取 {spu_name} 商品属性模板')
+
+        if not isinstance(product_type_id_list, list):
+            raise '参数错误: product_type_id_list 需要是列表'
+
+        cache_file = f'{self.config.auto_dir}/shein/attribute/attribute_template_{spu_name}.json'
+        attr_list = read_dict_from_file(cache_file, 3600 * 24 * 7)
+        if len(attr_list) > 0:
+            return attr_list
+
+        url = f"https://sso.geiwohuo.com/spmp-api-prefix/spmp/basic/query_attribute_templates"
+        payload = {
+            "category_id"         : category_id,
+            "for_update"          : True,
+            "product_type_id_list": product_type_id_list,
+            "spu_name"            : spu_name
+        }
+        response_text = fetch(self.web_page, url, payload)
+        error_code = response_text.get('code')
+        if str(error_code) != '0':
+            raise send_exception(json.dumps(response_text, ensure_ascii=False))
+        info = response_text.get('info')
+
+        data = info.get('data')[0]
+        write_dict_to_file(cache_file, data)
+        return data
+
+    def get_product_detail(self, spu_name):
+        cache_file = f'{self.config.auto_dir}/shein/product_detail/product_detail_{spu_name}.json'
+        info = read_dict_from_file(cache_file, 3600 * 24 * 7)
+        if len(info) > 0:
+            return info
+
+        log(f'正在获取 {spu_name} 商品详情')
+        url = f"https://sso.geiwohuo.com/spmp-api-prefix/spmp/product/get_product_detail"
+        payload = {
+            "spu_name": spu_name
+        }
+        response_text = fetch(self.web_page, url, payload)
+        error_code = response_text.get('code')
+        if str(error_code) != '0':
+            raise send_exception(json.dumps(response_text, ensure_ascii=False))
+        info = response_text.get('info')
+        write_dict_to_file(cache_file, info)
+        return info
+
     # 存储商品库
     def store_product_info(self):
         # todo 商品详情 属性 规格 图片 重量 与 尺寸
         skc_list = self.get_bak_base_info()
         cache_file = f'{self.config.auto_dir}/shein/sku_price/sku_price_{self.store_username}.json'
         dict_sku = read_dict_from_file(cache_file)
+        dict_product_detail = []
         for skc_item in skc_list:
             skc_item['store_username'] = self.store_username
             skc_item['store_name'] = self.store_name
             skc_item['store_manager'] = self.config.shein_store_manager.get(str(self.store_username).lower())
             spu = skc_item['spu']
+            if spu not in dict_product_detail:
+                dict_product_detail.append(spu)
+                material = self.get_product_attr(spu, '材质')
+                log(material)  # 这一步是为了获取 spu 详情和属性
 
             # 倒序遍历 skuList，安全删除
             for i in range(len(skc_item['skuList']) - 1, -1, -1):
@@ -657,6 +747,12 @@ class SheinLib:
         cache_file = f'{self.config.auto_dir}/shein/product/skc_list_{self.store_username}.json'
         write_dict_to_file_ex(cache_file, {self.store_username: skc_list}, [self.store_username])
 
+        skc_file = f'{self.config.auto_dir}/shein/product/skc_list_file.json'
+        write_dict_to_file_ex(skc_file, {self.store_username: cache_file}, [self.store_username])
+
+        detail_file = f'{self.config.auto_dir}/shein/product/product_detail_file.json'
+        write_dict_to_file_ex(detail_file, {self.store_username: dict_product_detail}, [self.store_username])
+
     # 获取备货信息列表
     def get_bak_base_info(self):
         log(f'获取备货信息列表 {self.store_name} {self.store_username}')

qrpa/shein_mysql.py

@@ -1,6 +1,7 @@
 import json
 
 from .mysql_module.shein_return_order_model import SheinReturnOrderManager
+from .mysql_module.shein_product_model import SheinProductManager
 from .fun_base import log
 
 class SheinMysql:
@@ -17,4 +18,37 @@ class SheinMysql:
         with open(json_file, 'r', encoding='utf-8') as f:
             dict = json.load(f)
             for store_username, data_list in dict.items():
-                manager.upsert_return_order_data(store_username, data_list)
+                manager.upsert_return_order_data(store_username, data_list)
+
+    def upsert_shein_product(self, json_file):
+        log(f'当前使用的数据库: {self.config.db.database_url}')
+        # 创建管理器实例
+        manager = SheinProductManager(self.config.db.database_url)
+        # 创建数据表
+        manager.create_tables()
+        with open(json_file, 'r', encoding='utf-8') as f:
+            file_list = json.load(f)
+            for store_username, store_skc_list_file in file_list.items():
+                with open(store_skc_list_file, 'r', encoding='utf-8') as f:
+                    dict_store_skc_list = json.load(f)
+                    for store_username, data_list in dict_store_skc_list.items():
+                        manager.upsert_product_data(data_list)
+
+    def upsert_shein_product_info(self, json_file):
+        log(f'当前使用的数据库: {self.config.db.database_url}')
+        # 创建管理器实例
+        manager = SheinProductManager(self.config.db.database_url)
+        # 创建数据表
+        manager.create_tables()
+        with open(json_file, 'r', encoding='utf-8') as f:
+            file_list = json.load(f)
+            for store_username, store_spu_list in file_list.items():
+                for spu in store_spu_list:
+                    product_detail_file = f'{self.config.auto_dir}/shein/product_detail/product_detail_{spu}.json'
+                    attribute_file = f'{self.config.auto_dir}/shein/attribute/attribute_template_{spu}.json'
+                    with open(product_detail_file, 'r', encoding='utf-8') as f:
+                        data_list = json.load(f)
+                        manager.upsert_product_detail(spu, 'product_detail', data_list)
+                    with open(attribute_file, 'r', encoding='utf-8') as f:
+                        data_list = json.load(f)
+                        manager.upsert_product_detail(spu, 'attribute_template', data_list)

qrpa/shein_ziniao.py

@@ -259,6 +259,7 @@ class ZiniaoBrowser:
         page = browser_context.pages[0]
         page.goto(launcher_page)
         page.wait_for_load_state('load')
+        page.wait_for_timeout(6000)
 
         run_func(page, store_username, store_name, task_key)
 
@@ -278,15 +279,14 @@ class ZiniaoTaskManager:
             is_skip_store: Optional[Callable] = None
     ):
         """运行单个店铺的任务"""
+        store_id = browser_info.get('browserOauth')
+        store_name = browser_info.get("browserName")
+        store_username = browser_info.get("store_username")
+
         retry_count = 0
         while True:
             try:
                 retry_count += 1
-
-                store_id = browser_info.get('browserOauth')
-                store_name = browser_info.get("browserName")
-                store_username = browser_info.get("store_username")
-
                 # 记录店铺账号与店铺别名对应关系
                 cache_file = f'{self.config.auto_dir}/shein_store_alias.json'
                 write_dict_to_file_ex(cache_file, {store_username: store_name}, [store_username])
@@ -349,7 +349,7 @@ class ZiniaoTaskManager:
                         self.browser.close_store(store_id)
                     break
             except:
-                send_exception(f'第{retry_count}次运行失败，准备重新打开店铺')
+                send_exception(f'第{retry_count}次运行失败，准备重新打开店铺: {store_username},{store_name},{store_id}')
                 if retry_count > 5:
                     break
 

{qrpa-1.0.65.dist-info → qrpa-1.0.68.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qrpa
-Version: 1.0.65
+Version: 1.0.68
 Summary: qsir's rpa library
 Author: QSir
 Author-email: QSir <1171725650@qq.com>

{qrpa-1.0.65.dist-info → qrpa-1.0.68.dist-info}/RECORD

@@ -10,11 +10,11 @@ qrpa/fun_file.py,sha256=yzjDV16WL5vRys7J4uQcNzIFkX4D5MAlSCwxcD-mwQo,11966
 qrpa/fun_web.py,sha256=F6cxwwOxB8twg2XK73oAvKla0FqpwDYJiM37CWmFwXo,6322
 qrpa/fun_win.py,sha256=-LnTeocdTt72NVH6VgLdpAT9_C5oV9okeudXG6CftMA,8034
 qrpa/shein_daily_report_model.py,sha256=H8oZmIN5Pyqe306W1_xuz87lOqLQ_LI5RjXbaxDkIzE,12589
-qrpa/shein_excel.py,sha256=vtmeEF_LbH6cVyitDDCNjuT76Nuov6Zvhsf-gIMd0H0,124780
-qrpa/shein_lib.py,sha256=jwJshOIiRtCMQWRwi0QwYIOCyLLE-AYnWPTDUSNgTu8,122127
-qrpa/shein_mysql.py,sha256=MGPJsz13evsLygZWhmitpPzUswiTKZ91VFvr8KA0M40,750
+qrpa/shein_excel.py,sha256=1F9QMMsaihDfwr0ajSSQdXksBCjuKSPwMynfP7Jlkso,124825
+qrpa/shein_lib.py,sha256=fTrSRnbAb1X9BUr6_ZrBjWDHbwbRz1wIe5kmGHs2r60,126731
+qrpa/shein_mysql.py,sha256=SrgBYPNWaKHIpD5Q7hqz3GWvytXO8u9SPvKxJvLwYsI,2720
 qrpa/shein_sqlite.py,sha256=ZQwD0Gz81q9WY7tY2HMEYvSF9r3N_G_Aur3bYfST9WY,5707
-qrpa/shein_ziniao.py,sha256=kAOirjawPf-VxBxHFNpdDHCN2zL-O1U-2xybpmt34Xs,18692
+qrpa/shein_ziniao.py,sha256=Xc4pgSa7PxgmR24rClgz_HcTRb2fdbXp6slzfpCOThI,18745
 qrpa/temu_chrome.py,sha256=CbtFy1QPan9xJdJcNZj-EsVGhUvv3ZTEPVDEA4-im40,2803
 qrpa/temu_excel.py,sha256=2hGw76YWzkTZGyFCuuUAab4oHptYX9a6U6yjpNsL7FE,6990
 qrpa/temu_lib.py,sha256=hYB59zsLS3m3NTic_duTwPMOTSxlHyQVa8OhHnHm-1g,7199
@@ -22,8 +22,9 @@ qrpa/time_utils.py,sha256=ef0hhbN_6b-gcnz5ETIVOoxemIMvcxGVGGIRnHnGaBo,29564
 qrpa/time_utils_example.py,sha256=shHOXKKF3QSzb0SHsNc34M61wEkkLuM30U9X1THKNS8,8053
 qrpa/wxwork.py,sha256=gIytG19DZ5g7Tsl0-W3EbjfSnpIqZw-ua24gcB78YEg,11264
 qrpa/mysql_module/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+qrpa/mysql_module/shein_product_model.py,sha256=qViI_Ik3SkXXxqJ1nXjimvfB_a5uiwW9RXL0fOreBao,18880
 qrpa/mysql_module/shein_return_order_model.py,sha256=Zt-bGOH_kCDbakW7uaTmqqo_qTT8v424yidcYSfWvWM,26562
-qrpa-1.0.65.dist-info/METADATA,sha256=YxYB_bf0U7NZAJ5t0Xf6P_L4gQuQZy2U5UhzLecpbEA,231
-qrpa-1.0.65.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-qrpa-1.0.65.dist-info/top_level.txt,sha256=F6T5igi0fhXDucPPUbmmSC0qFCDEsH5eVijfVF48OFU,5
-qrpa-1.0.65.dist-info/RECORD,,
+qrpa-1.0.68.dist-info/METADATA,sha256=g0-y3VWC4m7Jfhzv6ycjS7AQ3G5c6oHcE5l6X5j4x50,231
+qrpa-1.0.68.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+qrpa-1.0.68.dist-info/top_level.txt,sha256=F6T5igi0fhXDucPPUbmmSC0qFCDEsH5eVijfVF48OFU,5
+qrpa-1.0.68.dist-info/RECORD,,