快捷導(dǎo)航

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

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

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

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

準(zhǔn)備工作

在開(kāi)始之前，請(qǐng)確保已經(jīng)安裝了以下庫(kù)：

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

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

import wx
import fitz

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

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)建了一個(gè)帶有文件選擇器、密碼輸入框和加密按鈕的主窗口。

接下來(lái)，添加處理加密按鈕點(diǎn)擊事件的方法 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 方法中，我們獲取用戶(hù)選擇的 PDF 文件路徑和輸入的密碼。然后，使用 PyMuPDF 庫(kù)打開(kāi) PDF 文件，對(duì)其進(jìn)行加密并保存加密后的文件。

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

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

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

運(yùn)行應(yīng)用程序

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

全部代碼

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()

總結(jié)

本文介紹了如何使用 Python 和 wxPython 庫(kù)創(chuàng)建一個(gè)簡(jiǎn)單的圖形用戶(hù)界面應(yīng)用程序，用于對(duì) PDF 文件進(jìn)行加密。通過(guò)選擇 PDF 文件和輸入密碼，您可以加密 PDF 文件以保護(hù)其內(nèi)容的安全性。

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

您可能感興趣的文章: