qrpa 1.1.34__tar.gz → 1.1.36__tar.gz

{qrpa-1.1.34 → qrpa-1.1.36}/PKG-INFO

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

{qrpa-1.1.34 → qrpa-1.1.36}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "qrpa"
-version = "1.1.34"
+version = "1.1.36"
 description = "qsir's rpa library"
 authors = [{ name = "QSir", email = "1171725650@qq.com" }]
 readme = "README.md"

{qrpa-1.1.34 → qrpa-1.1.36}/qrpa/mysql_module/new_product_analysis_model.py

@@ -63,6 +63,7 @@ class SheinNewProductAnalysis(Base):
     ab_test = Column(Text, nullable=True, comment='AB测试数据(JSON)')
 
     # 分析字段（不参与更新）
+    front_price = Column(DECIMAL(10, 2), nullable=True, comment='前台价格')
     reason_analysis = Column(Text, nullable=True, comment='原因分析')
     optimization_action = Column(Text, nullable=True, comment='优化动作')
 
@@ -144,7 +145,7 @@ class NewProductAnalysisManager:
 
                 if existing:
                     # 更新现有记录（排除不更新的分析字段）
-                    exclude_fields = {'reason_analysis', 'optimization_action'}
+                    exclude_fields = {'front_price', 'reason_analysis', 'optimization_action'}
                     for key, value in data.items():
                         if key not in exclude_fields:
                             setattr(existing, key, value)
@@ -235,7 +236,13 @@ class NewProductAnalysisManager:
             prom_detail = prom.get('promDetail', [])
 
             # 根据promId长度判断取哪些字段
-            if len(str(prom_id)) > 6 and prom_detail:
+            if len(str(prom_id)) >= 11 and prom_detail:
+                # 营销工具：取第一个detail的数据
+                detail = prom_detail if prom_detail else {}
+                prom_info['attend_num_sum'] = detail.get('act_stock_num')
+                prom_info['supply_price'] = detail.get('sku_price')
+                prom_info['product_act_price'] = detail.get('act_sku_price')
+            elif len(str(prom_id)) >= 8 and prom_detail:
                 # 营销工具：取第一个detail的数据
                 detail = prom_detail[0] if prom_detail else {}
                 prom_info['attend_num_sum'] = detail.get('attend_num_sum')
@@ -328,6 +335,41 @@ class NewProductAnalysisManager:
         # 调用insert_data方法插入数据
         return self.insert_data(data_list)
 
