Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

metadata: now also displaying CreationDate, Creator, Keywords & Subject #73

Merged
merged 1 commit into from
Nov 12, 2024
Merged

metadata: now also displaying CreationDate, Creator, Keywords & Subject #73

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
66 changes: 46 additions & 20 deletions pdfly/metadata.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,9 +19,6 @@ class EncryptionData(BaseModel):
class MetaInfo(BaseModel):
encryption: Optional[EncryptionData] = None
pdf_file_version: str
title: Optional[str] = None
producer: Optional[str] = None
author: Optional[str] = None
pages: Optional[int] = None
page_mode: Optional[str] = None
page_layout: Optional[str] = None
Expand All @@ -30,6 +27,15 @@ class MetaInfo(BaseModel):
id2: Optional[bytes] = None
images: List[int] = []

# PDF /Info dictionary
author: Optional[str] = None
creation_date: Optional[datetime] = None
creator: Optional[str] = None
keywords: Optional[str] = None
producer: Optional[str] = None
subject: Optional[str] = None
title: Optional[str] = None

# OS Information
file_permissions: str
file_size: int # in bytes
Expand All @@ -43,12 +49,14 @@ def main(pdf: Path, output: OutputOptions) -> None:
if reader.is_encrypted:
pdf_stat = pdf.stat()
meta = MetaInfo(
encryption=EncryptionData(
v_value=reader._encryption.V,
revision=reader._encryption.R,
)
if reader.is_encrypted and reader._encryption
else None,
encryption=(
EncryptionData(
v_value=reader._encryption.V,
revision=reader._encryption.R,
)
if reader.is_encrypted and reader._encryption
else None
),
pdf_file_version=reader.stream.read(8).decode("utf-8"),
# OS Info
file_permissions=f"{stat.filemode(pdf_stat.st_mode)}",
Expand All @@ -66,12 +74,14 @@ def main(pdf: Path, output: OutputOptions) -> None:
pdf_id = reader.trailer.get("/ID")
meta = MetaInfo(
pages=len(reader.pages),
encryption=EncryptionData(
v_value=reader._encryption.V, # type: ignore
revision=reader._encryption.R, # type: ignore
)
if reader.is_encrypted and reader._encryption
else None,
encryption=(
EncryptionData(
v_value=reader._encryption.V, # type: ignore
revision=reader._encryption.R, # type: ignore
)
if reader.is_encrypted and reader._encryption
else None
),
page_mode=reader.page_mode,
pdf_file_version=pdf_file_version,
page_layout=reader.page_layout,
Expand All @@ -91,9 +101,14 @@ def main(pdf: Path, output: OutputOptions) -> None:
],
)
if info is not None:
meta.title = info.title
meta.producer = info.producer
meta.author = info.author
meta.creation_date = info.creation_date
meta.creator = info.creator
# Pending https://github.com/py-pdf/pypdf/pull/2939 to be able to access .keywords:
meta.keywords = info.get("/Keywords")
meta.producer = info.producer
meta.subject = info.subject
meta.title = info.title

if output == OutputOptions.json:
print(meta.json())
Expand All @@ -107,9 +122,20 @@ def main(pdf: Path, output: OutputOptions) -> None:
)
table.add_column("Value", style="white")

table.add_row("Title", meta.title)
table.add_row("Producer", meta.producer)
table.add_row("Author", meta.author)
if meta.title:
table.add_row("Title", meta.title)
if meta.author:
table.add_row("Author", meta.author)
if meta.creation_date:
table.add_row("CreationDate", str(meta.creation_date))
if meta.creator:
table.add_row("Creator", meta.creator)
if meta.producer:
table.add_row("Producer", meta.producer)
if meta.subject:
table.add_row("Subject", meta.subject)
if meta.keywords:
table.add_row("Keywords", meta.keywords)
table.add_row("Pages", f"{meta.pages:,}" if meta.pages else "unknown")
table.add_row("Encrypted", f"{meta.encryption}")
table.add_row("PDF File Version", meta.pdf_file_version)
Expand Down
Loading