qrpa 1.0.23__py3-none-any.whl → 1.0.25__py3-none-any.whl

qrpa/__init__.py

@@ -3,7 +3,8 @@ from .db_migrator import DatabaseMigrator, DatabaseConfig, RemoteConfig, create_
 
 from .shein_ziniao import ZiniaoRunner
 
-from .fun_base import log, send_exception, md5_string, hostname, get_safe_value, sanitize_filename
+# from .fun_base import log, send_exception, md5_string, hostname, get_safe_value, sanitize_filename, get_file_size, calculate_star_symbols
+from .fun_base import *
 
 from .time_utils import TimeUtils
 
@@ -18,3 +19,7 @@ from .shein_excel import SheinExcel
 from .shein_lib import SheinLib
 
 from .fun_excel import InsertImageV2
+
+from .temu_lib import TemuLib
+from .temu_excel import TemuExcel
+from .temu_chrome import temu_chrome_excute

qrpa/fun_base.py

@@ -105,3 +105,233 @@ def copy_file(source, destination):
         print(f"错误：没有权限复制到 '{destination}'")
     except Exception as e:
         print(f"错误：发生未知错误 - {e}")
+
+def get_file_size(file_path, human_readable=False):
+    """
+    获取文件大小
+
+    :param file_path: 文件路径
+    :param human_readable: 是否返回可读格式（KB, MB, GB）
+    :return: 文件大小（字节数或可读格式）
+    """
+    if not os.path.isfile(file_path):
+        raise FileNotFoundError(f"文件不存在: {file_path}")
+
+    size_bytes = os.path.getsize(file_path)
+
+    if not human_readable:
+        return size_bytes
+
+    # 转换为可读单位
+    units = ["B", "KB", "MB", "GB", "TB"]
+    size = float(size_bytes)
+    for unit in units:
+        if size < 1024:
+            return f"{size:.2f} {unit}"
+        size /= 1024
+
+def calculate_star_symbols(rating):
+    """
+    计算星级对应的符号组合（独立评分逻辑函数）
+    参数:
+    rating (int): 标准化评分（0-5）
+    返回:
+    str: 星级符号字符串（如★★★⭐☆）
+    """
+    full_stars = int(rating)
+    empty_stars = 5 - full_stars
+    star_string = '★' * full_stars
+    star_string += '☆' * empty_stars
+    return star_string
+
+def remove_columns(matrix, indices):
+    """
+    过滤二维列表，移除指定索引的列
+
+    参数:
+        matrix: 二维列表
+        indices: 需要移除的列索引列表
+
+    返回:
+        过滤后的二维列表
+    """
+    # 创建要保留的索引集合（排除需要移除的索引）
+    indices_to_keep = set(range(len(matrix[0]))) - set(indices)
+
+    # 遍历每行，只保留不在indices中的列
+    return [[row[i] for i in indices_to_keep] for row in matrix]
+
+
+def filter_columns(matrix, indices):
+    """
+    过滤二维列表，只保留指定索引的列
+
+    参数:
+        matrix: 二维列表
+        indices: 需要保留的列索引列表
+
+    返回:
+        过滤后的二维列表
+    """
+    # 转置矩阵，获取每一列
+    columns = list(zip(*matrix))
+
+    # 只保留指定索引的列
+    filtered_columns = [columns[i] for i in indices]
+
+    # 将过滤后的列转回二维列表
+    return [list(row) for row in zip(*filtered_columns)]
+
+
+# # 示例使用
+# matrix = [
+#     [1, 2, 3, 4],
+#     [5, 6, 7, 8],
+#     [9, 10, 11, 12]
+# ]
+#
+# # 只保留索引为 0 和 2 的列
+# filtered = filter_columns(matrix, [0, 2])
+# print(filtered)  # 输出: [[1, 3], [5, 7], [9, 11]]
+
+def add_column_to_2d_list(data, new_col, index=None):
+    """
+    给二维列表增加一列数据
+
+    :param data: 原始二维列表，例如 [[1, 2], [3, 4]]
+    :param new_col: 要添加的新列数据，例如 [10, 20]
+    :param index: 插入位置，默认为最后一列之后；支持负数索引
+    :return: 增加新列后的二维列表
+    """
+    if not data:
+        raise ValueError("原始数据为空")
+    if len(data) != len(new_col):
+        raise ValueError("新列长度必须与原始数据的行数相等")
+
+    new_data = []
+    for i, row in enumerate(data):
+        row = list(row)  # 防止修改原始数据
+        insert_at = index if index is not None else len(row)
+        row.insert(insert_at, new_col[i])
+        new_data.append(row)
+    return new_data
+
+
+def add_prefixed_column(data, header, value):
+    """
+    给二维列表增加第一列，第一行为 header，后面为 value。
+
+    :param data: 原始二维列表
+    :param header: 新列的标题
+    :param value: 新列内容（相同值）
+    :return: 增加新列后的二维列表
+    """
+    if not data:
+        raise ValueError("原始数据不能为空")
+
+    new_col = [header] + [value] * (len(data) - 1)
+    return [[new_col[i]] + row for i, row in enumerate(data)]
+
+
+def add_suffixed_column(data, header, value):
+    """
+    给二维列表增加第一列，第一行为 header，后面为 value。
+
+    :param data: 原始二维列表
+    :param header: 新列的标题
+    :param value: 新列内容（相同值）
+    :return: 增加新列后的二维列表
+    """
+    if not data:
+        raise ValueError("原始数据不能为空")
+
+    new_col = [header] + [value] * (len(data) - 1)
+    return [row + [new_col[i]] for i, row in enumerate(data)]
+
+
+def merge_2d_lists_keep_first_header(data1, data2):
+    """
+    合并两个二维列表，只保留第一个列表的标题（即第一行）。
+
+    :param data1: 第一个二维列表（包含标题）
+    :param data2: 第二个二维列表（包含标题）
+    :return: 合并后的二维列表
+    """
+    if not data1 or not isinstance(data1, list):
+        raise ValueError("data1 不能为空并且必须是二维列表")
+    if not data2 or not isinstance(data2, list):
+        raise ValueError("data2 不能为空并且必须是二维列表")
+
+    header = data1[0]
+    rows1 = data1[1:]
+    rows2 = data2[1:]
+
+    return [header] + rows1 + rows2
+
+
+def insert_total_row(data, row_index=1, label="合计"):
+    """
+    在指定行插入一行，第一列为 label，其余为空字符串。
+
+    :param data: 原始二维列表
+    :param row_index: 插入位置，默认插在第二行（索引1）
+    :param label: 第一列的标签内容，默认为 "合计"
+    :return: 新的二维列表
+    """
+    if not data or not isinstance(data, list):
+        raise ValueError("data 不能为空并且必须是二维列表")
+
+    num_cols = len(data[0])
+    new_row = [label] + [""] * (num_cols - 1)
+
+    return data[:row_index] + [new_row] + data[row_index:]
+
+
+def insert_empty_column_after(data, col_index, new_header="单价成本"):
+    """
+    在二维列表中指定列的后面插入一个新列，标题为 new_header，其余内容为空字符串。
+
+    :param data: 原始二维列表
+    :param col_index: 要插入的位置（在该列后面插入）
+    :param new_header: 新列的标题
+    :return: 新的二维列表
+    """
+    if not data or not isinstance(data, list):
+        raise ValueError("data 不能为空且必须是二维列表")
+
+    new_data = []
+    for i, row in enumerate(data):
+        row = list(row)  # 复制避免修改原数据
+        insert_value = new_header if i == 0 else ""
+        row.insert(col_index + 1, insert_value)
+        new_data.append(row)
+
+    return new_data
+
+
+def insert_empty_column_after_column_name(data, target_col_name, new_header="单价成本"):
+    """
+    在指定列名对应的列后面插入一个新列，标题为 new_header，其余行为空字符串。
+
+    :param data: 原始二维列表
+    :param target_col_name: 要在哪一列之后插入（通过列标题匹配）
+    :param new_header: 插入的新列标题
+    :return: 新的二维列表
+    """
+    if not data or not isinstance(data, list):
+        raise ValueError("data 不能为空且必须是二维列表")
+
+    header = data[0]
+    if target_col_name not in header:
+        raise ValueError(f"找不到列名：{target_col_name}")
+
+    col_index = header.index(target_col_name)
+
+    new_data = []
+    for i, row in enumerate(data):
+        row = list(row)  # 防止修改原始数据
+        insert_value = new_header if i == 0 else ""
+        row.insert(col_index + 1, insert_value)
+        new_data.append(row)
+
+    return new_data

