qrpa 1.0.15__py3-none-any.whl → 1.0.17__py3-none-any.whl

qrpa/__init__.py

@@ -9,9 +9,12 @@ from .time_utils import TimeUtils
 
 from .fun_file import read_dict_from_file, read_dict_from_file_ex, write_dict_to_file, write_dict_to_file_ex
 from .fun_file import get_progress_json_ex, check_progress_json_ex, done_progress_json_ex
+from .fun_file import delete_file, delete_file_simple
 
 from .fun_web import fetch, fetch_via_iframe, find_all_iframe, full_screen_shot
 from .fun_win import *
 
 from .shein_excel import SheinExcel
 from .shein_lib import SheinLib
+
+from .fun_excel import InsertImageV2

qrpa/fun_file.py

@@ -8,6 +8,152 @@ from datetime import date, datetime, timedelta, timezone
 
 from .fun_base import log
 
+import os
+import gc
+import time
+import psutil
+from pathlib import Path
+
+def delete_file(file_path):
+    """
+    删除文件的优化函数，自动处理已打开的文件
+
+    Args:
+        file_path (str): 要删除的文件路径
+
+    Returns:
+        bool: 删除成功返回True，失败返回False
+    """
+    try:
+        file_path = Path(file_path).resolve()  # 规范化路径
+
+        if not file_path.exists():
+            log(f"文件 {file_path} 不存在。")
+            return False
+
+        # 强制垃圾回收，关闭可能的文件句柄
+        gc.collect()
+
+        # 第一次尝试直接删除
+        try:
+            os.remove(file_path)
+            log(f"文件 {file_path} 已成功删除。")
+            return True
+        except PermissionError:
+            log(f"文件 {file_path} 可能被占用，尝试关闭相关进程...")
+
+            # 尝试找到并关闭占用该文件的进程
+            if close_file_handles(file_path):
+                # 等待一小段时间让系统释放句柄
+                time.sleep(0.1)
+
+                # 再次尝试删除
+                try:
+                    os.remove(file_path)
+                    log(f"文件 {file_path} 已成功删除。")
+                    return True
+                except PermissionError:
+                    log(f"错误：即使尝试关闭文件句柄后，仍无法删除文件 {file_path}。")
+                    return False
+            else:
+                log(f"错误：无法关闭文件 {file_path} 的句柄，删除失败。")
+                return False
+
+    except FileNotFoundError:
+        log(f"错误：文件 {file_path} 未找到。")
+        return False
+    except Exception as e:
+        log(f"错误：删除文件时发生未知错误：{e}")
+        return False
+
+def close_file_handles(file_path):
+    """
+    尝试关闭指定文件的所有句柄
+
+    Args:
+        file_path (Path): 文件路径
+
+    Returns:
+        bool: 成功关闭返回True，失败返回False
+    """
+    try:
+        file_path_str = str(file_path)
+        closed_any = False
+
+        # 遍历所有进程，查找打开该文件的进程
+        for proc in psutil.process_iter(['pid', 'name']):
+            try:
+                # 获取进程打开的文件列表
+                open_files = proc.open_files()
+                for open_file in open_files:
+                    if os.path.samefile(open_file.path, file_path_str):
+                        log(f"发现进程 {proc.info['name']} (PID: {proc.info['pid']}) 正在使用文件")
+
+                        # 如果是当前Python进程，尝试强制关闭文件句柄
+                        if proc.info['pid'] == os.getpid():
+                            closed_any = True
+                        else:
+                            # 对于其他进程，可以选择终止（谨慎使用）
+                            log(f"警告：文件被其他进程占用，请手动关闭应用程序")
+
+            except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
+                continue
+            except Exception as e:
+                continue
+
+        return closed_any
+
+    except Exception as e:
+        log(f"关闭文件句柄时出错：{e}")
+        return False
+
+def delete_file_simple(file_path):
+    """
+    简化版本的删除文件函数（不需要额外依赖）
+
+    Args:
+        file_path (str): 要删除的文件路径
+
+    Returns:
+        bool: 删除成功返回True，失败返回False
+    """
+    try:
+        file_path = os.path.abspath(file_path)  # 获取绝对路径
+
+        if not os.path.exists(file_path):
+            log(f"文件 {file_path} 不存在。")
+            return False
+
+        # 强制垃圾回收
+        gc.collect()
+
+        # 尝试删除
+        try:
+            os.remove(file_path)
+            log(f"文件 {file_path} 已成功删除。")
+            return True
+        except PermissionError:
+            # 在Windows上，尝试修改文件权限
+            if os.name == 'nt':  # Windows
+                try:
+                    os.chmod(file_path, 0o777)
+                    time.sleep(0.1)  # 短暂等待
+                    os.remove(file_path)
+                    log(f"文件 {file_path} 已成功删除。")
+                    return True
+                except Exception:
+                    pass
+
+            log(f"错误：没有权限删除文件 {file_path}，可能文件正在被使用。")
+            return False
+
+    except FileNotFoundError:
+        log(f"错误：文件 {file_path} 未找到。")
+        return False
+    except Exception as e:
+        log(f"错误：删除文件时发生未知错误：{e}")
+        return False
+
 def read_dict_from_file(file_path, cache_interval=3600 * 24 * 365 * 10):
     """
     从文件中读取字典。
@@ -170,4 +316,4 @@ def check_progress_json_ex(config, task_key, just_store_username=None):
                 return False
     else:
         log(f"进度文件不存在或为空: {progress_file}")
-    return True
+    return True

qrpa/shein_excel.py

@@ -79,3 +79,200 @@ class SheinExcel:
                 self.format_bak_advice(new_excel_path, sheet_name, mode)
 
         return new_excel_path_list
+
+    def write_activity_list(self):
+        cache_file = f'{self.config.auto_dir}/shein/activity_list/activity_list_{TimeUtils.today_date()}.json'
+        dict_activity = read_dict_from_file(cache_file)
+        all_data = []
+        header = []
+        for store_username, excel_data in dict_activity.items():
+            header = excel_data[:1]
+            all_data += excel_data[1:]
+
+        all_data = header + all_data
+
+        excel_path = create_file_path(self.config.excel_activity_list)
+        sheet_name = 'Sheet1'
+        write_data(excel_path, sheet_name, all_data)
+        self.format_activity_list(excel_path, sheet_name)
+
+    def format_activity_list(self, excel_path, sheet_name):
+        app, wb, sheet = open_excel(excel_path, sheet_name)
+        beautify_title(sheet)
+        add_borders(sheet)
+        column_to_left(sheet, ['活动信息'])
+        colorize_by_field(app, wb, sheet, '店铺名称')
+        autofit_column(sheet, ['店铺名称', '活动信息'])
+        wb.save()
+        close_excel(app, wb)
+
+    def write_jit_data(self):
+        excel_path_1 = create_file_path(self.config.Excel_Order_Type_1)
+        summary_excel_data_1 = []
+
+        cache_file_1 = f'{self.config.auto_dir}/shein/cache/jit_{TimeUtils.today_date()}_1_{TimeUtils.get_period()}.json'
+        dict_1 = read_dict_from_file(cache_file_1)
+        dict_store = read_dict_from_file(f'{self.config.auto_dir}/shein_store_alias.json')
+
+        header = []
+        for store_username, excel_data in dict_1.items():
+            store_name = dict_store.get(store_username)
+            sheet_name = store_name
+            write_data(excel_path_1, sheet_name, excel_data)
+            self.format_jit(excel_path_1, sheet_name)
+            header = excel_data[0]
+            summary_excel_data_1 += excel_data[1:]
+
+        if len(summary_excel_data_1) > 0:
+            sheet_name = 'Sheet1'
+            write_data(excel_path_1, sheet_name, [header] + summary_excel_data_1)
+            self.format_jit(excel_path_1, sheet_name)
+
+        excel_path_2 = create_file_path(self.config.Excel_Order_Type_2)
+        summary_excel_data_2 = []
+
+        cache_file_2 = f'{self.config.auto_dir}/shein/cache/jit_{TimeUtils.today_date()}_2_{TimeUtils.get_period()}.json'
+        dict_2 = read_dict_from_file(cache_file_2)
+
+        header = []
+        for store_username, excel_data in dict_2.items():
+            store_name = dict_store.get(store_username)
+            sheet_name = store_name
+            write_data(excel_path_2, sheet_name, excel_data)
+            self.format_jit(excel_path_2, sheet_name)
+            header = excel_data[0]
+            summary_excel_data_2 += excel_data[1:]
+
+        if len(summary_excel_data_2) > 0:
+            sheet_name = 'Sheet1'
+            write_data(excel_path_2, sheet_name, [header] + summary_excel_data_2)
+            self.format_jit(excel_path_2, sheet_name)
+
+    def format_jit(self, excel_path, sheet_name):
+        app, wb, sheet = open_excel(excel_path, sheet_name)
+        beautify_title(sheet)
+        add_borders(sheet)
+        colorize_by_field(app, wb, sheet, 'SKC')
+        column_to_left(sheet, ["商品信息", "近7天SKU销量/SKC销量/SKC曝光", "SKC点击率/SKC转化率", "自主参与活动"])
+        autofit_column(sheet,
+                       ['店铺名称', '商品信息', "近7天SKU销量/SKC销量/SKC曝光", "SKC点击率/SKC转化率", "自主参与活动"])
+        InsertImageV2(app, wb, sheet, ['SKC图片', 'SKU图片'])
+        wb.save()
+        close_excel(app, wb)
+
+    def write_week_report(self):
+        excel_path = create_file_path(self.config.excel_week_sales_report)
+        log(excel_path)
+
+        cache_file = f'{self.config.auto_dir}/shein/cache/week_sales_{TimeUtils.today_date()}.json'
+        dict = read_dict_from_file(cache_file)
+
+        summary_excel_data = []
+        header = []
+        for store_name, excel_data in dict.items():
+            sheet_name = store_name
+            write_data(excel_path, sheet_name, excel_data)
+            self.format_week_report(excel_path, sheet_name)
+            header = excel_data[0]
+            summary_excel_data += excel_data[1:]
+        summary_excel_data = [header] + summary_excel_data
+        sheet_name = 'Sheet1'
+        write_data(excel_path, sheet_name, summary_excel_data)
+        self.format_week_report(excel_path, sheet_name)
+
+    def format_week_report(self, excel_path, sheet_name):
+        app, wb, sheet = open_excel(excel_path, sheet_name)
+        beautify_title(sheet)
+        column_to_left(sheet, ['商品信息'])
+        format_to_money(sheet, ['申报价', '成本价', '毛利润', '利润'])
+        format_to_percent(sheet, ['支付率', '点击率', '毛利率'])
+        self.dealFormula(sheet)  # 有空再封装优化
+        colorize_by_field(app, wb, sheet, 'SPU')
+        autofit_column(sheet, ['店铺名称'])
+        specify_column_width(sheet, ['商品标题'], 150 / 6)
+        add_borders(sheet)
+        InsertImageV2(app, wb, sheet, ['SKC图片', 'SKU图片'], 'shein', 90)
+        wb.save()
+        close_excel(app, wb)
+
+    # 处理公式计算
+    def dealFormula(self, sheet):
+        # 增加列 周销增量 月销增量
+        col_week_increment = find_column_by_data(sheet, 1, '周销增量')
+        if col_week_increment is None:
+            col_week_increment = find_column_by_data(sheet, 1, '远30天销量')
+            log(f'{col_week_increment}:{col_week_increment}')
+            sheet.range(f'{col_week_increment}:{col_week_increment}').insert('right')
+            sheet.range(f'{col_week_increment}1').value = '周销增量'
+            log('已增加列 周销增量')
+
+        col_month_increment = find_column_by_data(sheet, 1, '月销增量')
+        if col_month_increment is None:
+            col_month_increment = find_column_by_data(sheet, 1, '总销量')
+            log(f'{col_month_increment}:{col_month_increment}')
+            sheet.range(f'{col_month_increment}:{col_month_increment}').insert('right')
+            sheet.range(f'{col_month_increment}1').value = '月销增量'
+            log('已增加列 月销增量')
+
+        col_month_profit = find_column_by_data(sheet, 1, '近30天利润')
+        if col_month_profit is None:
+            col_month_profit = find_column_by_data(sheet, 1, '总利润')
+            sheet.range(f'{col_month_profit}:{col_month_profit}').insert('right')
+            log((f'{col_month_profit}:{col_month_profit}'))
+            sheet.range(f'{col_month_profit}1').value = '近30天利润'
+            log('已增加列 近30天利润')
+
+        col_week_profit = find_column_by_data(sheet, 1, '近7天利润')
+        if col_week_profit is None:
+            col_week_profit = find_column_by_data(sheet, 1, '近30天利润')
+            sheet.range(f'{col_week_profit}:{col_week_profit}').insert('right')
+            log((f'{col_week_profit}:{col_week_profit}'))
+            sheet.range(f'{col_week_profit}1').value = '近7天利润'
+            log('已增加列 近7天利润')
+
+        # return
+
+        # 查找 申报价，成本价，毛利润，毛利润率 所在列
+        col_verify_price = find_column_by_data(sheet, 1, '申报价')
+        col_cost_price = find_column_by_data(sheet, 1, '成本价')
+        col_gross_profit = find_column_by_data(sheet, 1, '毛利润')
+        col_gross_margin = find_column_by_data(sheet, 1, '毛利率')
+
+        col_week_1 = find_column_by_data(sheet, 1, '近7天销量')
+        col_week_2 = find_column_by_data(sheet, 1, '远7天销量')
+        col_month_1 = find_column_by_data(sheet, 1, '近30天销量')
+        col_month_2 = find_column_by_data(sheet, 1, '远30天销量')
+
+        # 遍历可用行
+        used_range_row = sheet.range('A1').expand('down')
+        for i, cell in enumerate(used_range_row):
+            row = i + 1
+            if row < 2:
+                continue
+            rangeA = f'{col_verify_price}{row}'
+            rangeB = f'{col_cost_price}{row}'
+
+            rangeC = f'{col_week_increment}{row}'
+            rangeD = f'{col_month_increment}{row}'
+
+            # rangeE = f'{col_total_profit}{row}'
+            rangeF = f'{col_month_profit}{row}'
+            rangeG = f'{col_week_profit}{row}'
+
+            # 设置毛利润和毛利润率列公式与格式
+            sheet.range(f'{col_gross_profit}{row}').formula = f'=IF(ISNUMBER({rangeB}),{rangeA}-{rangeB},"")'
+            sheet.range(f'{col_gross_profit}{row}').number_format = '0.00'
+            sheet.range(f'{col_gross_margin}{row}').formula = f'=IF(ISNUMBER({rangeB}),({rangeA}-{rangeB})/{rangeA},"")'
+            sheet.range(f'{col_gross_margin}{row}').number_format = '0.00%'
+
+            sheet.range(rangeC).formula = f'={col_week_1}{row}-{col_week_2}{row}'
+            sheet.range(rangeC).number_format = '0'
+            sheet.range(rangeD).formula = f'={col_month_1}{row}-{col_month_2}{row}'
+            sheet.range(rangeD).number_format = '0'
+
+            # sheet.range(rangeE).formula = f'=IF(ISNUMBER({rangeB}),{col_total}{row}*{col_gross_profit}{row},"")'
+            # sheet.range(rangeE).number_format = '0.00'
+            sheet.range(rangeF).formula = f'=IF(ISNUMBER({rangeB}),{col_month_1}{row}*{col_gross_profit}{row},"")'
+            sheet.range(rangeF).number_format = '0.00'
+            sheet.range(rangeG).formula = f'=IF(ISNUMBER({rangeB}),{col_week_1}{row}*{col_gross_profit}{row},"")'
+            sheet.range(rangeG).number_format = '0.00'