+    def update_front_price(self, stat_date, skc, front_price):
+        """
+        更新指定SKC的前台价格
+        
+        Args:
+            stat_date (str): 统计日期
+            skc (str): 平台SKC
+            front_price (float): 前台价格
+            
+        Returns:
+            bool: 是否成功
+        """
+        session = self.Session()
+        try:
+            record = session.query(SheinNewProductAnalysis).filter(
+                SheinNewProductAnalysis.stat_date == stat_date,
+                SheinNewProductAnalysis.skc == skc
+            ).first()
+            
+            if record:
+                record.front_price = front_price
+                setattr(record, 'updated_at', datetime.now())
+                session.commit()
+                print(f"成功更新前台价格: {skc} ({stat_date}) -> {front_price}")
+                return True
+            else:
+                print(f"未找到记录: SKC={skc}, 日期={stat_date}")
+                return False
+        except Exception as e:
+            session.rollback()
+            print(f"更新前台价格失败: {e}")
+            raise
+        finally:
+            session.close()
+
     def delete_records_by_date(self, store_username, stat_date):
         """
         删除指定日期的记录
@@ -410,6 +452,23 @@ def get_records_by_skc(store_username, skc):
     manager = get_new_product_analysis_manager()
     return manager.get_records_by_skc(store_username, skc)
 
+
+def update_front_price(stat_date, skc, front_price):
+    """
+    更新前台价格的便捷函数
+
+    Args:
+        stat_date (str): 统计日期
+        skc (str): 平台SKC
+        front_price (float): 前台价格
+
+    Returns:
+        bool: 是否成功
+    """
+    manager = get_new_product_analysis_manager()
+    return manager.update_front_price(stat_date, skc, front_price)
+
+
 if __name__ == '__main__':
     # 测试代码
     manager = NewProductAnalysisManager()
@@ -427,3 +486,9 @@ if __name__ == '__main__':
         json_data = json.load(f)
     count = manager.import_from_json(json_data)
     print(f"成功导入 {count} 条记录")
+    
+    # 更新前台价格（手动设置，后续导入不会覆盖）
+    # manager.update_front_price('2025-10-15', 'si2409238815318914', 19.99)
+    
+    # 或使用便捷函数
+    # update_front_price('2025-10-15', 'si2409238815318914', 19.99)

qrpa-1.1.36/qrpa/mysql_module/shein_store_model.py

@@ -0,0 +1,594 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+店铺信息数据模型
+使用SQLAlchemy定义shein_store表结构
+"""
+
+from sqlalchemy import create_engine, Column, Integer, String, BigInteger, DateTime, Index
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+from datetime import datetime
+import json
+
+# 创建基类
+Base = declarative_base()
+
+class SheinStore(Base):
+    """
+    希音店铺信息表
+    存储店铺基础信息
+    """
+    __tablename__ = 'shein_store'
+
+    # 主键ID
+    id = Column(BigInteger, primary_key=True, autoincrement=True, comment='主键ID')
+
+    # 店铺账号信息（唯一）
+    user_name = Column(String(100), nullable=False, unique=True, comment='店铺账号')
+    store_username = Column(String(100), nullable=True, comment='店铺用户名')
+    store_name = Column(String(200), nullable=True, comment='店铺名称')
+    
+    # 用户ID信息
+    user_id = Column(BigInteger, nullable=True, comment='用户ID')
+    supplier_id = Column(BigInteger, nullable=True, comment='供应商ID')
+    main_user_name = Column(String(100), nullable=True, comment='主账号用户名')
+    main_user_id = Column(BigInteger, nullable=True, comment='主账号用户ID')
+    
+    # ULP信息
+    ulp_name = Column(String(200), nullable=True, comment='ULP名称')
+    ulp_en_name = Column(String(200), nullable=True, comment='ULP英文名称')
+    ulp_emplid = Column(String(100), nullable=True, comment='ULP员工ID')
+    is_ulp_login = Column(Integer, nullable=True, comment='是否ULP登录')
+    
+    # 时区信息
+    timezone = Column(String(100), nullable=True, comment='时区')
+    timezone_name = Column(String(200), nullable=True, comment='时区名称')
+    utc_timezone = Column(String(50), nullable=True, comment='UTC时区')
+    area_timezone = Column(String(100), nullable=True, comment='地区时区')
+    
+    # 系统配置
+    switch_new_menu = Column(Integer, nullable=True, comment='是否切换新菜单')
+    supplier_user_name = Column(String(100), nullable=True, comment='供应商用户名')
+    sso_top_nav = Column(Integer, nullable=True, comment='SSO顶部导航')
+    sso_host = Column(String(500), nullable=True, comment='SSO主机地址')
+    
+    # 类目信息
+    category_id = Column(BigInteger, nullable=True, comment='类目ID')
+    category_out_id = Column(BigInteger, nullable=True, comment='外部类目ID')
+    lv1_category_id = Column(BigInteger, nullable=True, comment='一级类目ID')
+    lv1_category_name = Column(String(200), nullable=True, comment='一级类目名称')
+    lv2_category_name = Column(String(200), nullable=True, comment='二级类目名称')
+    
+    # 其他信息
+    supplier_source = Column(Integer, nullable=True, comment='供应商来源')
+    external_id = Column(BigInteger, nullable=True, comment='外部ID')
+    store_code = Column(BigInteger, nullable=True, comment='店铺编码')
+    merchant_code = Column(String(100), nullable=True, comment='商户编码')
+    schat_id = Column(BigInteger, nullable=True, comment='Schat ID')
+    
+    # 管理字段（不参与更新）
+    store_manager_id = Column(BigInteger, default=0, comment='店长ID(用户表ID)')
+    is_deleted = Column(Integer, nullable=True, default=0, comment='软删除标志(0-未删除,1-已删除)')
+    
+    # 备注
+    remark = Column(String(500), nullable=True, comment='备注')
+    
+    # 时间戳
+    created_at = Column(DateTime, nullable=True, default=datetime.now, comment='创建时间')
+    updated_at = Column(DateTime, nullable=True, default=datetime.now, onupdate=datetime.now, comment='更新时间')
+
+    # 创建索引
+    __table_args__ = (
+        Index('idx_store_username', 'store_username',unique=True),
+    )
+
+    def __repr__(self):
+        return f"<SheinStore(id={self.id}, user_name='{self.user_name}', store_name='{self.store_name}')>"
+
+
+class SheinStoreManager:
+    """
+    店铺信息数据管理器
+    提供数据库操作相关方法
+    """
+
+    @classmethod
+    def __init__(self, database_url):
+        """
+        初始化数据库连接
+
+        Args:
+            database_url (str): 数据库连接URL
+        """
+        print(f"连接数据库: {database_url}")
+        self.engine = create_engine(database_url, echo=False)
+        self.Session = sessionmaker(bind=self.engine)
+
+    def create_table(self):
+        """
+        创建数据表
+        """
+        Base.metadata.create_all(self.engine)
+        print("店铺表创建成功")
+
+    def insert_data(self, data_list):
+        """
+        批量插入数据
+
+        Args:
+            data_list (list): 数据字典列表
+
+        Returns:
+            int: 成功插入的记录数
+        """
+        session = self.Session()
+        try:
+            insert_count = 0
+            update_count = 0
+            
+            for data in data_list:
+                # 检查是否存在（根据唯一索引：user_name）
+                existing = session.query(SheinStore).filter(
+                    SheinStore.user_name == data.get('user_name')
+                ).first()
+
+                if existing:
+                    # 更新现有记录（排除不更新的字段）
+                    exclude_fields = {'store_manager_id', 'is_deleted'}
+                    for key, value in data.items():
+                        if key not in exclude_fields:
+                            setattr(existing, key, value)
+                    setattr(existing, 'updated_at', datetime.now())
+                    update_count += 1
+                else:
+                    # 插入新记录
+                    new_record = SheinStore(**data)
+                    session.add(new_record)
+                    insert_count += 1
+
+            session.commit()
+            print(f"成功插入 {insert_count} 条记录，更新 {update_count} 条记录")
+            return insert_count
+        except Exception as e:
+            session.rollback()
+            print(f"插入数据失败: {e}")
+            raise
+        finally:
+            session.close()
+
+    def import_from_json_dict(self, json_dict):
+        """
+        从JSON字典数据导入到数据库（JSON格式：{userName: {店铺信息}}）
+        
+        Args:
+            json_dict (dict): JSON字典数据，key为userName，value为店铺信息
+            
+        Returns:
+            int: 成功导入的记录数
+        """
+        print(f"开始处理 {len(json_dict)} 个店铺数据")
+        data_list = []
+        skipped_count = 0
+
+        for user_name, item in json_dict.items():
+            # 验证必填字段
+            if not user_name:
+                print(f"⚠️ 警告: 缺少必填字段 user_name，跳过该记录")
+                skipped_count += 1
+                continue
+
+            # 构建数据库记录（字段名转换：驼峰转下划线）
+            record = {
+                'user_name': item.get('userName'),
+                'store_username': item.get('store_username'),
+                'store_name': item.get('store_name'),
+                'user_id': item.get('userId'),
+                'supplier_id': item.get('supplierId'),
+                'main_user_name': item.get('mainUserName'),
+                'main_user_id': item.get('mainUserId'),
+                'ulp_name': item.get('ulpName'),
+                'ulp_en_name': item.get('ulpEnName'),
+                'ulp_emplid': item.get('ulpEmplid'),
+                'is_ulp_login': item.get('isUlpLogin'),
+                'timezone': item.get('timezone'),
+                'timezone_name': item.get('timezoneName'),
+                'utc_timezone': item.get('utcTimezone'),
+                'area_timezone': item.get('areaTimezone'),
+                'switch_new_menu': item.get('switchNewMenu'),
+                'supplier_user_name': item.get('supplierUserName'),
+                'sso_top_nav': item.get('ssoTopNav'),
+                'sso_host': item.get('ssoHost'),
+                'category_id': item.get('categoryId'),
+                'category_out_id': item.get('categoryOutId'),
+                'lv1_category_id': item.get('lv1CategoryId'),
+                'lv1_category_name': item.get('lv1CategoryName'),
+                'lv2_category_name': item.get('lv2CategoryName'),
+                'supplier_source': item.get('supplierSource'),
+                'external_id': item.get('externalId'),
+                'store_code': item.get('storeCode'),
+                'merchant_code': item.get('merchantCode'),
+                'schat_id': item.get('schatId'),
+            }
+
+            print(f"✓ 处理店铺: {record['user_name']} - {record['store_name']}")
+            data_list.append(record)
+
+        if skipped_count > 0:
+            print(f"⚠️ 共跳过 {skipped_count} 条记录（缺少必填字段）")
+
+        # 调用insert_data方法插入数据
+        return self.insert_data(data_list)
+
+    def import_from_json_list(self, json_list):
+        """
+        从JSON列表数据导入到数据库（JSON格式：[{店铺信息}, {店铺信息}]）
+        
+        Args:
+            json_list (list): JSON列表数据
+            
+        Returns:
+            int: 成功导入的记录数
+        """
+        # 转换为字典格式，然后调用 import_from_json_dict
+        json_dict = {item.get('userName'): item for item in json_list if item.get('userName')}
+        return self.import_from_json_dict(json_dict)
+
+    def get_store_by_username(self, user_name, include_deleted=False):
+        """
+        根据用户名查询店铺信息
+        
+        Args:
+            user_name (str): 店铺账号
+            include_deleted (bool): 是否包含已删除的店铺，默认False
+            
+        Returns:
+            SheinStore: 店铺对象
+        """
+        session = self.Session()
+        try:
+            query = session.query(SheinStore).filter(
+                SheinStore.user_name == user_name
+            )
+            
+            # 默认只查询未删除的
+            if not include_deleted:
+                query = query.filter(SheinStore.is_deleted == 0)
+            
+            store = query.first()
+            return store
+        finally:
+            session.close()
+
+    def get_all_stores(self, include_deleted=False):
+        """
+        获取所有店铺列表
+        
+        Args:
+            include_deleted (bool): 是否包含已删除的店铺，默认False
+        
+        Returns:
+            list: 店铺列表
+        """
+        session = self.Session()
+        try:
+            query = session.query(SheinStore)
+            
+            # 默认只查询未删除的
+            if not include_deleted:
+                query = query.filter(SheinStore.is_deleted == 0)
+            
+            stores = query.all()
+            return stores
+        finally:
+            session.close()
+
+    def get_stores_by_category(self, lv1_category_name=None, lv2_category_name=None, include_deleted=False):
+        """
+        根据类目查询店铺列表
+        
+        Args:
+            lv1_category_name (str): 一级类目名称
+            lv2_category_name (str): 二级类目名称
+            include_deleted (bool): 是否包含已删除的店铺，默认False
+            
+        Returns:
+            list: 店铺列表
+        """
+        session = self.Session()
+        try:
+            query = session.query(SheinStore)
+            
+            # 默认只查询未删除的
+            if not include_deleted:
+                query = query.filter(SheinStore.is_deleted == 0)
+            
+            if lv1_category_name:
+                query = query.filter(SheinStore.lv1_category_name == lv1_category_name)
+            
+            if lv2_category_name:
+                query = query.filter(SheinStore.lv2_category_name == lv2_category_name)
+            
+            stores = query.all()
+            return stores
+        finally:
+            session.close()
+
+    def update_store_manager(self, user_name, manager_id):
+        """
+        更新店铺的店长ID
+        
+        Args:
+            user_name (str): 店铺账号
+            manager_id (int): 店长ID（用户表ID）
+            
+        Returns:
+            bool: 是否成功
+        """
+        session = self.Session()
+        try:
+            store = session.query(SheinStore).filter(
+                SheinStore.user_name == user_name
+            ).first()
+            
+            if store:
+                store.store_manager_id = manager_id
+                store.updated_at = datetime.now()
+                session.commit()
+                print(f"成功更新店长ID: {user_name} -> {manager_id}")
+                return True
+            else:
+                print(f"未找到店铺: {user_name}")
+                return False
+        except Exception as e:
+            session.rollback()
+            print(f"更新店长ID失败: {e}")
+            raise
+        finally:
+            session.close()
+    
+    def soft_delete_store(self, user_name):
+        """
+        软删除指定店铺（设置is_deleted=1）
+        
+        Args:
+            user_name (str): 店铺账号
+            
+        Returns:
+            bool: 是否成功
+        """
+        session = self.Session()
+        try:
+            store = session.query(SheinStore).filter(
+                SheinStore.user_name == user_name
+            ).first()
+            
+            if store:
+                store.is_deleted = 1
+                store.updated_at = datetime.now()
+                session.commit()
+                print(f"成功软删除店铺: {user_name}")
+                return True
+            else:
+                print(f"未找到店铺: {user_name}")
+                return False
+        except Exception as e:
+            session.rollback()
+            print(f"软删除失败: {e}")
+            raise
+        finally:
+            session.close()
+    
+    def restore_store(self, user_name):
+        """
+        恢复软删除的店铺（设置is_deleted=0）
+        
+        Args:
+            user_name (str): 店铺账号
+            
+        Returns:
+            bool: 是否成功
+        """
+        session = self.Session()
+        try:
+            store = session.query(SheinStore).filter(
+                SheinStore.user_name == user_name
+            ).first()
+            
+            if store:
+                store.is_deleted = 0
+                store.updated_at = datetime.now()
+                session.commit()
+                print(f"成功恢复店铺: {user_name}")
+                return True
+            else:
+                print(f"未找到店铺: {user_name}")
+                return False
+        except Exception as e:
+            session.rollback()
+            print(f"恢复失败: {e}")
+            raise
+        finally:
+            session.close()
+
+    def delete_store(self, user_name):
+        """
+        物理删除指定店铺（真正从数据库删除，慎用！）
+        
+        Args:
+            user_name (str): 店铺账号
+            
+        Returns:
+            int: 删除的记录数
+        """
+        session = self.Session()
+        try:
+            delete_count = session.query(SheinStore).filter(
+                SheinStore.user_name == user_name
+            ).delete()
+            session.commit()
+            print(f"成功物理删除 {delete_count} 条记录")
+            return delete_count
+        except Exception as e:
+            session.rollback()
+            print(f"删除数据失败: {e}")
+            raise
+        finally:
+            session.close()
+
+
+# 创建全局实例，用于快速访问
+_shein_store_manager = None
+
+
+def get_shein_store_manager(database_url):
+    """
+    获取店铺管理器实例
+
+    Args:
+        database_url (str): 数据库连接URL
+
+    Returns:
+        SheinStoreManager: 店铺管理器实例
+    """
+    global _shein_store_manager
+    if _shein_store_manager is None:
+        _shein_store_manager = SheinStoreManager(database_url)
+    return _shein_store_manager
+
+
+def insert_store_data(database_url, data_list):
+    """
+    插入店铺数据的便捷函数
+
+    Args:
+        database_url (str): 数据库连接URL
+        data_list (list): 数据字典列表
+
+    Returns:
+        int: 成功插入的记录数
+    """
+    manager = get_shein_store_manager(database_url)
+    return manager.insert_data(data_list)
+
+
+def import_from_json(database_url, json_data):
+    """
+    从JSON数据导入的便捷函数（自动识别字典或列表格式）
+
+    Args:
+        database_url (str): 数据库连接URL
+        json_data (dict|list): JSON数据（字典或列表）
+
+    Returns:
+        int: 成功导入的记录数
+    """
+    manager = get_shein_store_manager(database_url)
+    
+    if isinstance(json_data, dict):
+        return manager.import_from_json_dict(json_data)
+    elif isinstance(json_data, list):
+        return manager.import_from_json_list(json_data)
+    else:
+        raise ValueError("json_data 必须是字典或列表类型")
+
+
+def get_store_by_username(database_url, user_name, include_deleted=False):
+    """
+    根据用户名查询店铺的便捷函数
+
+    Args:
+        database_url (str): 数据库连接URL
+        user_name (str): 店铺账号
+        include_deleted (bool): 是否包含已删除的店铺，默认False
+
+    Returns:
+        SheinStore: 店铺对象
+    """
+    manager = get_shein_store_manager(database_url)
+    return manager.get_store_by_username(user_name, include_deleted)
+
+
+def update_store_manager(database_url, user_name, manager_id):
+    """
+    更新店长ID的便捷函数
+
+    Args:
+        database_url (str): 数据库连接URL
+        user_name (str): 店铺账号
+        manager_id (int): 店长ID（用户表ID）
+
+    Returns:
+        bool: 是否成功
+    """
+    manager = get_shein_store_manager(database_url)
+    return manager.update_store_manager(user_name, manager_id)
+
+
+def soft_delete_store(database_url, user_name):
+    """
+    软删除店铺的便捷函数
+
+    Args:
+        database_url (str): 数据库连接URL
+        user_name (str): 店铺账号
+
+    Returns:
+        bool: 是否成功
+    """
+    manager = get_shein_store_manager(database_url)
+    return manager.soft_delete_store(user_name)
+
+
+def restore_store(database_url, user_name):
+    """
+    恢复店铺的便捷函数
+
+    Args:
+        database_url (str): 数据库连接URL
+        user_name (str): 店铺账号
+
+    Returns:
+        bool: 是否成功
+    """
+    manager = get_shein_store_manager(database_url)
+    return manager.restore_store(user_name)
+
+
+if __name__ == '__main__':
+    # 测试代码
+    # 注意：需要提供数据库连接URL
+    database_url = "mysql+pymysql://root:123wyk@127.0.0.1:3306/lz"
+    
+    manager = SheinStoreManager(database_url)
+    
+    # 创建表
+    manager.create_table()
+
+    # 读取JSON文件
+    with open('../../docs/shein_user.json', 'r', encoding='utf-8') as f:
+        json_data = json.load(f)
+    
+    count = manager.import_from_json_dict(json_data)
+    print(f"成功导入 {count} 条店铺记录")
+    
+    # 更新店长ID（手动设置，后续导入不会覆盖）
+    # manager.update_store_manager('GS0628340', 123)  # 123是用户表的ID
+    
+    # 软删除测试
+    # manager.soft_delete_store('GS0628340')
+    
+    # 查询店铺（默认不包含已删除）
+    # store = manager.get_store_by_username('GS0628340')
+    # print(f"查询结果: {store}")
+    
+    # 查询店铺（包含已删除）
+    # store = manager.get_store_by_username('GS0628340', include_deleted=True)
+    # print(f"查询结果（包含已删除）: {store}")
+    
+    # 恢复店铺
+    # manager.restore_store('GS0628340')
+    
+    # 按类目查询示例
+    # stores = manager.get_stores_by_category(lv2_category_name='全托管-备CN-中国团队-内睡服装')
+    # print(f"找到 {len(stores)} 个店铺")

