qrpa 1.0.100__py3-none-any.whl → 1.1.2__py3-none-any.whl

qrpa/fun_excel.py

@@ -151,6 +151,34 @@ def wrap_column(sheet, columns=None, WrapText=True):
                 log(f'设置[{c}] 换行 {WrapText}')
                 sheet.range(f'{col_name}:{col_name}').api.WrapText = WrapText
 
+def sort_by_column_excel(sheet, sort_col: str, has_header=True, order="desc"):
+    """
+    对整个表格按照某一列排序
+
+    :param sheet: xlwings 的 sheet 对象
+    :param sort_col: 排序依据的列（如 'D'）
+    :param has_header: 是否有表头（默认 True）
+    :param order: 'asc' 升序，'desc' 降序
+    """
+    # 找到表格的最后一行和列
+    last_cell = sheet.used_range.last_cell
+    rng = sheet.range((1, 1), (last_cell.row, last_cell.column))
+
+    # 排序依据列
+    col_index = ord(sort_col.upper()) - ord('A') + 1
+    key = sheet.range((2 if has_header else 1, col_index)).api
+
+    # 排序顺序
+    order_val = 1 if order == "asc" else 2
+
+    # 调用 Excel 的 Sort 方法
+    rng.api.Sort(
+        Key1=key,
+        Order1=order_val,
+        Orientation=1,
+        Header=1 if has_header else 0
+    )
+
 def sort_by_column(data, col_index, header_rows=2, reverse=True):
     if not data or header_rows >= len(data):
         return data

qrpa/shein_excel.py

@@ -15,6 +15,78 @@ class SheinExcel:
         self.bridge = bridge
         pass
 
+    def write_product(self):
+        erp = self.config.erp_source
+        excel_path = create_file_path(self.config.excel_shein_skc_profit)
+        cache_file = f'{self.config.auto_dir}/shein/product/product_{TimeUtils.today_date()}.json'
+        dict_product = read_dict_from_file(cache_file)
+
+        skc_header = ['SKC', '商家SKC', 'SKC图片', '近7天利润', '近30天利润']
+        skc_excel_data = []
+        dict_skc = []
+
+        summary_excel_data = []
+        header = []
+        for store_username, excel_data in dict_product.items():
+            header = excel_data[0]
+            new_data = []
+            for row_item in excel_data[1:]:
+                supplier_sku = row_item[5]
+                row_item[10] = self.bridge.get_sku_cost(supplier_sku, erp)
+                new_data.append(row_item)
+
+                if row_item[2] not in dict_skc:
+                    dict_skc.append(row_item[2])
+                    stat_data = []
+                    stat_data.append(row_item[2])
+                    stat_data.append(row_item[3])
+                    stat_data.append(row_item[4])
+                    stat_data.append('')
+                    stat_data.append('')
+                    skc_excel_data.append(stat_data)
+
+            summary_excel_data += new_data
+
+        sheet_name = '商品库'
+
+        batch_excel_operations(excel_path, [
+            (sheet_name, 'write', [header] + summary_excel_data),
+            (sheet_name, 'format', self.format_product),
+        ])
+
+        sheet_name = 'Sheet1'
+        profit_data = [skc_header] + skc_excel_data
+        batch_excel_operations(excel_path, [
+            (sheet_name, 'write', sort_by_column(profit_data, 4, 1)),
+            (sheet_name, 'format', self.format_profit),
+            (sheet_name, 'format', sort_by_column_excel, 'E'),
+        ])
+
+    def format_profit(self, sheet):
+        beautify_title(sheet)
+        add_borders(sheet)
+        format_to_money(sheet, ['成本价', '核价', '利润'])
+        column_to_right(sheet, ['成本价', '核价', '利润'])
+        add_formula_for_column(sheet, '近7天利润', '=SUMIFS(商品库!L:L,商品库!P:P,A2)')
+        add_formula_for_column(sheet, '近30天利润', '=SUMIFS(商品库!M:M,商品库!P:P,A2)')
+        InsertImageV2(sheet, ['SKC图片'], 'shein', 90)
+
+    def format_product(self, sheet):
+        merge_by_column_v2(sheet, 'SPU', ['店铺信息', '产品信息'])
+        merge_by_column_v2(sheet, 'SKC', ['SKC图片', '商家SKC'])
+        beautify_title(sheet)
+        add_borders(sheet)
+        format_to_datetime(sheet, ['时间'])
+        format_to_money(sheet, ['成本价', '核价', '利润'])
+        column_to_right(sheet, ['成本价', '核价', '利润'])
+        autofit_column(sheet, ['产品信息'])
+        column_to_left(sheet, ['产品信息', '商家SKU', '商家SKC', '属性集'])
+        specify_column_width(sheet, ['店铺信息', '产品信息', '属性集'], 160 / 6)
+        specify_column_width(sheet, ['商家SKU', '商家SKC'], 220 / 6)
+        add_formula_for_column(sheet, '近7天利润', '=IF(ISNUMBER(K2), H2*(J2-K2),0)')
+        add_formula_for_column(sheet, '近30天利润', '=IF(ISNUMBER(K2), I2*(J2-K2),0)')
+        InsertImageV2(sheet, ['SKC图片'], 'shein', 150, '商家SKC', 'shein_skc_img')
+
     def write_week_ntb(self):
         excel_path = create_file_path(self.config.excel_week_report)
 

