python 實現批量文件加密功能

Posted on by

一、源碼

1、結果展示

  python自動化辦公現在可不是一個陌生的詞，也隨著人們對自己隱私越來越看重，就會出現這樣的需求：每人一個PDF文件，因有個人隱私信息，所以需要對文件進行加密且密碼為本人身份證後六位。
  廢話不多說，上結論，程序運行結果如下：

在這裡插入圖片描述

  我們去一探究竟，原文件如下：

在這裡插入圖片描述 

 獲取密碼如下：

在這裡插入圖片描述

  將文件加密以後結果如下：

在這裡插入圖片描述

  我們隨機打開一個驗證一下：

在這裡插入圖片描述

  輸入身份證後六位，作為密碼即可打開。

2、源碼

  可運行源碼如下：

import os
from PyPDF2 import PdfFileWriter, PdfFileReader
import pandas as pd

# 全局工作目錄
os.chdir("D:\加密")

def main():
    # 1、找文件
    # 獲取實習證明原文件的文件名
    rosterfile = os.listdir('.\原文件\\')
    # 獲取信息中表的姓名與密碼
    name,Password = getPassword()

    # 對每個文件進行加密並保存
    for prove in rosterfile:
        for Name in name:
            if Name in prove:
                i = name.index(Name)
                rosterLast = prove
                rosterNow = '.\文件加密版本\\' + prove
                password = Password[i]
                # 2、加密文件並輸出
                encryptionPDF(rosterLast,rosterNow,password)


# 獲取瞭表中姓名信息，並提取身份證後六位作為密碼
def getPassword():
    password = []
    name = []
    # 獲取Excel列，包括姓名、身份證號
    Name = pd.read_excel('.\名單.xlsx', usecols=[0],names=None)  # 讀取項目名稱列,不要列名
    Id = pd.read_excel('.\名單.xlsx', usecols=[1],names=None)  # 讀取項目名稱列,不要列名
    Name1 = Name.values.tolist()
    id = Id.values.tolist()
    # 提取身份證後六位作為密碼
    for i in id:
        password.append(str(i[0])[12:18:1])
    # 提取姓名為字符串，原來為list
    for j in Name1:
        name.append(j[0])
    return name,password

# 加密文件並輸出
# 變量說明：rosterLast-原PDF文件路徑,rosterNow-加密後輸出路徑,password-加密密碼
def encryptionPDF(rosterLast,rosterNow,password):
    pdf_reader = PdfFileReader(r'.\原文件\\%s'%(rosterLast))  # 輸入你想要操作的pdf文檔的位置/名稱
    pdf_writer = PdfFileWriter()

    for page in range(pdf_reader.getNumPages()):
        pdf_writer.addPage(pdf_reader.getPage(page))

    pdf_writer.encrypt(password)  # 寫入密碼並輸出到目的路徑
    with open(rosterNow, 'wb') as out:
        pdf_writer.write(out)
    print("文件:%s----已完成加密,密碼為：%s"%(rosterLast,password))

# 函數入口
if __name__ == '__main__':
    print("-----開始加密-----")
    main()
    print("-----加密完成-----")

二、源碼註釋

1、所使用依賴包

  共使用三個函數：

# 文件操作
import os

# 操作PDF文件
from PyPDF2 import PdfFileWriter, PdfFileReader

# 操作數據
import pandas as pd

  在這裡我隻是簡單說明用這些第三方庫幹瞭什麼，具體學習可參考官方文檔或者菜鳥教程。

2、函數功能

  編碼過程中也共定義有三個函數，其功能如下：

# 主函數，從這裡我們可以看到我整體是如何實現此需求的
def main():

# 功能函數，用來獲取文件密碼的，然後使用姓名與文件名做匹配，達到本人文件為本人身份證後六位的需求。
def getPassword():

#  功能函數，本次編碼的核心，前面都可以作為準備階段，就是為本函數用來加密文件而提供讀取、輸出路徑以及密碼的。
def encryptionPDF(rosterLast,rosterNow,password):

3、代碼拓展

  這隻是一個很簡單的需求，它並不具有普遍性，它可能僅在我電腦上可以運行，當你拷貝走源碼以後可能就沒法運行瞭，會報各種各樣的錯誤。
  如果你想直接用，那可能得按照我在編碼過程中準備的數據一模一樣的去準備。所以它其實隻能用作學習交流，想要直接使用不太現實。
  當然可以嘗試著在此基礎上增加功能或者增加代碼的可用性與門檻。例如：獲取用戶輸入的路徑，定義密碼模板等，讓代碼具有普遍性，可直接拷貝運行那種的。

三、可運行環境

  那如果你不想改源碼，想直接用那怎麼辦？
  沒問題！那你就跟著我來吧，一點一點來。
  首先在D盤創建一個文件夾名為：加密；註意！必須保證它在根目錄下！它的路徑為D:\加密。然後打開文件創建如下：必須一模一樣

在這裡插入圖片描述

  差不多瞭，原文件放在原文件夾裡面，運行以後的文件就在加密版本文件夾裡面。
  最重要的就是在名單的表格裡面把信息填好，這就是用來設置密碼的，表格格式如下：身份證號碼格式為數字

在這裡插入圖片描述

  表格隻有兩列！一列姓名、一列身份證號碼。身份證號碼為數字類型！

到此這篇關於python 實現批量文件加密功能的文章就介紹到這瞭,更多相關python 批量文件加密內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：