Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Utility function to detect if a PDF is encrypted

Browse Source
This commit is contained in:
Kovid Goyal 2019-08-05 12:11:48 +05:30
parent 6d449818c2
commit ec0515d409
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/calibre/ebooks/metadata/pdf.py
View File

@ -8,7 +8,7 @@ import os, subprocess, shutil, re
from functools import partial from functools import partial
from calibre import prints from calibre import prints
from calibre.constants import iswindows from calibre.constants import iswindows, ispy3
from calibre.ptempfile import TemporaryDirectory from calibre.ptempfile import TemporaryDirectory
from calibre.ebooks.metadata import ( from calibre.ebooks.metadata import (
MetaInformation, string_to_authors, check_isbn, check_doi) MetaInformation, string_to_authors, check_isbn, check_doi)
@ -97,6 +97,17 @@ def page_images(pdfpath, outputdir, first=1, last=1):
raise ValueError('Failed to render PDF, pdftoppm errorcode: %s'%e.returncode) raise ValueError('Failed to render PDF, pdftoppm errorcode: %s'%e.returncode)
def is_pdf_encrypted(path_to_pdf):
if not ispy3 and not isinstance(path_to_pdf, bytes):
path_to_pdf = path_to_pdf.encode('mbcs' if iswindows else 'utf-8')
pdfinfo = get_tools()[0]
raw = subprocess.check_output([pdfinfo, path_to_pdf])
q = re.search(br'^Encrypted:\s*(\S+)', raw, flags=re.MULTILINE)
if q is not None:
return q.group(1) == b'yes'
return False
def get_metadata(stream, cover=True): def get_metadata(stream, cover=True):
with TemporaryDirectory('_pdf_metadata_read') as pdfpath: with TemporaryDirectory('_pdf_metadata_read') as pdfpath:
stream.seek(0) stream.seek(0)