Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-06-23 15:30:45 -04:00
Code Issues Packages Projects Releases Wiki Activity

Make the PoDoFo metadata setting code re-useable

Browse Source
This commit is contained in:
Kovid Goyal 2019-07-03 18:18:55 +05:30
parent b5b0891421
commit 0470df53d5
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 38 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
src/calibre/utils/podofo/__init__.py
View File

@ -53,45 +53,50 @@ def set_metadata(stream, mi):
stream.seek(0) stream.seek(0)
def set_metadata_implementation(pdf_doc, title, authors, bkp, tags, xmp_packet):
title = prep(title)
touched = False
if title and title != pdf_doc.title:
pdf_doc.title = title
touched = True
author = prep(authors_to_string(authors))
if author and author != pdf_doc.author:
pdf_doc.author = author
touched = True
bkp = prep(bkp)
if bkp and bkp != pdf_doc.creator:
pdf_doc.creator = bkp
touched = True
try:
tags = prep(', '.join([x.strip() for x in tags if x.strip()]))
if tags != pdf_doc.keywords:
pdf_doc.keywords = tags
touched = True
except Exception:
pass
try:
current_xmp_packet = pdf_doc.get_xmp_metadata()
if current_xmp_packet:
from calibre.ebooks.metadata.xmp import merge_xmp_packet
xmp_packet = merge_xmp_packet(current_xmp_packet, xmp_packet)
pdf_doc.set_xmp_metadata(xmp_packet)
touched = True
except Exception:
pass
return touched
def set_metadata_(tdir, title, authors, bkp, tags, xmp_packet): def set_metadata_(tdir, title, authors, bkp, tags, xmp_packet):
podofo = get_podofo() podofo = get_podofo()
os.chdir(tdir) os.chdir(tdir)
p = podofo.PDFDoc() p = podofo.PDFDoc()
p.open('input.pdf') p.open('input.pdf')
title = prep(title)
touched = False
if title and title != p.title:
p.title = title
touched = True
author = prep(authors_to_string(authors))
if author and author != p.author:
p.author = author
touched = True
bkp = prep(bkp)
if bkp and bkp != p.creator:
p.creator = bkp
touched = True
try:
tags = prep(', '.join([x.strip() for x in tags if x.strip()]))
if tags != p.keywords:
p.keywords = tags
touched = True
except:
pass
try:
current_xmp_packet = p.get_xmp_metadata()
if current_xmp_packet:
from calibre.ebooks.metadata.xmp import merge_xmp_packet
xmp_packet = merge_xmp_packet(current_xmp_packet, xmp_packet)
p.set_xmp_metadata(xmp_packet)
touched = True
except:
pass
touched = set_metadata_implementation(p, title, authors, bkp, tags, xmp_packet)
if touched: if touched:
p.save('output.pdf') p.save('output.pdf')