{qrpa-1.1.34 → qrpa-1.1.36}/qrpa/shein_lib.py

@@ -1236,7 +1236,10 @@ class SheinLib:
             for prom_inf_ing in skc_item['promCampaign'].get('promInfIng') or []:
                 prom_id = prom_inf_ing['promId']
                 log('prom_id:', prom_id, len(prom_id))
-                if len(prom_id) >= 6:
+                if len(prom_id) >= 11:
+                    # 托管活动
+                    prom_inf_ing['promDetail'] = self.get_skc_activity_price_info(skc, prom_id)
+                elif len(prom_id) >= 8:
                     # 营销工具
                     prom_inf_ing['promDetail'] = self.query_goods_detail(prom_id)
                 else:
@@ -1246,7 +1249,9 @@ class SheinLib:
             for prom_inf_ready in skc_item['promCampaign'].get('promInfReady') or []:
                 prom_id = prom_inf_ready['promId']
                 log('prom_id:', prom_id, len(prom_id))
-                if len(prom_id) >= 6:
+                if len(prom_id) >= 11:
+                    prom_inf_ready['promDetail'] = self.get_skc_activity_price_info(skc, prom_id)
+                elif len(prom_id) >= 8:
                     prom_inf_ready['promDetail'] = self.query_goods_detail(prom_id)
                 else:
                     prom_inf_ready['promDetail'] = self.get_partake_activity_detail(prom_id, skc)
