Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
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
6 changes: 6 additions & 0 deletions pypdf/_writer.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,6 +169,9 @@ class PdfWriter(PdfDocCommon):
If false, pypdf will try to be forgiving and do something reasonable, but it will log
a warning message. It is a best-effort approach.

keep_initial_header: If true, the PDF header of the cloned document is kept
when using ``clone_from`` in non-incremental mode.

"""

def __init__(
Expand All @@ -179,6 +182,7 @@ def __init__(
full: bool = False,
strict: bool = False,
*,
keep_initial_header: bool = False,
incremental_clone_object_count_limit: Optional[int] = 500_000,
incremental_clone_object_id_limit: Optional[int] = 1_000_000,
) -> None:
Expand Down Expand Up @@ -304,6 +308,8 @@ def _get_clone_from(
clone_from = PdfReader(clone_from)
self.clone_document_from_reader(clone_from)
self._cloned = True
if keep_initial_header and not self.incremental:
self._header = clone_from.pdf_header.encode()
else:
self._pages = self._add_object(pages)
self._root_object = DictionaryObject(
Expand Down
9 changes: 9 additions & 0 deletions tests/test_writer.py
Original file line number Diff line number Diff line change
Expand Up @@ -994,6 +994,15 @@ def test_pdf_header():
assert writer.pdf_header == "%PDF-1.6"


def test_pdf_header_keep_initial_header():
reader = PdfReader(RESOURCE_ROOT / "crazyones.pdf")
writer = PdfWriter(clone_from=reader)
assert writer.pdf_header == "%PDF-1.3"

writer = PdfWriter(clone_from=reader, keep_initial_header=True)
assert writer.pdf_header == reader.pdf_header


def test_write_dict_stream_object(pdf_file_path):
stream = (
b"BT "
Expand Down
Loading