快捷導(dǎo)航

Python使用pymupdf實現(xiàn)PDF加密

更新時間：2023年08月23日 10:40:32 作者：winfredzhang

這篇文章主要介紹了如何使用 Python 和 wxPython 庫創(chuàng)建一個簡單的圖形用戶界面（GUI）應(yīng)用程序,用于對 PDF 文件進(jìn)行加密,感興趣的小伙伴可以了解下

PDF 文件是一種常見的文檔格式，但有時候我們希望對敏感信息進(jìn)行保護(hù)，以防止未經(jīng)授權(quán)的訪問。在本文中，我們將使用 Python 和 wxPython 庫創(chuàng)建一個簡單的圖形用戶界面（GUI）應(yīng)用程序，用于對 PDF 文件進(jìn)行加密。C:\pythoncode\new\PDFEncrypt.py

準(zhǔn)備工作

在開始之前，請確保已經(jīng)安裝了以下庫：

wxPython：在命令行中運行 pip install wxPython 進(jìn)行安裝
PyMuPDF（也稱為 fitz）：在命令行中運行 pip install PyMuPDF 進(jìn)行安裝創(chuàng)建 GUI 應(yīng)用程序

我們將使用 wxPython 庫創(chuàng)建 GUI 應(yīng)用程序。首先，導(dǎo)入必要的庫：

import wx
import fitz

接下來，創(chuàng)建一個主窗口類 MainFrame ，繼承自 wx.Frame 類：

class MainFrame(wx.Frame):
    def __init__(self):
        super().__init__(None, title="PDF Encryption", size=(400, 200))
        panel = wx.Panel(self)
        # 創(chuàng)建文件選擇器
        self.file_picker = wx.FilePickerCtrl(panel, message="Select a PDF file", style=wx.FLP_USE_TEXTCTRL)
        # 創(chuàng)建密碼輸入框
        self.password_text = wx.TextCtrl(panel, style=wx.TE_PASSWORD)
        # 創(chuàng)建加密按鈕
        encrypt_button = wx.Button(panel, label="Encrypt")
        encrypt_button.Bind(wx.EVT_BUTTON, self.on_encrypt_button)
        # 使用布局管理器設(shè)置組件的位置和大小
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(wx.StaticText(panel, label="PDF File:"), 0, wx.ALL, 5)
        sizer.Add(self.file_picker, 0, wx.EXPAND | wx.ALL, 5)
        sizer.Add(wx.StaticText(panel, label="Password:"), 0, wx.ALL, 5)
        sizer.Add(self.password_text, 0, wx.EXPAND | wx.ALL, 5)
        sizer.Add(encrypt_button, 0, wx.ALIGN_CENTER | wx.ALL, 5)
        panel.SetSizerAndFit(sizer)

以上代碼創(chuàng)建了一個帶有文件選擇器、密碼輸入框和加密按鈕的主窗口。

接下來，添加處理加密按鈕點擊事件的方法 on_encrypt_button ：

class MainFrame(wx.Frame):
    # ...
    def on_encrypt_button(self, event):
        filepath = self.file_picker.GetPath()
        password = self.password_text.GetValue()
        if filepath and password:
            try:
                doc = fitz.open(filepath)
                doc.encrypt(password)
                encrypted_filepath = filepath.replace(".pdf", "_encrypted.pdf")
                doc.save(encrypted_filepath)
                doc.close()
                wx.MessageBox("PDF file encrypted successfully!", "Success", wx.OK | wx.ICON_INFORMATION)
            except Exception as e:
                wx.MessageBox(f"An error occurred: {str(e)}", "Error", wx.OK | wx.ICON_ERROR)
        else:
            wx.MessageBox("Please select a PDF file and enter a password.", "Error", wx.OK | wx.ICON_ERROR)

在 on_encrypt_button 方法中，我們獲取用戶選擇的 PDF 文件路徑和輸入的密碼。然后，使用 PyMuPDF 庫打開 PDF 文件，對其進(jìn)行加密并保存加密后的文件。

最后，創(chuàng)建一個應(yīng)用程序類 App ，并運行主循環(huán)：

class App(wx.App):
    def OnInit(self):
        frame = MainFrame()
        frame.Show()
        return True
if __name__ == "__main__":
    app = App()
    app.MainLoop()

以上代碼創(chuàng)建了一個應(yīng)用程序類 App ，并在 if __name__ == "__main__": 代碼塊中運行應(yīng)用程序的主循環(huán)。

運行應(yīng)用程序

保存上述代碼為 pdf_encryption.py 文件，然后在命令行中運行 python pdf_encryption.py 。應(yīng)用程序窗口將打開，您可以選擇一個 PDF 文件并輸入密碼來加密它。

全部代碼

import wx
import fitz
class MainFrame(wx.Frame):
    def __init__(self):
        super().__init__(None, title="PDF Encryption", size=(400, 200))
        panel = wx.Panel(self)
        self.file_picker = wx.FilePickerCtrl(panel, message="Select a PDF file", style=wx.FLP_USE_TEXTCTRL)
        self.password_text = wx.TextCtrl(panel, style=wx.TE_PASSWORD)
        encrypt_button = wx.Button(panel, label="Encrypt")
        encrypt_button.Bind(wx.EVT_BUTTON, self.on_encrypt_button)
        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(wx.StaticText(panel, label="PDF File:"), 0, wx.ALL, 5)
        sizer.Add(self.file_picker, 0, wx.EXPAND | wx.ALL, 5)
        sizer.Add(wx.StaticText(panel, label="Password:"), 0, wx.ALL, 5)
        sizer.Add(self.password_text, 0, wx.EXPAND | wx.ALL, 5)
        sizer.Add(encrypt_button, 0, wx.ALIGN_CENTER | wx.ALL, 5)
        panel.SetSizerAndFit(sizer)
    def on_encrypt_button(self, event):
        filepath = self.file_picker.GetPath()
        password = self.password_text.GetValue()
        if filepath and password:
            try:
                doc = fitz.open(filepath)
                # doc.encrypt(password)
                perm = int(
                    fitz.PDF_PERM_ACCESSIBILITY # always use this
                            | fitz.PDF_PERM_PRINT # permit printing
                            | fitz.PDF_PERM_COPY # permit copying
                            | fitz.PDF_PERM_ANNOTATE # permit annotations
                ) # 可以打印，復(fù)制，添加注釋                
                owner_pass = "owner" # owner password
                user_pass = password # "user" # user password
                encrypt_meth = fitz.PDF_ENCRYPT_AES_256 # strongest algorithm
                encrypted_filepath = filepath.replace(".pdf", "_encrypted.pdf")
                # doc.save(encrypted_filepath)
                doc.save(encrypted_filepath,encryption=encrypt_meth,owner_pw=owner_pass,permissions=perm,user_pw=user_pass) 
                # doc.save(encrypted_filepath)
                doc.close()
                wx.MessageBox("PDF file encrypted successfully!", "Success", wx.OK | wx.ICON_INFORMATION)
            except Exception as e:
                wx.MessageBox(f"An error occurred: {str(e)}", "Error", wx.OK | wx.ICON_ERROR)
        else:
            wx.MessageBox("Please select a PDF file and enter a password.", "Error", wx.OK | wx.ICON_ERROR)
class App(wx.App):
    def OnInit(self):
        frame = MainFrame()
        frame.Show()
        return True
if __name__ == "__main__":
    app = App()
    app.MainLoop()