qrpa/shein_lib.py

@@ -1,3 +1,4 @@
+from docker.rq.config import today_date
 from .fun_file import read_dict_from_file, write_dict_to_file, read_dict_from_file_ex, write_dict_to_file_ex
 from .fun_base import log, send_exception, md5_string, get_safe_value, NetWorkIdleTimeout
 from .fun_web import fetch, fetch_get, full_screen_shot, safe_goto
@@ -80,11 +81,9 @@ class SheinLib:
 
         while retries < MAX_RETRIES:
             try:
-
                 retries += 1
 
                 while not web_page.locator('//div[contains(text(),"商家后台")]').nth(1).is_visible():
-
                     try:
                         current_url = web_page.url
                         current_title = web_page.title()
@@ -170,7 +169,6 @@ class SheinLib:
                     web_page.wait_for_timeout(3000)
 
                     if 'https://sso.geiwohuo.com/#/home' in web_page.url:
-
                         if 'SHEIN全球商家中心' in web_page.title() or '后台首页' in web_page.title() or '商家后台' in web_page.title():
                             log(web_page.title(), '中断循环', self.store_username, self.store_name)
                             web_page.wait_for_timeout(5000)
@@ -354,7 +352,6 @@ class SheinLib:
         return list_item
 
     def get_return_order_list(self, start_date, end_date, only_yesterday=1):
-
         log(f'获取退货列表: {self.store_username} {self.store_name} {start_date} {end_date}')
 
         page_num = 1
@@ -865,6 +862,63 @@ class SheinLib:
         write_dict_to_file(cache_file, info)
         return info
 