qrpa/fun_excel.py

@@ -12,6 +12,7 @@ import concurrent.futures
 from collections import defaultdict
 import threading
 from playwright.sync_api import sync_playwright
+import psutil
 
 from .fun_base import log, sanitize_filename, create_file_path, copy_file, add_https, send_exception
 
@@ -89,13 +90,13 @@ def set_cell_prefix_red(cell, n, color_name):
     except Exception as e:
         print(f"设置字体颜色失败: {e}")
 
-def sort_by_column(data, col_index, start_row=2, reverse=True):
-    if not data or start_row >= len(data):
+def sort_by_column(data, col_index, header_rows=2, reverse=True):
+    if not data or header_rows >= len(data):
         return data
 
     try:
-        header = data[:start_row]
-        new_data_sorted = data[start_row:]
+        header = data[:header_rows]
+        new_data_sorted = data[header_rows:]
 
         def get_key(row):
             value = row[col_index]
@@ -541,11 +542,11 @@ def insert_fixed_scale_image(sheet, cell, image_path, scale=1.0):
 
     return None
 
-def InsertImageV2(app, wb, sheet, columns=None, platform='shein', img_width=150, img_save_key=None, dir_name=None, cell_height_with_img=False):
+def InsertImageV2(sheet, columns=None, platform='shein', img_width=150, img_save_key=None, dir_name=None, cell_height_with_img=False, start_row=2):
     if not columns:
         return
 
