Python实现Excel文件整文件密码保护的专业指南

import pandas as pd

def create_excel_file(data: dict, file_path: str):
    """
    使用pandas创建Excel文件。

    Args:
        data (dict): 包含列名和数据的字典。
        file_path (str): 输出Excel文件的路径。
    """
    df = pd.DataFrame(data)
    try:
        # 使用xlsxwriter引擎，也可以是openpyxl
        df.to_excel(file_path, index=False, engine='xlsxwriter')
        print(f"未加密Excel文件 '{file_path}' 创建成功。")
    except Exception as e:
        print(f"创建Excel文件失败: {e}")

# 示例数据
sample_data = {
    '姓名': ['张三', '李四', '王五'],
    '年龄': [25, 30, 28],
    '城市': ['北京', '上海', '广州']
}
unencrypted_file = "unencrypted_data.xlsx"
create_excel_file(sample_data, unencrypted_file)

import subprocess
import os

def encrypt_excel_file(input_file: str, output_file: str, password: str):
    """
    使用msoffice-crypt工具加密Excel文件。

    Args:
        input_file (str): 待加密的Excel文件路径。
        output_file (str): 加密后Excel文件的输出路径。
        password (str): 用于加密文件的密码。
    """
    # 检查输入文件是否存在
    if not os.path.exists(input_file):
        print(f"错误：输入文件 '{input_file}' 不存在。")
        return

    # msoffice-crypt 命令格式: msoffice-crypt -e -p <password> <input_file> <output_file>
    # -e: 加密模式
    # -p: 指定密码
    command = ["msoffice-crypt", "-e", "-p", password, input_file, output_file]

    try:
        # 执行命令行命令
        # check=True: 如果命令返回非零退出码，则抛出CalledProcessError
        # capture_output=True: 捕获stdout和stderr
        # text=True: 以文本模式处理输出
        result = subprocess.run(command, check=True, capture_output=True, text=True)

        print(f"Excel文件 '{input_file}' 成功加密为 '{output_file}'。")
        if result.stdout:
            print(f"msoffice-crypt 标准输出:\n{result.stdout.strip()}")
        if result.stderr:
            print(f"msoffice-crypt 标准错误:\n{result.stderr.strip()}")

        # 加密成功后，可以选择删除原始未加密文件
        # os.remove(input_file)
        # print(f"原始文件 '{input_file}' 已删除。")

    except FileNotFoundError:
        print("错误：'msoffice-crypt' 命令未找到。请确保该工具已安装且位于系统PATH中。")
    except subprocess.CalledProcessError as e:
        print(f"加密失败。错误码: {e.returncode}")
        print(f"命令输出:\n{e.stdout.strip()}")
        print(f"错误信息:\n{e.stderr.strip()}")
    except Exception as e:
        print(f"发生未知错误: {e}")

# 定义加密密码和输出文件路径
encryption_password = "YourSecurePassword123!" # 实际应用中应从安全配置中获取
encrypted_file = "encrypted_data_protected.xlsx"

# 执行加密
encrypt_excel_file(unencrypted_file, encrypted_file, encryption_password)