@@ -2373,6 +2378,7 @@ class SheinLib:
             list_item += response_text['info']['data']
             time.sleep(0.1)
 
+        log(list_item)
         write_dict_to_file(cache_file, list_item)
         return list_item
 
@@ -2398,6 +2404,7 @@ class SheinLib:
             raise send_exception(json.dumps(response_text, ensure_ascii=False))
         list_item = response_text['info']['data']
 
+        log(list_item)
         write_dict_to_file(cache_file, list_item)
         return list_item
 
@@ -3930,3 +3937,247 @@ class SheinLib:
             log('临时下载文件已清理')
 
         return output_file_path
+
+    def query_hosting_info_list(self):
+        """
+        查询店铺活动托管规则列表
+        
+        Returns:
+            list: 托管规则列表，每个元素包含：
+                - hosting_id: 托管规则ID
+                - scene_type: 场景类型
+                - state: 状态
+                - hosting_name: 托管规则名称
+                - hosting_tools_id: 托管工具ID
+                - hosting_tools_state: 托管工具状态
+                - time_zone: 时区
+                - create_user: 创建用户
+                - last_update_user: 最后更新用户
+                - insert_time: 创建时间
+                - last_update_time: 最后更新时间
+                - exist_act_goods: 是否存在活动商品
+        """
+        log(f'正在获取 {self.store_name} 店铺活动托管规则列表')
+
+        cache_file = f'{self.config.auto_dir}/shein/cache/hosting_info_list_{self.store_username}_{TimeUtils.today_date()}.json'
+        hosting_list = read_dict_from_file_ex(cache_file, self.store_username, 3600 * 8)
+        if len(hosting_list) > 0:
+            log('返回缓存数据')
+            return hosting_list
+
+        url = "https://sso.geiwohuo.com/mrs-api-prefix/promotion/hosting/query_hosting_info_list"
+        payload = {}
+
+        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))
+
+        hosting_list = response_text.get('info', [])
+        log(f'获取到 {len(hosting_list)} 条托管规则')
+
+        write_dict_to_file_ex(cache_file, {self.store_username: hosting_list}, [self.store_username])
+
+        return hosting_list
+
+    def query_hosting_activity_goods(self, hosting_id, goods_states=None):
+        """
+        查询托管活动参与的商品
+        
+        Args:
+            hosting_id: 托管规则ID
+            goods_states: 商品状态列表，默认为[1]（在售）
+            
+        Returns:
+            list: 参与托管活动的商品列表，每个元素包含：
+                - goods_state: 商品状态
+                - skc_info_list: SKC信息列表
+                    - skc_id: SKC ID
+                    - skc_name: SKC名称
+                    - goods_name: 商品名称
+                    - image_url: 图片URL
+                    - act_stock_num: 活动库存数量
+                    - act_sales_num: 活动销量
+                    - activity_info: 活动信息
+                    - sku_info_list: SKU信息列表
+        """
+        if goods_states is None:
+            goods_states = [1]
+
+        log(f'正在获取 {self.store_name} 托管活动商品列表 hosting_id={hosting_id}')
+
+        cache_file = f'{self.config.auto_dir}/shein/cache/hosting_activity_goods_{self.store_username}_{hosting_id}_{TimeUtils.today_date()}.json'
+        goods_list = read_dict_from_file_ex(cache_file, self.store_username, 3600 * 8)
+        if len(goods_list) > 0:
+            log('返回缓存数据')
+            return goods_list
+
+        page_num = 1
+        page_size = 100
+
+        url = "https://sso.geiwohuo.com/mrs-api-prefix/promotion/hosting/query_hosting_activity_goods"
+        payload = {
+            "goods_states": goods_states,
+            "hosting_id"  : str(hosting_id),
+            "page_num"    : page_num,
+            "page_size"   : page_size
+        }
+
+        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))
+
+        goods_list = response_text.get('info', [])
+        if not goods_list:
+            log('未获取到商品数据')
+            return []
+
+        # 获取第一页数据
+        first_item = goods_list[0] if goods_list else {}
+        skc_info_list = first_item.get('skc_info_list', {})
+        all_data = skc_info_list.get('data', [])
+        meta = skc_info_list.get('meta', {})
+        total = meta.get('count', 0)
+
+        log(f'第1页获取到 {len(all_data)} 条商品，总数: {total}')
+
+        # 如果有多页，继续获取
+        if total > page_size:
+            totalPage = math.ceil(total / page_size)
+            for page in range(2, totalPage + 1):
+                log(f'获取托管活动商品列表 第{page}/{totalPage}页')
+                payload['page_num'] = page
+                response_text = fetch(self.web_page, url, payload)
+                error_code = response_text.get('code')
+                if str(error_code) != '0':
+                    log(f'获取第{page}页失败: {response_text}')
+                    continue
+
+                page_goods_list = response_text.get('info', [])
+                if page_goods_list:
+                    page_data = page_goods_list[0].get('skc_info_list', {}).get('data', [])
+                    all_data.extend(page_data)
+                    log(f'第{page}页获取到 {len(page_data)} 条商品')
+
+                time.sleep(0.1)
+
+        log(f'总共获取到 {len(all_data)} 条商品')
+
+        # 保存缓存
+        write_dict_to_file_ex(cache_file, {self.store_username: all_data}, [self.store_username])
+
+        return all_data
+
+    def get_skc_activity_price_info(self, skc, activity_id):
+        """
+        根据SKC和活动ID获取供货价、活动价和活动库存
+        
+        Args:
+            skc: SKC名称
+            activity_id: 活动ID（可以是字符串或整数）
+            
+        Returns:
+            dict: 包含以下键值的字典，如果未找到则返回None：
+                - sku_price: SKU供货价（取第一个SKU的价格）
+                - act_sku_price: SKU活动价（取第一个SKU的活动价）
+                - act_stock_num: 活动库存数量
+                - skc_name: SKC名称
+                - goods_name: 商品名称
+                - activity_id: 活动ID
+                - currency: 币种
+                - image_url: 商品图片
+        """
+        log(f'获取SKC活动价格信息: skc={skc}, activity_id={activity_id}')
+
+        # 转换activity_id为整数进行比较
+        try:
+            target_activity_id = int(activity_id)
+        except (ValueError, TypeError):
+            log(f'无效的activity_id: {activity_id}')
+            return None
+
+        # 缓存文件，使用skc和activity_id作为缓存key
+        cache_file = f'{self.config.auto_dir}/shein/cache/skc_activity_price_{self.store_username}_{skc}_{activity_id}_{TimeUtils.today_date()}.json'
+        cached_data = read_dict_from_file(cache_file, 3600 * 8)
+        if cached_data:
+            log('返回缓存的价格信息')
+            return cached_data
+
+        # 获取所有托管规则
+        hosting_list = self.query_hosting_info_list()
+
+        if not hosting_list:
+            log('未找到任何托管规则')
+            return None
+
+        # 遍历所有托管规则，查找匹配的SKC和活动
+        for hosting in hosting_list:
+            hosting_id = hosting.get('hosting_id')
+            if not hosting_id:
+                continue
+
+            log(f'查询托管规则: hosting_id={hosting_id}, hosting_name={hosting.get("hosting_name")}')
+
+            # 获取该托管规则下的商品
+            goods_list = self.query_hosting_activity_goods(hosting_id)
+
+            # 在商品列表中查找匹配的SKC
+            for goods_item in goods_list:
+                skc_name = goods_item.get('skc_name', '')
+
+                # 匹配SKC名称
+                if skc_name != skc:
+                    continue
+
+                # 检查活动信息
+                activity_info = goods_item.get('activity_info', {})
+                goods_activity_id = activity_info.get('activity_id')
+
+                # 匹配活动ID
+                try:
+                    if int(goods_activity_id) != target_activity_id:
+                        continue
+                except (ValueError, TypeError):
+                    continue
+
+                log(f'找到匹配的SKC: {skc_name}, activity_id={goods_activity_id}')
+
+                # 提取活动库存
+                act_stock_num = goods_item.get('act_stock_num', 0)
+
+                # 获取第一个SKU的价格信息
+                sku_info_list = goods_item.get('sku_info_list', [])
+                if not sku_info_list:
+                    log(f'SKC {skc_name} 没有SKU信息')
+                    continue
+
+                first_sku = sku_info_list[0]
+                sku_price = first_sku.get('sku_price', 0)
+                act_sku_price = first_sku.get('act_sku_price', 0)
+                currency = first_sku.get('currency', 'CNY')
+
+                # 构建返回结果
+                result = {
+                    'skc_name'     : skc_name,
+                    'goods_name'   : goods_item.get('goods_name', ''),
+                    'image_url'    : goods_item.get('image_url', ''),
+                    'activity_id'  : goods_activity_id,
+                    'act_stock_num': act_stock_num,
+                    'sku_price'    : sku_price,
+                    'act_sku_price': act_sku_price,
+                    'currency'     : currency,
+                    'start_time'   : activity_info.get('start_time', ''),
+                    'end_time'     : activity_info.get('end_time', ''),
+                    'time_zone'    : activity_info.get('time_zone', ''),
+                }
+
+                log(f'SKC供货价: {sku_price} {currency}, 活动价: {act_sku_price} {currency}, 活动库存: {act_stock_num}')
+
+                # 保存缓存
+                write_dict_to_file(cache_file, result)
+
+                return result
+
+        log(f'未找到匹配的SKC和活动: skc={skc}, activity_id={activity_id}')
+        return None

{qrpa-1.1.34 → qrpa-1.1.36}/qrpa.egg-info/PKG-INFO

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

{qrpa-1.1.34 → qrpa-1.1.36}/qrpa.egg-info/SOURCES.txt

@@ -33,5 +33,6 @@ qrpa/mysql_module/new_product_analysis_model.py
 qrpa/mysql_module/shein_ledger_model.py
 qrpa/mysql_module/shein_product_model.py
 qrpa/mysql_module/shein_return_order_model.py
+qrpa/mysql_module/shein_store_model.py
 tests/test_db_migrator.py
 tests/test_wxwork.py