-    minimize(app)
+    minimize(sheet.book.app)
 
     # 清空所有图片
     clear_all_pictures(sheet)
@@ -574,7 +575,7 @@ def InsertImageV2(app, wb, sheet, columns=None, platform='shein', img_width=150,
 
     # 预计算所有单元格的合并区域信息 (优化点1)
     area_map = {}
-    for row in range(2, last_row + 1):
+    for row in range(start_row, last_row + 1):
         log(f'计算 {row}/{last_row}')  # 如果数据量非常大，这里的日志会影响性能，可以考虑优化
         for col_letter in col_letter_map.values():
             cell_ref = f'{col_letter}{row}'
@@ -615,7 +616,7 @@ def InsertImageV2(app, wb, sheet, columns=None, platform='shein', img_width=150,
                 }
 
     # 处理图片插入 (优化点2)
-    for row in range(2, last_row + 1):
+    for row in range(start_row, last_row + 1):
         for img_col_name, col_letter in col_letter_map.items():
             cell_ref = f'{col_letter}{row}'
             cell_range = sheet.range(cell_ref)
@@ -1174,8 +1175,8 @@ def write_data(excel_path, sheet_name, data, format_to_text_colunm=None):
     wb.save()
     close_excel(app, wb)
 
-def colorize_by_field(app, wb, sheet, field):
-    minimize(app)
+def colorize_by_field(sheet, field):
+    minimize(sheet.book.app)
     # 读取数据
     field_column = find_column_by_data(sheet, 1, field)  # 假设 SPU 在 C 列
     if field_column is None:
@@ -2061,6 +2062,21 @@ def format_to_number(sheet, columns=None, decimal_places=2):
 #                 else:
 #                     sheet.range(f'{col_name}:{col_name}').number_format = f'0.{"0" * decimal_places}'
 
+def hidden_columns(sheet, columns=None):
+    if columns is None:
+        return
+    used_range_col = sheet.range('A1').expand('right')
+    for j, cell in enumerate(used_range_col):
+        col = j + 1
+        col_name = index_to_column_name(col)
+        col_val = sheet.range(f'{col_name}1').value
+        if col_val is None:
+            continue
+        for c in columns:
+            if c in col_val:
+                log(f'设置[{c}] 隐藏')
+                sheet.range(f'{col_name}:{col_name}').column_width = 0
+
 def column_to_right(sheet, columns=None):
     if columns is None:
         return
@@ -2076,7 +2092,7 @@ def column_to_right(sheet, columns=None):
                 # 水平对齐： # -4108：居中 # -4131：左对齐 # -4152：右对齐
                 # 垂直对齐： # -4108：居中 # -4160：顶部对齐 # -4107：底部对齐
                 # 所有列水平居中和垂直居中
-                log(f'设置[{c}] 水平垂直居中')
+                log(f'设置[{c}] 水平右对齐')
                 sheet.range(f'{col_name}:{col_name}').api.HorizontalAlignment = -4152
                 sheet.range(f'{col_name}:{col_name}').api.VerticalAlignment = -4108
 
@@ -2117,6 +2133,16 @@ def beautify_title(sheet):
         sheet.range(f'{col_name}:{col_name}').api.VerticalAlignment = -4108
     sheet.autofit()
 
+def set_body_style(sheet, row_start, row_end=None):
+    if row_end is None:
+        row_end = get_last_used_row(sheet)
+
+    range = sheet.range(f'{row_start}:{row_end}')
+    # 设置字体名称
+    range.font.name = 'Calibri'
+    # 设置字体大小
+    range.font.size = 11
+
 def set_title_style(sheet, rows=2):
     col = get_max_column_letter(sheet)
     range = sheet.range(f'A1:{col}{rows}')
@@ -2405,6 +2431,7 @@ def format_excel_with_lock(excel_path, sheet_name, format_func, *args, **kwargs)
         log(f"格式化失败: {e}")
         return False
 
+# 经过观察 fortmat时 传入函数需要为类函数且第二个参数必须是 sheet
 def batch_excel_operations(excel_path, operations):
     """
     批量 Excel 操作函数，一次性打开 Excel 执行多个操作
@@ -2439,17 +2466,19 @@ def batch_excel_operations(excel_path, operations):
             sheet.activate()
 
             if operation_type == 'write':
-                data, format_to_text_colunm = args[:2]
+                data, format_to_text_colunm = args[0], args[1:] if len(args) > 1 else None
                 # 清空工作表
                 sheet.clear()
                 # 格式化文本列
-                format_to_text_v2(sheet, format_to_text_colunm)
+                if format_to_text_colunm:
+                    format_to_text_v2(sheet, format_to_text_colunm)
                 # 写入数据
                 sheet.range('A1').value = data
                 log(f"批量操作：写入数据到 {sheet_name}")
 
             elif operation_type == 'format':
                 format_func, format_args = args[0], args[1:] if len(args) > 1 else ()
+                log('格式化入参', *format_args)
                 # 执行格式化
                 format_func(sheet, *format_args)
                 log(f"批量操作：格式化工作表 {sheet_name}")
@@ -2478,6 +2507,26 @@ def batch_excel_operations(excel_path, operations):
     finally:
         # 释放锁但不关闭 Excel（保持复用）
         excel_lock_manager.release_excel_lock(excel_path)
+        close_excel_with_lock(excel_path, app, wb, True)
+
+def close_excel_file(file_path):
+    file_path = os.path.abspath(file_path).lower()
+
+    for proc in psutil.process_iter(['pid', 'name']):
+        if proc.info['name'] and proc.info['name'].lower() in ['excel.exe', 'wps.exe']:  # 只找 Excel
+            try:
+                for f in proc.open_files():
+                    if os.path.abspath(f.path).lower() == file_path:
+                        print(f"文件被 Excel 占用 (PID: {proc.pid})，正在关闭进程...")
+                        proc.terminate()
+                        proc.wait(timeout=3)
+                        print("已关闭。")
+                        return True
+            except (psutil.NoSuchProcess, psutil.AccessDenied):
+                continue
+
+    print("文件没有被 Excel 占用。")
+    return False
 
 def force_close_excel_file(excel_path):
     """
@@ -2673,3 +2722,6 @@ def get_excel_status(excel_path):
         'operation_count': excel_lock_manager.get_operation_count(excel_path),
         'has_lock'       : excel_lock_manager.get_file_lock(excel_path).locked()
     }
+
+def get_last_used_row(sheet):
+    return sheet.used_range.last_cell.row