+    def get_product(self):
+        excel_data = [
+            ['店铺信息', '产品信息', 'SKC', '商家SKC', 'SKC图片', '商家SKU', '属性集', '近7天销量', '近30天销量', '核价', 'ERP成本价', '近7天利润', '近30天利润', '导出时间', 'SPU', 'SKC_FOR_STAT']
+        ]
+        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)
+        for skc_item in skc_list:
+            categoryName = skc_item['categoryName']
+            spu = skc_item['spu']
+            skc = skc_item['skc']
+            supplierCode = skc_item['supplierCode']
+            skc_img = skc_item['picUrl']
+            shelfDate = skc_item['shelfDate']
+            shelfDays = skc_item['shelfDays']
+            shelfStatusName = skc_item['shelfStatus']['name']
+            # if shelfStatusName != '已下架':
+            #     continue
+            for sku_item in skc_item['skuList']:
+                supplierSku = sku_item['supplierSku']
+                attr = sku_item['attr']
+                sku = sku_item['skuCode']
+                c7dSaleCnt = sku_item['c7dSaleCnt']
+                c30dSaleCnt = sku_item['c30dSaleCnt']
+                if attr == '合计' or int(c30dSaleCnt) == 0:
+                    log(f'跳过: {supplierSku},近30天销量: {c30dSaleCnt}')
+                    continue
+
+                price = dict_sku[sku]
+
+                product_info = f'SPU: {spu}\n商品分类: {categoryName}\n上架日期: {shelfDate}\n上架天数: {shelfDays}\n上架状态: {shelfStatusName}'
+
+                store_info = f'{self.store_username}\n{self.store_name}\n{self.config.shein_store_manager.get(self.store_username)}'
+
+                row_item = []
+                row_item.append(store_info)
+                row_item.append(product_info)
+                row_item.append(skc)
+                row_item.append(supplierCode)
+                row_item.append(skc_img)
+                row_item.append(supplierSku)
+                row_item.append(attr)
+                row_item.append(c7dSaleCnt)
+                row_item.append(c30dSaleCnt)
+                row_item.append(price)
+                row_item.append('')
+                row_item.append('')
+                row_item.append('')
+                row_item.append(TimeUtils.current_datetime())
+                row_item.append(spu)
+                row_item.append(skc)
+                excel_data.append(row_item)
+
+        cache_file = f'{self.config.auto_dir}/shein/product/product_{TimeUtils.today_date()}.json'
+        write_dict_to_file_ex(cache_file, {self.store_username: excel_data}, [self.store_username])
+        return excel_data
+
     # 存储商品库
     def store_product_info(self):
         # todo 商品详情 属性 规格 图片 重量 与 尺寸

qrpa/shein_mysql.py

@@ -1,62 +1,62 @@
-import json
-
-from .mysql_module.shein_return_order_model import SheinReturnOrderManager
-from .mysql_module.shein_product_model import SheinProductManager
-from .fun_base import log
-
-import os
-
-class SheinMysql:
-    def __init__(self, config):
-        self.config = config
-
-    def upsert_shein_return_order(self, json_file):
-        log(f'当前使用的数据库: {self.config.db.database_url}')
-        # 创建管理器实例
-        manager = SheinReturnOrderManager(self.config.db.database_url)
-        # 创建数据表
-        manager.create_tables()
-        # 读取JSON文件
-        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)
-
-    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'
-                    if os.path.exists(product_detail_file):
-                        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)
-                    else:
-                        log(f'文件不存在: {product_detail_file}')
-                    if os.path.exists(attribute_file):
-                        with open(attribute_file, 'r', encoding='utf-8') as f:
-                            data_list = json.load(f)
-                            manager.upsert_product_detail(spu, 'attribute_template', data_list)
-                    else:
-                        log(f'文件不存在: {attribute_file}')
+import json
+
+from .mysql_module.shein_return_order_model import SheinReturnOrderManager
+from .mysql_module.shein_product_model import SheinProductManager
+from .fun_base import log
+
+import os
+
+class SheinMysql:
+    def __init__(self, config):
+        self.config = config
+
+    def upsert_shein_return_order(self, json_file):
+        log(f'当前使用的数据库: {self.config.db.database_url}')
+        # 创建管理器实例
+        manager = SheinReturnOrderManager(self.config.db.database_url)
+        # 创建数据表
+        manager.create_tables()
+        # 读取JSON文件
+        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)
+
+    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'
+                    if os.path.exists(product_detail_file):
+                        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)
+                    else:
+                        log(f'文件不存在: {product_detail_file}')
+                    if os.path.exists(attribute_file):
+                        with open(attribute_file, 'r', encoding='utf-8') as f:
+                            data_list = json.load(f)
+                            manager.upsert_product_detail(spu, 'attribute_template', data_list)
+                    else:
+                        log(f'文件不存在: {attribute_file}')