快捷導(dǎo)航

Python批量加密Excel文件的實(shí)現(xiàn)示例

更新時間：2023年12月08日 15:31:53 作者：Sitin濤哥

在日常工作中,保護(hù)敏感數(shù)據(jù)是至關(guān)重要的,本文主要介紹了Python批量加密Excel文件的實(shí)現(xiàn)示例,具有一定的參考價值,感興趣的可以了解一下

安裝所需庫

首先，確保已經(jīng)安裝了 openpyxl 庫，它是一個強(qiáng)大的處理Excel文件的庫。

pip install openpyxl

加密單個Excel文件

先從加密單個Excel文件開始。以下是一個基本的示例代碼：

from openpyxl import load_workbook
from openpyxl.styles import Protection

def encrypt_excel(file_path, password):
    workbook = load_workbook(file_path)
    for sheet in workbook.sheetnames:
        ws = workbook[sheet]
        ws.protection.sheet = True
        ws.protection.password = password

    encrypted_file_path = file_path.replace(".xlsx", "_encrypted.xlsx")
    workbook.save(encrypted_file_path)

# 使用示例
file_path = "example.xlsx"
password = "securepassword"
encrypt_excel(file_path, password)

這個函數(shù)將保護(hù)Excel文件中的每個工作表，并使用提供的密碼進(jìn)行加密。加密后的文件會保存在原文件名基礎(chǔ)上追加 "_encrypted"。

批量加密多個Excel文件

如果有一個文件夾中的多個Excel文件需要批量加密，我們可以編寫一個函數(shù)來實(shí)現(xiàn)這一任務(wù)。

以下是一個示例代碼：

import os

def batch_encrypt_excel(folder_path, password):
    for file_name in os.listdir(folder_path):
        if file_name.endswith(".xlsx"):
            file_path = os.path.join(folder_path, file_name)
            encrypt_excel(file_path, password)

# 使用示例
folder_path = "folder_with_excel_files"
password = "securepassword"
batch_encrypt_excel(folder_path, password)

這個函數(shù)遍歷指定文件夾中的所有Excel文件，并對每個文件調(diào)用之前編寫的 encrypt_excel 函數(shù)進(jìn)行加密。

解密Excel文件

當(dāng)需要查看或編輯加密的Excel文件時，需要提供正確的密碼進(jìn)行解密。

以下是解密的示例代碼：

def decrypt_excel(file_path, password):
    workbook = load_workbook(file_path, read_only=False, keep_vba=True)
    for sheet in workbook.sheetnames:
        ws = workbook[sheet]
        ws.protection.sheet = False
        ws.protection.password = password

    decrypted_file_path = file_path.replace("_encrypted.xlsx", "_decrypted.xlsx")
    workbook.save(decrypted_file_path)

# 使用示例
encrypted_file_path = "example_encrypted.xlsx"
password = "securepassword"
decrypt_excel(encrypted_file_path, password)

這個函數(shù)將解密Excel文件中的每個工作表，并使用提供的密碼進(jìn)行解密。解密后的文件會保存在原文件名基礎(chǔ)上追加 "_decrypted"。

自定義加密和解密選項(xiàng)

除了基本的加密和解密，可能希望在保護(hù)文件時進(jìn)行一些自定義設(shè)置，比如只允許特定操作或者只允許特定用戶編輯。Protection 類提供了一些選項(xiàng)，可以根據(jù)需求進(jìn)行定制。

以下是一個示例，演示如何限制對工作表進(jìn)行排序和篩選的權(quán)限：

from openpyxl.styles import Protection

def custom_encrypt_excel(file_path, password):
    workbook = load_workbook(file_path)
    for sheet in workbook.sheetnames:
        ws = workbook[sheet]
        ws.protection.sheet = True
        ws.protection.password = password
        ws.protection.enableFormatCells = False  # 禁止格式更改
        ws.protection.enableInsertRows = False    # 禁止插入行
        ws.protection.enableSort = False          # 禁止排序
        ws.protection.enableAutoFilter = False    # 禁止自動篩選

    encrypted_file_path = file_path.replace(".xlsx", "_custom_encrypted.xlsx")
    workbook.save(encrypted_file_path)

# 使用示例
file_path = "example.xlsx"
password = "securepassword"
custom_encrypt_excel(file_path, password)

使用 getpass 模塊隱藏密碼輸入

為了提高安全性，你可以使用 getpass 模塊隱藏密碼的輸入過程，確保密碼不會明文顯示在終端上。

以下是一個示例代碼：

from getpass import getpass

def get_password():
    return getpass("Enter the password: ")

# 使用示例
password = get_password()
print("Password entered:", password)

在實(shí)際應(yīng)用中，可以調(diào)整代碼以在加密/解密操作中使用 get_password 函數(shù)，保障密碼的安全性。

異常處理

在實(shí)際應(yīng)用中，始終要考慮到潛在的錯誤。在文件操作中，文件可能不存在、密碼錯誤等情況都是需要處理的異常。

以下是一個簡單的異常處理示例：

from openpyxl import load_workbook
from openpyxl.utils.exceptions import InvalidFileException

def encrypt_excel(file_path, password):
    try:
        workbook = load_workbook(file_path)
        for sheet in workbook.sheetnames:
            ws = workbook[sheet]
            ws.protection.sheet = True
            ws.protection.password = password

        encrypted_file_path = file_path.replace(".xlsx", "_encrypted.xlsx")
        workbook.save(encrypted_file_path)
        print(f"Encryption successful. Encrypted file saved at: {encrypted_file_path}")

    except InvalidFileException:
        print(f"Error: Invalid file format. Please provide a valid Excel file.")
    except Exception as e:
        print(f"An unexpected error occurred: {e}")

# 使用示例
file_path = "nonexistent_file.xlsx"
password = "securepassword"
encrypt_excel(file_path, password)

通過合理的異常處理，你可以更好地應(yīng)對各種可能發(fā)生的問題，提高腳本的魯棒性。

安全注意事項(xiàng)

密碼管理： 確保安全地管理密碼，不要將密碼明文存儲在腳本或其他不安全的地方。
備份文件： 在加密或解密操作前，最好先備份原始文件，以防不測之需。

總結(jié)

在這篇文章中，深入研究了如何通過Python實(shí)現(xiàn)批量加密和解密Excel文件，為保護(hù)敏感數(shù)據(jù)提供了全面的指南。我們從安裝必要的庫、加密單個文件，到批量加密多個文件和解密文件，甚至介紹了如何自定義加密選項(xiàng)和提高安全性。示例代碼詳細(xì)演示了每個步驟的操作，能夠輕松應(yīng)用這一實(shí)用技能。

除了基礎(chǔ)的操作外，還介紹了一些高級主題，如自定義加密選項(xiàng)，通過 getpass 模塊隱藏密碼輸入，以及合理的異常處理。這些補(bǔ)充內(nèi)容更好地理解并應(yīng)用于實(shí)際場景中。

通過本文，不僅能夠加密和解密Excel文件，還能夠根據(jù)需求進(jìn)行一些高級的定制。在日常工作中，這項(xiàng)技能可用于確保數(shù)據(jù)的安全性，同時提供了靈活的數(shù)據(jù)處理手段。

到此這篇關(guān)于Python批量加密Excel文件的實(shí)現(xiàn)示例的文章就介紹到這了,更多相關(guān)Python批量加密Excel內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: