Kicad/kicad-libraries
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
b2ff6a2b48ef091a0097b38fdf17cbe07bd81551
kicad-libraries/3rdparty/plugins/com_gitlab_dennevi_Board2Pdf/pypdf/_writer.py
tieutuanbao b2ff6a2b48 Initial commit
2025-08-28 03:07:33 +07:00

3381 lines
129 KiB
Python
Raw Blame History

# Copyright (c) 2006, Mathieu Fenniak
# Copyright (c) 2007, Ashish Kulkarni <kulkarni.ashish@gmail.com>
#
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions and the following disclaimer in the documentation
# and/or other materials provided with the distribution.
# * The name of the author may not be used to endorse or promote products
# derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
import decimal
import enum
import hashlib
import re
import struct
import uuid
from io import BytesIO, FileIO, IOBase
from itertools import compress
from pathlib import Path
from types import TracebackType
from typing import (
IO,
Any,
Callable,
Dict,
Iterable,
List,
Optional,
Pattern,
Tuple,
Type,
Union,
cast,
)
from ._cmap import _default_fonts_space_width, build_char_map_from_dict
from ._doc_common import DocumentInformation, PdfDocCommon
from ._encryption import EncryptAlgorithm, Encryption
from ._page import PageObject
from ._page_labels import nums_clear_range, nums_insert, nums_next
from ._reader import PdfReader
from ._utils import (
StrByteType,
StreamType,
_get_max_pdf_version_header,
deprecate,
deprecate_no_replacement,
deprecation_with_replacement,
logger_warning,
)
from .constants import AnnotationDictionaryAttributes as AA
from .constants import CatalogAttributes as CA
from .constants import (
CatalogDictionary,
FileSpecificationDictionaryEntries,
GoToActionArguments,
ImageType,
InteractiveFormDictEntries,
PageLabelStyle,
TypFitArguments,
UserAccessPermissions,
)
from .constants import Core as CO
from .constants import FieldDictionaryAttributes as FA
from .constants import PageAttributes as PG
from .constants import PagesAttributes as PA
from .constants import TrailerKeys as TK
from .errors import PyPdfError
from .generic import (
PAGE_FIT,
ArrayObject,
BooleanObject,
ByteStringObject,
ContentStream,
DecodedStreamObject,
Destination,
DictionaryObject,
Fit,
FloatObject,
IndirectObject,
NameObject,
NullObject,
NumberObject,
PdfObject,
RectangleObject,
StreamObject,
TextStringObject,
TreeObject,
ViewerPreferences,
create_string_object,
hex_to_rgb,
is_null_or_none,
)
from .pagerange import PageRange, PageRangeSpec
from .types import (
AnnotationSubtype,
BorderArrayType,
LayoutType,
OutlineItemType,
OutlineType,
PagemodeType,
)
from .xmp import XmpInformation
ALL_DOCUMENT_PERMISSIONS = UserAccessPermissions.all()
DEFAULT_FONT_HEIGHT_IN_MULTILINE = 12
class ObjectDeletionFlag(enum.IntFlag):
NONE = 0
TEXT = enum.auto()
LINKS = enum.auto()
ATTACHMENTS = enum.auto()
OBJECTS_3D = enum.auto()
ALL_ANNOTATIONS = enum.auto()
XOBJECT_IMAGES = enum.auto()
INLINE_IMAGES = enum.auto()
DRAWING_IMAGES = enum.auto()
IMAGES = XOBJECT_IMAGES | INLINE_IMAGES | DRAWING_IMAGES
def _rolling_checksum(stream: BytesIO, blocksize: int = 65536) -> str:
hash = hashlib.md5()
for block in iter(lambda: stream.read(blocksize), b""):
hash.update(block)
return hash.hexdigest()
class PdfWriter(PdfDocCommon):
"""
Write a PDF file out, given pages produced by another class or through
cloning a PDF file during initialization.
Typically data is added from a :class:`PdfReader<pypdf.PdfReader>`.
Args:
clone_from: identical to fileobj (for compatibility)
incremental: If true, loads the document and set the PdfWriter in incremental mode.
When writing incrementally, the original document is written first and new/modified
content is appended. To be used for signed document/forms to keep signature valid.
full: If true, loads all the objects (always full if incremental = True).
This parameters may allows to load very big PDFs.
"""
def __init__(
self,
fileobj: Union[None, PdfReader, StrByteType, Path] = "",
clone_from: Union[None, PdfReader, StrByteType, Path] = None,
incremental: bool = False,
full: bool = False,
) -> None:
self.incremental = incremental or full
"""
Returns if the PdfWriter object has been started in incremental mode.
"""
self._objects: List[Optional[PdfObject]] = []
"""
The indirect objects in the PDF.
For the incremental case, it will be filled with None
in clone_reader_document_root.
"""
self._original_hash: List[int] = []
"""
List of hashes after import; used to identify changes.
"""
self._idnum_hash: Dict[bytes, Tuple[IndirectObject, List[IndirectObject]]] = {}
"""
Maps hash values of indirect objects to the list of IndirectObjects.
This is used for compression.
"""
self._id_translated: Dict[int, Dict[int, int]] = {}
"""List of already translated IDs.
dict[id(pdf)][(idnum, generation)]
"""
self._ID: Union[ArrayObject, None] = None
self._info_obj: Optional[PdfObject]
if self.incremental:
if isinstance(fileobj, (str, Path)):
with open(fileobj, "rb") as f:
fileobj = BytesIO(f.read(-1))
if isinstance(fileobj, BytesIO):
fileobj = PdfReader(fileobj)
if not isinstance(fileobj, PdfReader):
raise PyPdfError("Invalid type for incremental mode")
self._reader = fileobj # prev content is in _reader.stream
self._header = fileobj.pdf_header.encode()
self._readonly = True # !!!TODO: to be analysed
else:
self._header = b"%PDF-1.3"
self._info_obj = self._add_object(
DictionaryObject(
{NameObject("/Producer"): create_string_object("pypdf")}
)
)
def _get_clone_from(
fileobj: Union[None, PdfReader, str, Path, IO[Any], BytesIO],
clone_from: Union[None, PdfReader, str, Path, IO[Any], BytesIO],
) -> Union[None, PdfReader, str, Path, IO[Any], BytesIO]:
if isinstance(fileobj, (str, Path, IO, BytesIO)) and (
fileobj == "" or clone_from is not None
):
return clone_from
cloning = True
if isinstance(fileobj, (str, Path)) and (
not Path(str(fileobj)).exists()
or Path(str(fileobj)).stat().st_size == 0
):
cloning = False
if isinstance(fileobj, (IOBase, BytesIO)):
t = fileobj.tell()
if fileobj.seek(0, 2) == 0:
cloning = False
fileobj.seek(t, 0)
if cloning:
clone_from = fileobj
return clone_from
clone_from = _get_clone_from(fileobj, clone_from)
# to prevent overwriting
self.temp_fileobj = fileobj
self.fileobj = ""
self._with_as_usage = False
self._cloned = False
# The root of our page tree node.
pages = DictionaryObject()
pages.update(
{
NameObject(PA.TYPE): NameObject("/Pages"),
NameObject(PA.COUNT): NumberObject(0),
NameObject(PA.KIDS): ArrayObject(),
}
)
self.flattened_pages = []
self._encryption: Optional[Encryption] = None
self._encrypt_entry: Optional[DictionaryObject] = None
if clone_from is not None:
if not isinstance(clone_from, PdfReader):
clone_from = PdfReader(clone_from)
self.clone_document_from_reader(clone_from)
self._cloned = True
else:
self._pages = self._add_object(pages)
# root object
self._root_object = DictionaryObject()
self._root_object.update(
{
NameObject(PA.TYPE): NameObject(CO.CATALOG),
NameObject(CO.PAGES): self._pages,
}
)
self._add_object(self._root_object)
if full and not incremental:
self.incremental = False
if isinstance(self._ID, list):
if isinstance(self._ID[0], TextStringObject):
self._ID[0] = ByteStringObject(self._ID[0].get_original_bytes())
if isinstance(self._ID[1], TextStringObject):
self._ID[1] = ByteStringObject(self._ID[1].get_original_bytes())
# for commonality
@property
def is_encrypted(self) -> bool:
"""
Read-only boolean property showing whether this PDF file is encrypted.
Note that this property, if true, will remain true even after the
:meth:`decrypt()<pypdf.PdfReader.decrypt>` method is called.
"""
return False
@property
def root_object(self) -> DictionaryObject:
"""
Provide direct access to PDF Structure.
Note:
Recommended only for read access.
"""
return self._root_object
@property
def _info(self) -> Optional[DictionaryObject]:
"""
Provide access to "/Info". Standardized with PdfReader.
Returns:
/Info Dictionary; None if the entry does not exist
"""
return (
None
if self._info_obj is None
else cast(DictionaryObject, self._info_obj.get_object())
)
@_info.setter
def _info(self, value: Optional[Union[IndirectObject, DictionaryObject]]) -> None:
if value is None:
try:
self._objects[self._info_obj.indirect_reference.idnum - 1] = None # type: ignore
except (KeyError, AttributeError):
pass
self._info_obj = None
else:
if self._info_obj is None:
self._info_obj = self._add_object(DictionaryObject())
obj = cast(DictionaryObject, self._info_obj.get_object())
obj.clear()
obj.update(cast(DictionaryObject, value.get_object()))
@property
def xmp_metadata(self) -> Optional[XmpInformation]:
"""XMP (Extensible Metadata Platform) data."""
return cast(XmpInformation, self.root_object.xmp_metadata)
@xmp_metadata.setter
def xmp_metadata(self, value: Optional[XmpInformation]) -> None:
"""XMP (Extensible Metadata Platform) data."""
if value is None:
if "/Metadata" in self.root_object:
del self.root_object["/Metadata"]
else:
self.root_object[NameObject("/Metadata")] = value
return self.root_object.xmp_metadata # type: ignore
@property
def with_as_usage(self) -> bool:
deprecate_no_replacement("with_as_usage", "6.0")
return self._with_as_usage
@with_as_usage.setter
def with_as_usage(self, value: bool) -> None:
deprecate_no_replacement("with_as_usage", "6.0")
self._with_as_usage = value
def __enter__(self) -> "PdfWriter":
"""Store how writer is initialized by 'with'."""
c: bool = self._cloned
t = self.temp_fileobj
self.__init__() # type: ignore
self._cloned = c
self._with_as_usage = True
self.fileobj = t # type: ignore
return self
def __exit__(
self,
exc_type: Optional[Type[BaseException]],
exc: Optional[BaseException],
traceback: Optional[TracebackType],
) -> None:
"""Write data to the fileobj."""
if self.fileobj and not self._cloned:
self.write(self.fileobj)
def _repr_mimebundle_(
self,
include: Union[None, Iterable[str]] = None,
exclude: Union[None, Iterable[str]] = None,
) -> Dict[str, Any]:
"""
Integration into Jupyter Notebooks.
This method returns a dictionary that maps a mime-type to its
representation.
See https://ipython.readthedocs.io/en/stable/config/integrating.html
"""
pdf_data = BytesIO()
self.write(pdf_data)
data = {
"application/pdf": pdf_data,
}
if include is not None:
# Filter representations based on include list
data = {k: v for k, v in data.items() if k in include}
if exclude is not None:
# Remove representations based on exclude list
data = {k: v for k, v in data.items() if k not in exclude}
return data
@property
def pdf_header(self) -> str:
"""
Read/Write property of the PDF header that is written.
This should be something like ``'%PDF-1.5'``. It is recommended to set
the lowest version that supports all features which are used within the
PDF file.
Note: `pdf_header` returns a string but accepts bytes or str for writing
"""
return self._header.decode()
@pdf_header.setter
def pdf_header(self, new_header: Union[str, bytes]) -> None:
if isinstance(new_header, str):
new_header = new_header.encode()
self._header = new_header
def _add_object(self, obj: PdfObject) -> IndirectObject:
if (
getattr(obj, "indirect_reference", None) is not None
and obj.indirect_reference.pdf == self # type: ignore
):
return obj.indirect_reference # type: ignore
# check for /Contents in Pages (/Contents in annotation are strings)
if isinstance(obj, DictionaryObject) and isinstance(
obj.get(PG.CONTENTS, None), (ArrayObject, DictionaryObject)
):
obj[NameObject(PG.CONTENTS)] = self._add_object(obj[PG.CONTENTS])
self._objects.append(obj)
obj.indirect_reference = IndirectObject(len(self._objects), 0, self)
return obj.indirect_reference
def get_object(
self,
indirect_reference: Union[int, IndirectObject],
) -> PdfObject:
if isinstance(indirect_reference, int):
obj = self._objects[indirect_reference - 1]
elif indirect_reference.pdf != self:
raise ValueError("pdf must be self")
else:
obj = self._objects[indirect_reference.idnum - 1]
assert obj is not None # clarification for mypy
return obj
def _replace_object(
self,
indirect_reference: Union[int, IndirectObject],
obj: PdfObject,
) -> PdfObject:
if isinstance(indirect_reference, IndirectObject):
if indirect_reference.pdf != self:
raise ValueError("PDF must be self")
indirect_reference = indirect_reference.idnum
gen = self._objects[indirect_reference - 1].indirect_reference.generation # type: ignore
if (
getattr(obj, "indirect_reference", None) is not None
and obj.indirect_reference.pdf != self # type: ignore
):
obj = obj.clone(self)
self._objects[indirect_reference - 1] = obj
obj.indirect_reference = IndirectObject(indirect_reference, gen, self)
assert isinstance(obj, PdfObject) # clarification for mypy
return obj
def _add_page(
self,
page: PageObject,
index: int,
excluded_keys: Iterable[str] = (),
) -> PageObject:
if not isinstance(page, PageObject) or page.get(PA.TYPE, None) != CO.PAGE:
raise ValueError("Invalid page object")
assert self.flattened_pages is not None, "for mypy"
page_org = page
excluded_keys = list(excluded_keys)
excluded_keys += [PA.PARENT, "/StructParents"]
# acrobat does not accept to have two indirect ref pointing on the same
# page; therefore in order to add easily multiple copies of the same
# page, we need to create a new dictionary for the page, however the
# objects below (including content) are not duplicated:
try: # delete an already existing page
del self._id_translated[id(page_org.indirect_reference.pdf)][ # type: ignore
page_org.indirect_reference.idnum # type: ignore
]
except Exception:
pass
page = cast(
"PageObject", page_org.clone(self, False, excluded_keys).get_object()
)
if page_org.pdf is not None:
other = page_org.pdf.pdf_header
self.pdf_header = _get_max_pdf_version_header(self.pdf_header, other)
node, idx = self._get_page_in_node(index)
page[NameObject(PA.PARENT)] = node.indirect_reference
if idx >= 0:
cast(ArrayObject, node[PA.KIDS]).insert(idx, page.indirect_reference)
self.flattened_pages.insert(index, page)
else:
cast(ArrayObject, node[PA.KIDS]).append(page.indirect_reference)
self.flattened_pages.append(page)
cpt = 1000
while not is_null_or_none(node):
node = cast(DictionaryObject, node.get_object())
node[NameObject(PA.COUNT)] = NumberObject(cast(int, node[PA.COUNT]) + 1)
node = node.get(PA.PARENT, None)
cpt -= 1
if cpt < 0:
raise PyPdfError("Too many recursive calls!")
return page
def set_need_appearances_writer(self, state: bool = True) -> None:
"""
Sets the "NeedAppearances" flag in the PDF writer.
The "NeedAppearances" flag indicates whether the appearance dictionary
for form fields should be automatically generated by the PDF viewer or
if the embedded appearance should be used.
Args:
state: The actual value of the NeedAppearances flag.
Returns:
None
"""
# See §12.7.2 and §7.7.2 for more information:
# https://opensource.adobe.com/dc-acrobat-sdk-docs/pdfstandards/PDF32000_2008.pdf
try:
# get the AcroForm tree
if CatalogDictionary.ACRO_FORM not in self._root_object:
self._root_object[
NameObject(CatalogDictionary.ACRO_FORM)
] = self._add_object(DictionaryObject())
need_appearances = NameObject(InteractiveFormDictEntries.NeedAppearances)
cast(DictionaryObject, self._root_object[CatalogDictionary.ACRO_FORM])[
need_appearances
] = BooleanObject(state)
except Exception as exc: # pragma: no cover
logger_warning(
f"set_need_appearances_writer({state}) catch : {exc}", __name__
)
def create_viewer_preferences(self) -> ViewerPreferences:
o = ViewerPreferences()
self._root_object[
NameObject(CatalogDictionary.VIEWER_PREFERENCES)
] = self._add_object(o)
return o
def add_page(
self,
page: PageObject,
excluded_keys: Iterable[str] = (),
) -> PageObject:
"""
Add a page to this PDF file.
Recommended for advanced usage including the adequate excluded_keys.
The page is usually acquired from a :class:`PdfReader<pypdf.PdfReader>`
instance.
Args:
page: The page to add to the document. Should be
an instance of :class:`PageObject<pypdf._page.PageObject>`
excluded_keys:
Returns:
The added PageObject.
"""
assert self.flattened_pages is not None, "mypy"
return self._add_page(page, len(self.flattened_pages), excluded_keys)
def insert_page(
self,
page: PageObject,
index: int = 0,
excluded_keys: Iterable[str] = (),
) -> PageObject:
"""
Insert a page in this PDF file. The page is usually acquired from a
:class:`PdfReader<pypdf.PdfReader>` instance.
Args:
page: The page to add to the document.
index: Position at which the page will be inserted.
excluded_keys:
Returns:
The added PageObject.
"""
assert self.flattened_pages is not None, "mypy"
if index < 0:
index = len(self.flattened_pages) + index
if index < 0:
raise ValueError("Invalid index value")
if index >= len(self.flattened_pages):
return self.add_page(page, excluded_keys)
else:
return self._add_page(page, index, excluded_keys)
def _get_page_number_by_indirect(
self, indirect_reference: Union[None, int, NullObject, IndirectObject]
) -> Optional[int]:
"""
Generate _page_id2num.
Args:
indirect_reference:
Returns:
The page number or None
"""
# to provide same function as in PdfReader
if is_null_or_none(indirect_reference):
return None
assert indirect_reference is not None, "mypy"
if isinstance(indirect_reference, int):
indirect_reference = IndirectObject(indirect_reference, 0, self)
obj = indirect_reference.get_object()
if isinstance(obj, PageObject):
return obj.page_number
return None
def add_blank_page(
self, width: Optional[float] = None, height: Optional[float] = None
) -> PageObject:
"""
Append a blank page to this PDF file and return it.
If no page size is specified, use the size of the last page.
Args:
width: The width of the new page expressed in default user
space units.
height: The height of the new page expressed in default
user space units.
Returns:
The newly appended page.
Raises:
PageSizeNotDefinedError: if width and height are not defined
and previous page does not exist.
"""
page = PageObject.create_blank_page(self, width, height)
return self.add_page(page)
def insert_blank_page(
self,
width: Optional[Union[float, decimal.Decimal]] = None,
height: Optional[Union[float, decimal.Decimal]] = None,
index: int = 0,
) -> PageObject:
"""
Insert a blank page to this PDF file and return it.
If no page size is specified, use the size of the last page.
Args:
width: The width of the new page expressed in default user
space units.
height: The height of the new page expressed in default
user space units.
index: Position to add the page.
Returns:
The newly inserted page.
Raises:
PageSizeNotDefinedError: if width and height are not defined
and previous page does not exist.
"""
if width is None or height is None and index < self.get_num_pages():
oldpage = self.pages[index]
width = oldpage.mediabox.width
height = oldpage.mediabox.height
page = PageObject.create_blank_page(self, width, height)
self.insert_page(page, index)
return page
@property
def open_destination(
self,
) -> Union[None, Destination, TextStringObject, ByteStringObject]:
return super().open_destination
@open_destination.setter
def open_destination(self, dest: Union[None, str, Destination, PageObject]) -> None:
if dest is None:
try:
del self._root_object["/OpenAction"]
except KeyError:
pass
elif isinstance(dest, str):
self._root_object[NameObject("/OpenAction")] = TextStringObject(dest)
elif isinstance(dest, Destination):
self._root_object[NameObject("/OpenAction")] = dest.dest_array
elif isinstance(dest, PageObject):
self._root_object[NameObject("/OpenAction")] = Destination(
"Opening",
dest.indirect_reference
if dest.indirect_reference is not None
else NullObject(),
PAGE_FIT,
).dest_array
def add_js(self, javascript: str) -> None:
"""
Add JavaScript which will launch upon opening this PDF.
Args:
javascript: Your JavaScript.
>>> output.add_js("this.print({bUI:true,bSilent:false,bShrinkToFit:true});")
# Example: This will launch the print window when the PDF is opened.
"""
# Names / JavaScript preferred to be able to add multiple scripts
if "/Names" not in self._root_object:
self._root_object[NameObject(CA.NAMES)] = DictionaryObject()
names = cast(DictionaryObject, self._root_object[CA.NAMES])
if "/JavaScript" not in names:
names[NameObject("/JavaScript")] = DictionaryObject(
{NameObject("/Names"): ArrayObject()}
)
js_list = cast(
ArrayObject, cast(DictionaryObject, names["/JavaScript"])["/Names"]
)
js = DictionaryObject()
js.update(
{
NameObject(PA.TYPE): NameObject("/Action"),
NameObject("/S"): NameObject("/JavaScript"),
NameObject("/JS"): TextStringObject(f"{javascript}"),
}
)
# We need a name for parameterized JavaScript in the PDF file,
# but it can be anything.
js_list.append(create_string_object(str(uuid.uuid4())))
js_list.append(self._add_object(js))
def add_attachment(self, filename: str, data: Union[str, bytes]) -> None:
"""
Embed a file inside the PDF.
Reference:
https://opensource.adobe.com/dc-acrobat-sdk-docs/pdfstandards/PDF32000_2008.pdf
Section 7.11.3
Args:
filename: The filename to display.
data: The data in the file.
"""
# We need three entries:
# * The file's data
# * The /Filespec entry
# * The file's name, which goes in the Catalog
# The entry for the file
# Sample:
# 8 0 obj
# <<
# /Length 12
# /Type /EmbeddedFile
# >>
# stream
# Hello world!
# endstream
# endobj
if isinstance(data, str):
data = data.encode("latin-1")
file_entry = DecodedStreamObject()
file_entry.set_data(data)
file_entry.update({NameObject(PA.TYPE): NameObject("/EmbeddedFile")})
# The Filespec entry
# Sample:
# 7 0 obj
# <<
# /Type /Filespec
# /F (hello.txt)
# /EF << /F 8 0 R >>
# >>
# endobj
ef_entry = DictionaryObject()
ef_entry.update({NameObject("/F"): self._add_object(file_entry)})
filespec = DictionaryObject()
filespec.update(
{
NameObject(PA.TYPE): NameObject("/Filespec"),
NameObject(FileSpecificationDictionaryEntries.F): create_string_object(
filename
), # Perhaps also try TextStringObject
NameObject(FileSpecificationDictionaryEntries.EF): ef_entry,
}
)
# Then create the entry for the root, as it needs
# a reference to the Filespec
# Sample:
# 1 0 obj
# <<
# /Type /Catalog
# /Outlines 2 0 R
# /Pages 3 0 R
# /Names << /EmbeddedFiles << /Names [(hello.txt) 7 0 R] >> >>
# >>
# endobj
if CA.NAMES not in self._root_object:
self._root_object[NameObject(CA.NAMES)] = self._add_object(
DictionaryObject()
)
if "/EmbeddedFiles" not in cast(DictionaryObject, self._root_object[CA.NAMES]):
embedded_files_names_dictionary = DictionaryObject(
{NameObject(CA.NAMES): ArrayObject()}
)
cast(DictionaryObject, self._root_object[CA.NAMES])[
NameObject("/EmbeddedFiles")
] = self._add_object(embedded_files_names_dictionary)
else:
embedded_files_names_dictionary = cast(
DictionaryObject,
cast(DictionaryObject, self._root_object[CA.NAMES])["/EmbeddedFiles"],
)
cast(ArrayObject, embedded_files_names_dictionary[CA.NAMES]).extend(
[create_string_object(filename), filespec]
)
def append_pages_from_reader(
self,
reader: PdfReader,
after_page_append: Optional[Callable[[PageObject], None]] = None,
) -> None:
"""
Copy pages from reader to writer. Includes an optional callback
parameter which is invoked after pages are appended to the writer.
``append`` should be preferred.
Args:
reader: a PdfReader object from which to copy page
annotations to this writer object. The writer's annots
will then be updated.
after_page_append:
Callback function that is invoked after each page is appended to
the writer. Signature includes a reference to the appended page
(delegates to append_pages_from_reader). The single parameter of
the callback is a reference to the page just appended to the
document.
"""
# Get page count from writer and reader
reader_num_pages = len(reader.pages)
# Copy pages from reader to writer
for reader_page_number in range(reader_num_pages):
reader_page = reader.pages[reader_page_number]
writer_page = self.add_page(reader_page)
# Trigger callback, pass writer page as parameter
if callable(after_page_append):
after_page_append(writer_page)
def _update_field_annotation(
self,
field: DictionaryObject,
anno: DictionaryObject,
font_name: str = "",
font_size: float = -1,
) -> None:
# Calculate rectangle dimensions
_rct = cast(RectangleObject, anno[AA.Rect])
rct = RectangleObject((0, 0, abs(_rct[2] - _rct[0]), abs(_rct[3] - _rct[1])))
# Extract font information
da = anno.get_inherited(
AA.DA,
cast(DictionaryObject, self.root_object[CatalogDictionary.ACRO_FORM]).get(
AA.DA, None
),
)
if da is None:
da = TextStringObject("/Helv 0 Tf 0 g")
else:
da = da.get_object()
font_properties = da.replace("\n", " ").replace("\r", " ").split(" ")
font_properties = [x for x in font_properties if x != ""]
if font_name:
font_properties[font_properties.index("Tf") - 2] = font_name
else:
font_name = font_properties[font_properties.index("Tf") - 2]
font_height = (
font_size
if font_size >= 0
else float(font_properties[font_properties.index("Tf") - 1])
)
if font_height == 0:
if field.get(FA.Ff, 0) & FA.FfBits.Multiline:
font_height = DEFAULT_FONT_HEIGHT_IN_MULTILINE
else:
font_height = rct.height - 2
font_properties[font_properties.index("Tf") - 1] = str(font_height)
da = " ".join(font_properties)
y_offset = rct.height - 1 - font_height
# Retrieve font information from local DR ...
dr: Any = cast(
DictionaryObject,
cast(
DictionaryObject,
anno.get_inherited(
"/DR",
cast(
DictionaryObject, self.root_object[CatalogDictionary.ACRO_FORM]
).get("/DR", DictionaryObject()),
),
).get_object(),
)
dr = dr.get("/Font", DictionaryObject()).get_object()
# _default_fonts_space_width keys is the list of Standard fonts
if font_name not in dr and font_name not in _default_fonts_space_width:
# ...or AcroForm dictionary
dr = cast(
Dict[Any, Any],
cast(
DictionaryObject, self.root_object[CatalogDictionary.ACRO_FORM]
).get("/DR", {}),
)
dr = dr.get_object().get("/Font", DictionaryObject()).get_object()
font_res = dr.get(font_name, None)
if not is_null_or_none(font_res):
font_res = cast(DictionaryObject, font_res.get_object())
font_subtype, _, font_encoding, font_map = build_char_map_from_dict(
200, font_res
)
try: # get rid of width stored in -1 key
del font_map[-1]
except KeyError:
pass
font_full_rev: Dict[str, bytes]
if isinstance(font_encoding, str):
font_full_rev = {
v: k.encode(font_encoding) for k, v in font_map.items()
}
else:
font_full_rev = {v: bytes((k,)) for k, v in font_encoding.items()}
font_encoding_rev = {v: bytes((k,)) for k, v in font_encoding.items()}
for kk, v in font_map.items():
font_full_rev[v] = font_encoding_rev.get(kk, kk)
else:
logger_warning(f"Font dictionary for {font_name} not found.", __name__)
font_full_rev = {}
# Retrieve field text and selected values
field_flags = field.get(FA.Ff, 0)
if field.get(FA.FT, "/Tx") == "/Ch" and field_flags & FA.FfBits.Combo == 0:
txt = "\n".join(anno.get_inherited(FA.Opt, []))
sel = field.get("/V", [])
if not isinstance(sel, list):
sel = [sel]
else: # /Tx
txt = field.get("/V", "")
sel = []
# Escape parentheses (pdf 1.7 reference, table 3.2 Literal Strings)
txt = txt.replace("\\", "\\\\").replace("(", r"\(").replace(")", r"\)")
# Generate appearance stream
ap_stream = generate_appearance_stream(
txt, sel, da, font_full_rev, rct, font_height, y_offset
)
# Create appearance dictionary
dct = DecodedStreamObject.initialize_from_dictionary(
{
NameObject("/Type"): NameObject("/XObject"),
NameObject("/Subtype"): NameObject("/Form"),
NameObject("/BBox"): rct,
"__streamdata__": ByteStringObject(ap_stream),
"/Length": 0,
}
)
if AA.AP in anno:
for k, v in cast(DictionaryObject, anno[AA.AP]).get("/N", {}).items():
if k not in {"/BBox", "/Length", "/Subtype", "/Type", "/Filter"}:
dct[k] = v
# Update Resources with font information if necessary
if font_res is not None:
dct[NameObject("/Resources")] = DictionaryObject(
{
NameObject("/Font"): DictionaryObject(
{
NameObject(font_name): getattr(
font_res, "indirect_reference", font_res
)
}
)
}
)
if AA.AP not in anno:
anno[NameObject(AA.AP)] = DictionaryObject(
{NameObject("/N"): self._add_object(dct)}
)
elif "/N" not in cast(DictionaryObject, anno[AA.AP]):
cast(DictionaryObject, anno[NameObject(AA.AP)])[
NameObject("/N")
] = self._add_object(dct)
else: # [/AP][/N] exists
n = anno[AA.AP]["/N"].indirect_reference.idnum # type: ignore
self._objects[n - 1] = dct
dct.indirect_reference = IndirectObject(n, 0, self)
FFBITS_NUL = FA.FfBits(0)
def update_page_form_field_values(
self,
page: Union[PageObject, List[PageObject], None],
fields: Dict[str, Any],
flags: FA.FfBits = FFBITS_NUL,
auto_regenerate: Optional[bool] = True,
) -> None:
"""
Update the form field values for a given page from a fields dictionary.
Copy field texts and values from fields to page.
If the field links to a parent object, add the information to the parent.
Args:
page: `PageObject` - references **PDF writer's page** where the
annotations and field data will be updated.
`List[Pageobject]` - provides list of pages to be processed.
`None` - all pages.
fields: a Python dictionary of:
* field names (/T) as keys and text values (/V) as value
* field names (/T) as keys and list of text values (/V) for multiple choice list
* field names (/T) as keys and tuple of:
* text values (/V)
* font id (e.g. /F1, the font id must exist)
* font size (0 for autosize)
flags: A set of flags from :class:`~pypdf.constants.FieldDictionaryAttributes.FfBits`.
auto_regenerate: Set/unset the need_appearances flag;
the flag is unchanged if auto_regenerate is None.
"""
if CatalogDictionary.ACRO_FORM not in self._root_object:
raise PyPdfError("No /AcroForm dictionary in PdfWriter Object")
af = cast(DictionaryObject, self._root_object[CatalogDictionary.ACRO_FORM])
if InteractiveFormDictEntries.Fields not in af:
raise PyPdfError("No /Fields dictionary in Pdf in PdfWriter Object")
if isinstance(auto_regenerate, bool):
self.set_need_appearances_writer(auto_regenerate)
# Iterate through pages, update field values
if page is None:
page = list(self.pages)
if isinstance(page, list):
for p in page:
if PG.ANNOTS in p: # just to prevent warnings
self.update_page_form_field_values(p, fields, flags, None)
return None
if PG.ANNOTS not in page:
logger_warning("No fields to update on this page", __name__)
return
for writer_annot in page[PG.ANNOTS]: # type: ignore
writer_annot = cast(DictionaryObject, writer_annot.get_object())
if writer_annot.get("/Subtype", "") != "/Widget":
continue
if "/FT" in writer_annot and "/T" in writer_annot:
writer_parent_annot = writer_annot
else:
writer_parent_annot = writer_annot.get(
PG.PARENT, DictionaryObject()
).get_object()
for field, value in fields.items():
if not (
self._get_qualified_field_name(writer_parent_annot) == field
or writer_parent_annot.get("/T", None) == field
):
continue
if (
writer_parent_annot.get("/FT", None) == "/Ch"
and "/I" in writer_parent_annot
):
del writer_parent_annot["/I"]
if flags:
writer_annot[NameObject(FA.Ff)] = NumberObject(flags)
if isinstance(value, list):
lst = ArrayObject(TextStringObject(v) for v in value)
writer_parent_annot[NameObject(FA.V)] = lst
elif isinstance(value, tuple):
writer_annot[NameObject(FA.V)] = TextStringObject(
value[0],
)
else:
writer_parent_annot[NameObject(FA.V)] = TextStringObject(value)
if writer_parent_annot.get(FA.FT) in ("/Btn"):
# case of Checkbox button (no /FT found in Radio widgets
v = NameObject(value)
if v not in writer_annot[NameObject(AA.AP)][NameObject("/N")]:
v = NameObject("/Off")
# other cases will be updated through the for loop
writer_annot[NameObject(AA.AS)] = v
elif (
writer_parent_annot.get(FA.FT) == "/Tx"
or writer_parent_annot.get(FA.FT) == "/Ch"
):
# textbox
if isinstance(value, tuple):
self._update_field_annotation(
writer_parent_annot, writer_annot, value[1], value[2]
)
else:
self._update_field_annotation(writer_parent_annot, writer_annot)
elif (
writer_annot.get(FA.FT) == "/Sig"
): # deprecated # not implemented yet
# signature
logger_warning("Signature forms not implemented yet", __name__)
def reattach_fields(
self, page: Optional[PageObject] = None
) -> List[DictionaryObject]:
"""
Parse annotations within the page looking for orphan fields and
reattach then into the Fields Structure.
Args:
page: page to analyze.
If none is provided, all pages will be analyzed.
Returns:
list of reattached fields.
"""
lst = []
if page is None:
for p in self.pages:
lst += self.reattach_fields(p)
return lst
try:
af = cast(DictionaryObject, self._root_object[CatalogDictionary.ACRO_FORM])
except KeyError:
af = DictionaryObject()
self._root_object[NameObject(CatalogDictionary.ACRO_FORM)] = af
try:
fields = cast(ArrayObject, af[InteractiveFormDictEntries.Fields])
except KeyError:
fields = ArrayObject()
af[NameObject(InteractiveFormDictEntries.Fields)] = fields
if "/Annots" not in page:
return lst
annots = cast(ArrayObject, page["/Annots"])
for idx in range(len(annots)):
ano = annots[idx]
indirect = isinstance(ano, IndirectObject)
ano = cast(DictionaryObject, ano.get_object())
if ano.get("/Subtype", "") == "/Widget" and "/FT" in ano:
if (
"indirect_reference" in ano.__dict__
and ano.indirect_reference in fields
):
continue
if not indirect:
annots[idx] = self._add_object(ano)
fields.append(ano.indirect_reference)
lst.append(ano)
return lst
def clone_reader_document_root(self, reader: PdfReader) -> None:
"""
Copy the reader document root to the writer and all sub-elements,
including pages, threads, outlines,... For partial insertion, ``append``
should be considered.
Args:
reader: PdfReader from which the document root should be copied.
"""
self._info_obj = None
if self.incremental:
self._objects = [None] * cast(int, reader.trailer["/Size"])
for i in range(len(self._objects) - 1):
o = reader.get_object(i + 1)
if o is not None:
self._objects[i] = o.replicate(self)
else:
self._objects.clear()
self._root_object = reader.root_object.clone(self)
self._pages = self._root_object.raw_get("/Pages")
assert len(self._objects) <= cast(int, reader.trailer["/Size"]) # for pytest
# must be done here before rewriting
if self.incremental:
self._original_hash = [
(obj.hash_bin() if obj is not None else 0) for obj in self._objects
]
self._flatten()
assert self.flattened_pages is not None
for p in self.flattened_pages:
self._replace_object(cast(IndirectObject, p.indirect_reference).idnum, p)
if not self.incremental:
p[NameObject("/Parent")] = self._pages
if not self.incremental:
cast(DictionaryObject, self._pages.get_object())[
NameObject("/Kids")
] = ArrayObject([p.indirect_reference for p in self.flattened_pages])
def clone_document_from_reader(
self,
reader: PdfReader,
after_page_append: Optional[Callable[[PageObject], None]] = None,
) -> None:
"""
Create a copy (clone) of a document from a PDF file reader cloning
section '/Root' and '/Info' and '/ID' of the pdf.
Args:
reader: PDF file reader instance from which the clone
should be created.
after_page_append:
Callback function that is invoked after each page is appended to
the writer. Signature includes a reference to the appended page
(delegates to append_pages_from_reader). The single parameter of
the callback is a reference to the page just appended to the
document.
"""
self.clone_reader_document_root(reader)
inf = reader._info
if self.incremental:
if inf is not None:
self._info_obj = cast(
IndirectObject, inf.clone(self).indirect_reference
)
assert isinstance(self._info, DictionaryObject), "for mypy"
self._original_hash[
self._info_obj.indirect_reference.idnum - 1
] = self._info.hash_bin()
elif inf is not None:
self._info_obj = self._add_object(
DictionaryObject(cast(DictionaryObject, inf.get_object()))
)
# else: _info_obj = None done in clone_reader_document_root()
try:
self._ID = cast(ArrayObject, reader._ID).clone(self)
except AttributeError:
pass
if callable(after_page_append):
for page in cast(
ArrayObject, cast(DictionaryObject, self._pages.get_object())["/Kids"]
):
after_page_append(page.get_object())
def _compute_document_identifier(self) -> ByteStringObject:
stream = BytesIO()
self._write_pdf_structure(stream)
stream.seek(0)
return ByteStringObject(_rolling_checksum(stream).encode("utf8"))
def generate_file_identifiers(self) -> None:
"""
Generate an identifier for the PDF that will be written.
The only point of this is ensuring uniqueness. Reproducibility is not
required.
When a file is first written, both identifiers shall be set to the same value.
If both identifiers match when a file reference is resolved, it is very
likely that the correct and unchanged file has been found. If only the first
identifier matches, a different version of the correct file has been found.
see 14.4 "File Identifiers".
"""
if self._ID:
id1 = self._ID[0]
id2 = self._compute_document_identifier()
else:
id1 = self._compute_document_identifier()
id2 = id1
self._ID = ArrayObject((id1, id2))
def encrypt(
self,
user_password: str,
owner_password: Optional[str] = None,
use_128bit: bool = True,
permissions_flag: UserAccessPermissions = ALL_DOCUMENT_PERMISSIONS,
*,
algorithm: Optional[str] = None,
) -> None:
"""
Encrypt this PDF file with the PDF Standard encryption handler.
Args:
user_password: The password which allows for opening
and reading the PDF file with the restrictions provided.
owner_password: The password which allows for
opening the PDF files without any restrictions. By default,
the owner password is the same as the user password.
use_128bit: flag as to whether to use 128bit
encryption. When false, 40bit encryption will be used.
By default, this flag is on.
permissions_flag: permissions as described in
Table 3.20 of the PDF 1.7 specification. A bit value of 1 means
the permission is granted.
Hence an integer value of -1 will set all flags.
Bit position 3 is for printing, 4 is for modifying content,
5 and 6 control annotations, 9 for form fields,
10 for extraction of text and graphics.
algorithm: encrypt algorithm. Values may be one of "RC4-40", "RC4-128",
"AES-128", "AES-256-R5", "AES-256". If it is valid,
`use_128bit` will be ignored.
"""
if owner_password is None:
owner_password = user_password
if algorithm is not None:
try:
alg = getattr(EncryptAlgorithm, algorithm.replace("-", "_"))
except AttributeError:
raise ValueError(f"Algorithm '{algorithm}' NOT supported")
else:
alg = EncryptAlgorithm.RC4_128
if not use_128bit:
alg = EncryptAlgorithm.RC4_40
self.generate_file_identifiers()
assert self._ID
self._encryption = Encryption.make(alg, permissions_flag, self._ID[0])
# in case call `encrypt` again
entry = self._encryption.write_entry(user_password, owner_password)
if self._encrypt_entry:
# replace old encrypt_entry
assert self._encrypt_entry.indirect_reference is not None
entry.indirect_reference = self._encrypt_entry.indirect_reference
self._objects[entry.indirect_reference.idnum - 1] = entry
else:
self._add_object(entry)
self._encrypt_entry = entry
def write_stream(self, stream: StreamType) -> None:
if hasattr(stream, "mode") and "b" not in stream.mode:
logger_warning(
f"File <{stream.name}> to write to is not in binary mode. "
"It may not be written to correctly.",
__name__,
)
# deprecated to be removed in pypdf 6.0.0 :
# if not self._root:
# self._root = self._add_object(self._root_object)
# self._sweep_indirect_references(self._root)
if self.incremental:
self._reader.stream.seek(0)
stream.write(self._reader.stream.read(-1))
if len(self.list_objects_in_increment()) > 0:
self._write_increment(stream) # writes objs, Xref stream and startx
else:
object_positions, free_objects = self._write_pdf_structure(stream)
xref_location = self._write_xref_table(
stream, object_positions, free_objects
)
self._write_trailer(stream, xref_location)
def write(self, stream: Union[Path, StrByteType]) -> Tuple[bool, IO[Any]]:
"""
Write the collection of pages added to this object out as a PDF file.
Args:
stream: An object to write the file to. The object can support
the write method and the tell method, similar to a file object, or
be a file path, just like the fileobj, just named it stream to keep
existing workflow.
Returns:
A tuple (bool, IO).
"""
my_file = False
if stream == "":
raise ValueError(f"Output({stream=}) is empty.")
if isinstance(stream, (str, Path)):
stream = FileIO(stream, "wb")
my_file = True
self.write_stream(stream)
if my_file:
stream.close()
else:
stream.flush()
return my_file, stream
def list_objects_in_increment(self) -> List[IndirectObject]:
"""
For debugging/analysis.
Provides the list of new/modified objects that will be written
in the increment.
Deleted objects will not be freed but will become orphans.
Returns:
List of (new / modified) IndirectObjects
"""
return [
cast(IndirectObject, self._objects[i]).indirect_reference
for i in range(len(self._objects))
if (
self._objects[i] is not None
and (
i >= len(self._original_hash)
or cast(PdfObject, self._objects[i]).hash_bin()
!= self._original_hash[i]
)
)
]
def _write_increment(self, stream: StreamType) -> None:
object_positions = {}
object_blocks = []
current_start = -1
current_stop = -2
for i, obj in enumerate(self._objects):
if self._objects[i] is not None and (
i >= len(self._original_hash)
or cast(PdfObject, self._objects[i]).hash_bin()
!= self._original_hash[i]
):
idnum = i + 1
assert isinstance(obj, PdfObject) # mypy
# first write new/modified object
object_positions[idnum] = stream.tell()
stream.write(f"{idnum} 0 obj\n".encode())
""" encryption is not operational
if self._encryption and obj != self._encrypt_entry:
obj = self._encryption.encrypt_object(obj, idnum, 0)
"""
obj.write_to_stream(stream)
stream.write(b"\nendobj\n")
# prepare xref
if idnum != current_stop:
if current_start > 0:
object_blocks.append(
[current_start, current_stop - current_start]
)
current_start = idnum
current_stop = idnum + 1
assert current_start > 0, "for pytest only"
object_blocks.append([current_start, current_stop - current_start])
# write incremented xref
xref_location = stream.tell()
xr_id = len(self._objects) + 1
stream.write(f"{xr_id} 0 obj".encode())
init_data = {
NameObject("/Type"): NameObject("/XRef"),
NameObject("/Size"): NumberObject(xr_id + 1),
NameObject("/Root"): self.root_object.indirect_reference,
NameObject("/Filter"): NameObject("/FlateDecode"),
NameObject("/Index"): ArrayObject(
[NumberObject(_it) for _su in object_blocks for _it in _su]
),
NameObject("/W"): ArrayObject(
[NumberObject(1), NumberObject(4), NumberObject(1)]
),
"__streamdata__": b"",
}
if self._info is not None and (
self._info.indirect_reference.idnum - 1 # type: ignore
>= len(self._original_hash)
or cast(IndirectObject, self._info).hash_bin() # kept for future
!= self._original_hash[
self._info.indirect_reference.idnum - 1 # type: ignore
]
):
init_data[NameObject(TK.INFO)] = self._info.indirect_reference
init_data[NameObject(TK.PREV)] = NumberObject(self._reader._startxref)
if self._ID:
init_data[NameObject(TK.ID)] = self._ID
xr = StreamObject.initialize_from_dictionary(init_data)
xr.set_data(
b"".join(
[struct.pack(b">BIB", 1, _pos, 0) for _pos in object_positions.values()]
)
)
xr.write_to_stream(stream)
stream.write(f"\nstartxref\n{xref_location}\n%%EOF\n".encode()) # eof
def _write_pdf_structure(self, stream: StreamType) -> Tuple[List[int], List[int]]:
object_positions = []
free_objects = [] # will contain list of all free entries
stream.write(self.pdf_header.encode() + b"\n")
stream.write(b"%\xE2\xE3\xCF\xD3\n")
for i, obj in enumerate(self._objects):
if obj is not None:
idnum = i + 1
object_positions.append(stream.tell())
stream.write(f"{idnum} 0 obj\n".encode())
if self._encryption and obj != self._encrypt_entry:
obj = self._encryption.encrypt_object(obj, idnum, 0)
obj.write_to_stream(stream)
stream.write(b"\nendobj\n")
else:
object_positions.append(-1)
free_objects.append(i + 1)
free_objects.append(0) # add 0 to loop in accordance with PDF spec
return object_positions, free_objects
def _write_xref_table(
self, stream: StreamType, object_positions: List[int], free_objects: List[int]
) -> int:
xref_location = stream.tell()
stream.write(b"xref\n")
stream.write(f"0 {len(self._objects) + 1}\n".encode())
stream.write(f"{free_objects[0]:0>10} {65535:0>5} f \n".encode())
free_idx = 1
for offset in object_positions:
if offset > 0:
stream.write(f"{offset:0>10} {0:0>5} n \n".encode())
else:
stream.write(f"{free_objects[free_idx]:0>10} {1:0>5} f \n".encode())
free_idx += 1
return xref_location
def _write_trailer(self, stream: StreamType, xref_location: int) -> None:
"""
Write the PDF trailer to the stream.
To quote the PDF specification:
[The] trailer [gives] the location of the cross-reference table and
of certain special objects within the body of the file.
"""
stream.write(b"trailer\n")
trailer = DictionaryObject(
{
NameObject(TK.SIZE): NumberObject(len(self._objects) + 1),
NameObject(TK.ROOT): self.root_object.indirect_reference,
}
)
if self._info is not None:
trailer[NameObject(TK.INFO)] = self._info.indirect_reference
if self._ID is not None:
trailer[NameObject(TK.ID)] = self._ID
if self._encrypt_entry:
trailer[NameObject(TK.ENCRYPT)] = self._encrypt_entry.indirect_reference
trailer.write_to_stream(stream)
stream.write(f"\nstartxref\n{xref_location}\n%%EOF\n".encode()) # eof
@property
def metadata(self) -> Optional[DocumentInformation]:
"""
Retrieve/set the PDF file's document information dictionary, if it exists.
Args:
value: dict with the entries to be set. if None : remove the /Info entry from the pdf.
Note that some PDF files use (xmp)metadata streams instead of document
information dictionaries, and these metadata streams will not be
accessed by this function.
"""
return super().metadata
@metadata.setter
def metadata(
self,
value: Optional[Union[DocumentInformation, DictionaryObject, Dict[Any, Any]]],
) -> None:
if value is None:
self._info = None
else:
if self._info is not None:
self._info.clear()
else:
self._info = DictionaryObject()
self.add_metadata(value)
def add_metadata(self, infos: Dict[str, Any]) -> None:
"""
Add custom metadata to the output.
Args:
infos: a Python dictionary where each key is a field
and each value is your new metadata.
"""
args = {}
if isinstance(infos, PdfObject):
infos = cast(DictionaryObject, infos.get_object())
for key, value in list(infos.items()):
if isinstance(value, PdfObject):
value = value.get_object()
args[NameObject(key)] = create_string_object(str(value))
assert isinstance(self._info, DictionaryObject)
self._info.update(args)
def compress_identical_objects(
self,
remove_identicals: bool = True,
remove_orphans: bool = True,
) -> None:
"""
Parse the PDF file and merge objects that have the same hash.
This will make objects common to multiple pages.
Recommended to be used just before writing output.
Args:
remove_identicals: Remove identical objects.
remove_orphans: Remove unreferenced objects.
"""
def replace_in_obj(
obj: PdfObject, crossref: Dict[IndirectObject, IndirectObject]
) -> None:
if isinstance(obj, DictionaryObject):
key_val = obj.items()
elif isinstance(obj, ArrayObject):
key_val = enumerate(obj) # type: ignore
else:
return
assert isinstance(obj, (DictionaryObject, ArrayObject))
for k, v in key_val:
if isinstance(v, IndirectObject):
orphans[v.idnum - 1] = False
if v in crossref:
obj[k] = crossref[v]
else:
"""the filtering on DictionaryObject and ArrayObject only
will be performed within replace_in_obj"""
replace_in_obj(v, crossref)
# _idnum_hash :dict[hash]=(1st_ind_obj,[other_indir_objs,...])
self._idnum_hash = {}
orphans = [True] * len(self._objects)
# look for similar objects
for idx, obj in enumerate(self._objects):
if obj is None:
continue
assert isinstance(obj.indirect_reference, IndirectObject)
h = obj.hash_value()
if remove_identicals and h in self._idnum_hash:
self._idnum_hash[h][1].append(obj.indirect_reference)
self._objects[idx] = None
else:
self._idnum_hash[h] = (obj.indirect_reference, [])
# generate the dict converting others to 1st
cnv = {v[0]: v[1] for v in self._idnum_hash.values() if len(v[1]) > 0}
cnv_rev: Dict[IndirectObject, IndirectObject] = {}
for k, v in cnv.items():
cnv_rev.update(zip(v, (k,) * len(v)))
# replace reference to merged objects
for obj in self._objects:
if isinstance(obj, (DictionaryObject, ArrayObject)):
replace_in_obj(obj, cnv_rev)
# remove orphans (if applicable)
orphans[self.root_object.indirect_reference.idnum - 1] = False # type: ignore
orphans[self._info.indirect_reference.idnum - 1] = False # type: ignore
try:
orphans[self._ID.indirect_reference.idnum - 1] = False # type: ignore
except AttributeError:
pass
for i in compress(range(len(self._objects)), orphans):
self._objects[i] = None
def _sweep_indirect_references(
self,
root: Union[
ArrayObject,
BooleanObject,
DictionaryObject,
FloatObject,
IndirectObject,
NameObject,
PdfObject,
NumberObject,
TextStringObject,
NullObject,
],
) -> None: # deprecated
"""
Resolving any circular references to Page objects.
Circular references to Page objects can arise when objects such as
annotations refer to their associated page. If these references are not
properly handled, the PDF file will contain multiple copies of the same
Page object. To address this problem, Page objects store their original
object reference number. This method adds the reference number of any
circularly referenced Page objects to an external reference map. This
ensures that self-referencing trees reference the correct new object
location, rather than copying in a new copy of the Page object.
Args:
root: The root of the PDF object tree to sweep.
"""
deprecate(
"_sweep_indirect_references has been removed, please report to dev team if this warning is observed",
)
def _resolve_indirect_object(
self, data: IndirectObject
) -> IndirectObject: # deprecated
"""
Resolves an indirect object to an indirect object in this PDF file.
If the input indirect object already belongs to this PDF file, it is
returned directly. Otherwise, the object is retrieved from the input
object's PDF file using the object's ID number and generation number. If
the object cannot be found, a warning is logged and a `NullObject` is
returned.
If the object is not already in this PDF file, it is added to the file's
list of objects and assigned a new ID number and generation number of 0.
The hash value of the object is then added to the `_idnum_hash`
dictionary, with the corresponding `IndirectObject` reference as the
value.
Args:
data: The `IndirectObject` to resolve.
Returns:
The resolved `IndirectObject` in this PDF file.
Raises:
ValueError: If the input stream is closed.
"""
deprecate(
"_resolve_indirect_object has been removed, please report to dev team if this warning is observed",
)
return IndirectObject(0, 0, self)
def get_reference(self, obj: PdfObject) -> IndirectObject:
idnum = self._objects.index(obj) + 1
ref = IndirectObject(idnum, 0, self)
assert ref.get_object() == obj
return ref
def get_outline_root(self) -> TreeObject:
if CO.OUTLINES in self._root_object:
# Table 3.25 Entries in the catalog dictionary
outline = cast(TreeObject, self._root_object[CO.OUTLINES])
if not isinstance(outline, TreeObject):
t = TreeObject(outline)
self._replace_object(outline.indirect_reference.idnum, t)
outline = t
idnum = self._objects.index(outline) + 1
outline_ref = IndirectObject(idnum, 0, self)
assert outline_ref.get_object() == outline
else:
outline = TreeObject()
outline.update({})
outline_ref = self._add_object(outline)
self._root_object[NameObject(CO.OUTLINES)] = outline_ref
return outline
def get_threads_root(self) -> ArrayObject:
"""
The list of threads.
See §12.4.3 of the PDF 1.7 or PDF 2.0 specification.
Returns:
An array (possibly empty) of Dictionaries with ``/F`` and
``/I`` properties.
"""
if CO.THREADS in self._root_object:
# Table 3.25 Entries in the catalog dictionary
threads = cast(ArrayObject, self._root_object[CO.THREADS])
else:
threads = ArrayObject()
self._root_object[NameObject(CO.THREADS)] = threads
return threads
@property
def threads(self) -> ArrayObject:
"""
Read-only property for the list of threads.
See §8.3.2 from PDF 1.7 spec.
Each element is a dictionaries with ``/F`` and ``/I`` keys.
"""
return self.get_threads_root()
def add_outline_item_destination(
self,
page_destination: Union[IndirectObject, PageObject, TreeObject],
parent: Union[None, TreeObject, IndirectObject] = None,
before: Union[None, TreeObject, IndirectObject] = None,
is_open: bool = True,
) -> IndirectObject:
page_destination = cast(PageObject, page_destination.get_object())
if isinstance(page_destination, PageObject):
return self.add_outline_item_destination(
Destination(
f"page #{page_destination.page_number}",
cast(IndirectObject, page_destination.indirect_reference),
Fit.fit(),
)
)
if parent is None:
parent = self.get_outline_root()
page_destination[NameObject("/%is_open%")] = BooleanObject(is_open)
parent = cast(TreeObject, parent.get_object())
page_destination_ref = self._add_object(page_destination)
if before is not None:
before = before.indirect_reference
parent.insert_child(
page_destination_ref,
before,
self,
page_destination.inc_parent_counter_outline
if is_open
else (lambda x, y: 0),
)
if "/Count" not in page_destination:
page_destination[NameObject("/Count")] = NumberObject(0)
return page_destination_ref
def add_outline_item_dict(
self,
outline_item: OutlineItemType,
parent: Union[None, TreeObject, IndirectObject] = None,
before: Union[None, TreeObject, IndirectObject] = None,
is_open: bool = True,
) -> IndirectObject:
outline_item_object = TreeObject()
outline_item_object.update(outline_item)
"""code currently unreachable
if "/A" in outline_item:
action = DictionaryObject()
a_dict = cast(DictionaryObject, outline_item["/A"])
for k, v in list(a_dict.items()):
action[NameObject(str(k))] = v
action_ref = self._add_object(action)
outline_item_object[NameObject("/A")] = action_ref
"""
return self.add_outline_item_destination(
outline_item_object, parent, before, is_open
)
def add_outline_item(
self,
title: str,
page_number: Union[None, PageObject, IndirectObject, int],
parent: Union[None, TreeObject, IndirectObject] = None,
before: Union[None, TreeObject, IndirectObject] = None,
color: Optional[Union[Tuple[float, float, float], str]] = None,
bold: bool = False,
italic: bool = False,
fit: Fit = PAGE_FIT,
is_open: bool = True,
) -> IndirectObject:
"""
Add an outline item (commonly referred to as a "Bookmark") to the PDF file.
Args:
title: Title to use for this outline item.
page_number: Page number this outline item will point to.
parent: A reference to a parent outline item to create nested
outline items.
before:
color: Color of the outline item's font as a red, green, blue tuple
from 0.0 to 1.0 or as a Hex String (#RRGGBB)
bold: Outline item font is bold
italic: Outline item font is italic
fit: The fit of the destination page.
Returns:
The added outline item as an indirect object.
"""
page_ref: Union[None, NullObject, IndirectObject, NumberObject]
if isinstance(italic, Fit): # it means that we are on the old params
if fit is not None and page_number is None:
page_number = fit
return self.add_outline_item(
title, page_number, parent, None, before, color, bold, italic, is_open=is_open
)
if page_number is None:
action_ref = None
else:
if isinstance(page_number, IndirectObject):
page_ref = page_number
elif isinstance(page_number, PageObject):
page_ref = page_number.indirect_reference
elif isinstance(page_number, int):
try:
page_ref = self.pages[page_number].indirect_reference
except IndexError:
page_ref = NumberObject(page_number)
if page_ref is None:
logger_warning(
f"can not find reference of page {page_number}",
__name__,
)
page_ref = NullObject()
dest = Destination(
NameObject("/" + title + " outline item"),
page_ref,
fit,
)
action_ref = self._add_object(
DictionaryObject(
{
NameObject(GoToActionArguments.D): dest.dest_array,
NameObject(GoToActionArguments.S): NameObject("/GoTo"),
}
)
)
outline_item = self._add_object(
_create_outline_item(action_ref, title, color, italic, bold)
)
if parent is None:
parent = self.get_outline_root()
return self.add_outline_item_destination(outline_item, parent, before, is_open)
def add_outline(self) -> None:
raise NotImplementedError(
"This method is not yet implemented. Use :meth:`add_outline_item` instead."
)
def add_named_destination_array(
self, title: TextStringObject, destination: Union[IndirectObject, ArrayObject]
) -> None:
named_dest = self.get_named_dest_root()
i = 0
while i < len(named_dest):
if title < named_dest[i]:
named_dest.insert(i, destination)
named_dest.insert(i, TextStringObject(title))
return
else:
i += 2
named_dest.extend([TextStringObject(title), destination])
return
def add_named_destination_object(
self,
page_destination: PdfObject,
) -> IndirectObject:
page_destination_ref = self._add_object(page_destination.dest_array) # type: ignore
self.add_named_destination_array(
cast("TextStringObject", page_destination["/Title"]), page_destination_ref # type: ignore
)
return page_destination_ref
def add_named_destination(
self,
title: str,
page_number: int,
) -> IndirectObject:
page_ref = self.get_object(self._pages)[PA.KIDS][page_number] # type: ignore
dest = DictionaryObject()
dest.update(
{
NameObject(GoToActionArguments.D): ArrayObject(
[page_ref, NameObject(TypFitArguments.FIT_H), NumberObject(826)]
),
NameObject(GoToActionArguments.S): NameObject("/GoTo"),
}
)
dest_ref = self._add_object(dest)
if not isinstance(title, TextStringObject):
title = TextStringObject(str(title))
self.add_named_destination_array(title, dest_ref)
return dest_ref
def remove_links(self) -> None:
"""Remove links and annotations from this output."""
for page in self.pages:
self.remove_objects_from_page(page, ObjectDeletionFlag.ALL_ANNOTATIONS)
def remove_annotations(
self, subtypes: Optional[Union[AnnotationSubtype, Iterable[AnnotationSubtype]]]
) -> None:
"""
Remove annotations by annotation subtype.
Args:
subtypes: subtype or list of subtypes to be removed.
Examples are: "/Link", "/FileAttachment", "/Sound",
"/Movie", "/Screen", ...
If you want to remove all annotations, use subtypes=None.
"""
for page in self.pages:
self._remove_annots_from_page(page, subtypes)
def _remove_annots_from_page(
self,
page: Union[IndirectObject, PageObject, DictionaryObject],
subtypes: Optional[Iterable[str]],
) -> None:
page = cast(DictionaryObject, page.get_object())
if PG.ANNOTS in page:
i = 0
while i < len(cast(ArrayObject, page[PG.ANNOTS])):
an = cast(ArrayObject, page[PG.ANNOTS])[i]
obj = cast(DictionaryObject, an.get_object())
if subtypes is None or cast(str, obj["/Subtype"]) in subtypes:
if isinstance(an, IndirectObject):
self._objects[an.idnum - 1] = NullObject() # to reduce PDF size
del page[PG.ANNOTS][i] # type:ignore
else:
i += 1
def remove_objects_from_page(
self,
page: Union[PageObject, DictionaryObject],
to_delete: Union[ObjectDeletionFlag, Iterable[ObjectDeletionFlag]],
) -> None:
"""
Remove objects specified by ``to_delete`` from the given page.
Args:
page: Page object to clean up.
to_delete: Objects to be deleted; can be a ``ObjectDeletionFlag``
or a list of ObjectDeletionFlag
"""
if isinstance(to_delete, (list, tuple)):
for to_d in to_delete:
self.remove_objects_from_page(page, to_d)
return
assert isinstance(to_delete, ObjectDeletionFlag)
if to_delete & ObjectDeletionFlag.LINKS:
return self._remove_annots_from_page(page, ("/Link",))
if to_delete & ObjectDeletionFlag.ATTACHMENTS:
return self._remove_annots_from_page(
page, ("/FileAttachment", "/Sound", "/Movie", "/Screen")
)
if to_delete & ObjectDeletionFlag.OBJECTS_3D:
return self._remove_annots_from_page(page, ("/3D",))
if to_delete & ObjectDeletionFlag.ALL_ANNOTATIONS:
return self._remove_annots_from_page(page, None)
jump_operators = []
if to_delete & ObjectDeletionFlag.DRAWING_IMAGES:
jump_operators = (
[b"w", b"J", b"j", b"M", b"d", b"i"]
+ [b"W", b"W*"]
+ [b"b", b"b*", b"B", b"B*", b"S", b"s", b"f", b"f*", b"F", b"n"]
+ [b"m", b"l", b"c", b"v", b"y", b"h", b"re"]
+ [b"sh"]
)
if to_delete & ObjectDeletionFlag.TEXT:
jump_operators = [b"Tj", b"TJ", b"'", b'"']
def clean(content: ContentStream, images: List[str], forms: List[str]) -> None:
nonlocal jump_operators, to_delete
i = 0
while i < len(content.operations):
operands, operator = content.operations[i]
if (
(
operator == b"INLINE IMAGE"
and (to_delete & ObjectDeletionFlag.INLINE_IMAGES)
)
or (operator in jump_operators)
or (
operator == b"Do"
and (to_delete & ObjectDeletionFlag.XOBJECT_IMAGES)
and (operands[0] in images)
)
):
del content.operations[i]
else:
i += 1
content.get_data() # this ensures ._data is rebuilt from the .operations
def clean_forms(
elt: DictionaryObject, stack: List[DictionaryObject]
) -> Tuple[List[str], List[str]]:
nonlocal to_delete
# elt in recursive call is a new ContentStream object, so we have to check the indirect_reference
if (elt in stack) or (
hasattr(elt, "indirect_reference")
and any(
elt.indirect_reference == getattr(x, "indirect_reference", -1)
for x in stack
)
):
# to prevent infinite looping
return [], [] # pragma: no cover
try:
d = cast(
Dict[Any, Any],
cast(DictionaryObject, elt["/Resources"])["/XObject"],
)
except KeyError:
d = {}
images = []
forms = []
for k, v in d.items():
o = v.get_object()
try:
content: Any = None
if (
to_delete & ObjectDeletionFlag.XOBJECT_IMAGES
and o["/Subtype"] == "/Image"
):
content = NullObject() # to delete the image keeping the entry
images.append(k)
if o["/Subtype"] == "/Form":
forms.append(k)
if isinstance(o, ContentStream):
content = o
else:
content = ContentStream(o, self)
content.update(
{
k1: v1
for k1, v1 in o.items()
if k1 not in ["/Length", "/Filter", "/DecodeParms"]
}
)
try:
content.indirect_reference = o.indirect_reference
except AttributeError: # pragma: no cover
pass
stack.append(elt)
clean_forms(content, stack) # clean subforms
if content is not None:
if isinstance(v, IndirectObject):
self._objects[v.idnum - 1] = content
else:
# should only occur with pdf not respecting pdf spec
# where streams must be indirected.
d[k] = self._add_object(content) # pragma: no cover
except (TypeError, KeyError):
pass
for im in images:
del d[im] # for clean-up
if isinstance(elt, StreamObject): # for /Form
if not isinstance(elt, ContentStream): # pragma: no cover
e = ContentStream(elt, self)
e.update(elt.items())
elt = e
clean(elt, images, forms) # clean the content
return images, forms
if not isinstance(page, PageObject):
page = PageObject(self, page.indirect_reference) # pragma: no cover
if "/Contents" in page:
content = cast(ContentStream, page.get_contents())
images, forms = clean_forms(page, [])
clean(content, images, forms)
page.replace_contents(content)
def remove_images(
self,
to_delete: ImageType = ImageType.ALL,
) -> None:
"""
Remove images from this output.
Args:
to_delete : The type of images to be deleted
(default = all images types)
"""
if isinstance(to_delete, bool):
to_delete = ImageType.ALL
i = (
(
ObjectDeletionFlag.XOBJECT_IMAGES
if to_delete & ImageType.XOBJECT_IMAGES
else ObjectDeletionFlag.NONE
)
| (
ObjectDeletionFlag.INLINE_IMAGES
if to_delete & ImageType.INLINE_IMAGES
else ObjectDeletionFlag.NONE
)
| (
ObjectDeletionFlag.DRAWING_IMAGES
if to_delete & ImageType.DRAWING_IMAGES
else ObjectDeletionFlag.NONE
)
)
for page in self.pages:
self.remove_objects_from_page(page, i)
def remove_text(self) -> None:
"""Remove text from this output."""
for page in self.pages:
self.remove_objects_from_page(page, ObjectDeletionFlag.TEXT)
def add_uri(
self,
page_number: int,
uri: str,
rect: RectangleObject,
border: Optional[ArrayObject] = None,
) -> None:
"""
Add an URI from a rectangular area to the specified page.
Args:
page_number: index of the page on which to place the URI action.
uri: URI of resource to link to.
rect: :class:`RectangleObject<pypdf.generic.RectangleObject>` or
array of four integers specifying the clickable rectangular area
``[xLL, yLL, xUR, yUR]``, or string in the form
``"[ xLL yLL xUR yUR ]"``.
border: if provided, an array describing border-drawing
properties. See the PDF spec for details. No border will be
drawn if this argument is omitted.
"""
page_link = self.get_object(self._pages)[PA.KIDS][page_number] # type: ignore
page_ref = cast(Dict[str, Any], self.get_object(page_link))
border_arr: BorderArrayType
if border is not None:
border_arr = [NumberObject(n) for n in border[:3]]
if len(border) == 4:
dash_pattern = ArrayObject([NumberObject(n) for n in border[3]])
border_arr.append(dash_pattern)
else:
border_arr = [NumberObject(2), NumberObject(2), NumberObject(2)]
if isinstance(rect, str):
rect = NumberObject(rect)
elif isinstance(rect, RectangleObject):
pass
else:
rect = RectangleObject(rect)
lnk2 = DictionaryObject()
lnk2.update(
{
NameObject("/S"): NameObject("/URI"),
NameObject("/URI"): TextStringObject(uri),
}
)
lnk = DictionaryObject()
lnk.update(
{
NameObject(AA.Type): NameObject("/Annot"),
NameObject(AA.Subtype): NameObject("/Link"),
NameObject(AA.P): page_link,
NameObject(AA.Rect): rect,
NameObject("/H"): NameObject("/I"),
NameObject(AA.Border): ArrayObject(border_arr),
NameObject("/A"): lnk2,
}
)
lnk_ref = self._add_object(lnk)
if PG.ANNOTS in page_ref:
page_ref[PG.ANNOTS].append(lnk_ref)
else:
page_ref[NameObject(PG.ANNOTS)] = ArrayObject([lnk_ref])
_valid_layouts = (
"/NoLayout",
"/SinglePage",
"/OneColumn",
"/TwoColumnLeft",
"/TwoColumnRight",
"/TwoPageLeft",
"/TwoPageRight",
)
def _get_page_layout(self) -> Optional[LayoutType]:
try:
return cast(LayoutType, self._root_object["/PageLayout"])
except KeyError:
return None
def _set_page_layout(self, layout: Union[NameObject, LayoutType]) -> None:
"""
Set the page layout.
Args:
layout: The page layout to be used.
.. list-table:: Valid ``layout`` arguments
:widths: 50 200
* - /NoLayout
- Layout explicitly not specified
* - /SinglePage
- Show one page at a time
* - /OneColumn
- Show one column at a time
* - /TwoColumnLeft
- Show pages in two columns, odd-numbered pages on the left
* - /TwoColumnRight
- Show pages in two columns, odd-numbered pages on the right
* - /TwoPageLeft
- Show two pages at a time, odd-numbered pages on the left
* - /TwoPageRight
- Show two pages at a time, odd-numbered pages on the right
"""
if not isinstance(layout, NameObject):
if layout not in self._valid_layouts:
logger_warning(
f"Layout should be one of: {'', ''.join(self._valid_layouts)}",
__name__,
)
layout = NameObject(layout)
self._root_object.update({NameObject("/PageLayout"): layout})
def set_page_layout(self, layout: LayoutType) -> None:
"""
Set the page layout.
Args:
layout: The page layout to be used
.. list-table:: Valid ``layout`` arguments
:widths: 50 200
* - /NoLayout
- Layout explicitly not specified
* - /SinglePage
- Show one page at a time
* - /OneColumn
- Show one column at a time
* - /TwoColumnLeft
- Show pages in two columns, odd-numbered pages on the left
* - /TwoColumnRight
- Show pages in two columns, odd-numbered pages on the right
* - /TwoPageLeft
- Show two pages at a time, odd-numbered pages on the left
* - /TwoPageRight
- Show two pages at a time, odd-numbered pages on the right
"""
self._set_page_layout(layout)
@property
def page_layout(self) -> Optional[LayoutType]:
"""
Page layout property.
.. list-table:: Valid ``layout`` values
:widths: 50 200
* - /NoLayout
- Layout explicitly not specified
* - /SinglePage
- Show one page at a time
* - /OneColumn
- Show one column at a time
* - /TwoColumnLeft
- Show pages in two columns, odd-numbered pages on the left
* - /TwoColumnRight
- Show pages in two columns, odd-numbered pages on the right
* - /TwoPageLeft
- Show two pages at a time, odd-numbered pages on the left
* - /TwoPageRight
- Show two pages at a time, odd-numbered pages on the right
"""
return self._get_page_layout()
@page_layout.setter
def page_layout(self, layout: LayoutType) -> None:
self._set_page_layout(layout)
_valid_modes = (
"/UseNone",
"/UseOutlines",
"/UseThumbs",
"/FullScreen",
"/UseOC",
"/UseAttachments",
)
def _get_page_mode(self) -> Optional[PagemodeType]:
try:
return cast(PagemodeType, self._root_object["/PageMode"])
except KeyError:
return None
@property
def page_mode(self) -> Optional[PagemodeType]:
"""
Page mode property.
.. list-table:: Valid ``mode`` values
:widths: 50 200
* - /UseNone
- Do not show outline or thumbnails panels
* - /UseOutlines
- Show outline (aka bookmarks) panel
* - /UseThumbs
- Show page thumbnails panel
* - /FullScreen
- Fullscreen view
* - /UseOC
- Show Optional Content Group (OCG) panel
* - /UseAttachments
- Show attachments panel
"""
return self._get_page_mode()
@page_mode.setter
def page_mode(self, mode: PagemodeType) -> None:
if isinstance(mode, NameObject):
mode_name: NameObject = mode
else:
if mode not in self._valid_modes:
logger_warning(
f"Mode should be one of: {', '.join(self._valid_modes)}", __name__
)
mode_name = NameObject(mode)
self._root_object.update({NameObject("/PageMode"): mode_name})
def add_annotation(
self,
page_number: Union[int, PageObject],
annotation: Dict[str, Any],
) -> DictionaryObject:
"""
Add a single annotation to the page.
The added annotation must be a new annotation.
It cannot be recycled.
Args:
page_number: PageObject or page index.
annotation: Annotation to be added (created with annotation).
Returns:
The inserted object.
This can be used for popup creation, for example.
"""
page = page_number
if isinstance(page, int):
page = self.pages[page]
elif not isinstance(page, PageObject):
raise TypeError("page: invalid type")
to_add = cast(DictionaryObject, _pdf_objectify(annotation))
to_add[NameObject("/P")] = page.indirect_reference
if page.annotations is None:
page[NameObject("/Annots")] = ArrayObject()
assert page.annotations is not None
# Internal link annotations need the correct object type for the
# destination
if to_add.get("/Subtype") == "/Link" and "/Dest" in to_add:
tmp = cast(Dict[Any, Any], to_add[NameObject("/Dest")])
dest = Destination(
NameObject("/LinkName"),
tmp["target_page_index"],
Fit(
fit_type=tmp["fit"], fit_args=dict(tmp)["fit_args"]
), # I have no clue why this dict-hack is necessary
)
to_add[NameObject("/Dest")] = dest.dest_array
page.annotations.append(self._add_object(to_add))
if to_add.get("/Subtype") == "/Popup" and NameObject("/Parent") in to_add:
cast(DictionaryObject, to_add["/Parent"].get_object())[
NameObject("/Popup")
] = to_add.indirect_reference
return to_add
def clean_page(self, page: Union[PageObject, IndirectObject]) -> PageObject:
"""
Perform some clean up in the page.
Currently: convert NameObject named destination to TextStringObject
(required for names/dests list)
Args:
page:
Returns:
The cleaned PageObject
"""
page = cast("PageObject", page.get_object())
for a in page.get("/Annots", []):
a_obj = a.get_object()
d = a_obj.get("/Dest", None)
act = a_obj.get("/A", None)
if isinstance(d, NameObject):
a_obj[NameObject("/Dest")] = TextStringObject(d)
elif act is not None:
act = act.get_object()
d = act.get("/D", None)
if isinstance(d, NameObject):
act[NameObject("/D")] = TextStringObject(d)
return page
def _create_stream(
self, fileobj: Union[Path, StrByteType, PdfReader]
) -> Tuple[IOBase, Optional[Encryption]]:
# If the fileobj parameter is a string, assume it is a path
# and create a file object at that location. If it is a file,
# copy the file's contents into a BytesIO stream object; if
# it is a PdfReader, copy that reader's stream into a
# BytesIO stream.
# If fileobj is none of the above types, it is not modified
encryption_obj = None
stream: IOBase
if isinstance(fileobj, (str, Path)):
with FileIO(fileobj, "rb") as f:
stream = BytesIO(f.read())
elif isinstance(fileobj, PdfReader):
if fileobj._encryption:
encryption_obj = fileobj._encryption
orig_tell = fileobj.stream.tell()
fileobj.stream.seek(0)
stream = BytesIO(fileobj.stream.read())
# reset the stream to its original location
fileobj.stream.seek(orig_tell)
elif hasattr(fileobj, "seek") and hasattr(fileobj, "read"):
fileobj.seek(0)
filecontent = fileobj.read()
stream = BytesIO(filecontent)
else:
raise NotImplementedError(
"Merging requires an object that PdfReader can parse. "
"Typically, that is a Path or a string representing a Path, "
"a file object, or an object implementing .seek and .read. "
"Passing a PdfReader directly works as well."
)
return stream, encryption_obj
def append(
self,
fileobj: Union[StrByteType, PdfReader, Path],
outline_item: Union[
str, None, PageRange, Tuple[int, int], Tuple[int, int, int], List[int]
] = None,
pages: Union[
None,
PageRange,
Tuple[int, int],
Tuple[int, int, int],
List[int],
List[PageObject],
] = None,
import_outline: bool = True,
excluded_fields: Optional[Union[List[str], Tuple[str, ...]]] = None,
) -> None:
"""
Identical to the :meth:`merge()<merge>` method, but assumes you want to
concatenate all pages onto the end of the file instead of specifying a
position.
Args:
fileobj: A File Object or an object that supports the standard
read and seek methods similar to a File Object. Could also be a
string representing a path to a PDF file.
outline_item: Optionally, you may specify a string to build an
outline (aka 'bookmark') to identify the beginning of the
included file.
pages: Can be a :class:`PageRange<pypdf.pagerange.PageRange>`
or a ``(start, stop[, step])`` tuple
or a list of pages to be processed
to merge only the specified range of pages from the source
document into the output document.
import_outline: You may prevent the source document's
outline (collection of outline items, previously referred to as
'bookmarks') from being imported by specifying this as ``False``.
excluded_fields: Provide the list of fields/keys to be ignored
if ``/Annots`` is part of the list, the annotation will be ignored
if ``/B`` is part of the list, the articles will be ignored
"""
if excluded_fields is None:
excluded_fields = ()
if isinstance(outline_item, (tuple, list, PageRange)):
if isinstance(pages, bool):
if not isinstance(import_outline, bool):
excluded_fields = import_outline
import_outline = pages
pages = outline_item
self.merge(
None,
fileobj,
None,
pages,
import_outline,
excluded_fields,
)
else: # if isinstance(outline_item,str):
self.merge(
None,
fileobj,
outline_item,
pages,
import_outline,
excluded_fields,
)
def merge(
self,
position: Optional[int],
fileobj: Union[Path, StrByteType, PdfReader],
outline_item: Optional[str] = None,
pages: Optional[Union[PageRangeSpec, List[PageObject]]] = None,
import_outline: bool = True,
excluded_fields: Optional[Union[List[str], Tuple[str, ...]]] = (),
) -> None:
"""
Merge the pages from the given file into the output file at the
specified page number.
Args:
position: The *page number* to insert this file. File will
be inserted after the given number.
fileobj: A File Object or an object that supports the standard
read and seek methods similar to a File Object. Could also be a
string representing a path to a PDF file.
outline_item: Optionally, you may specify a string to build an outline
(aka 'bookmark') to identify the
beginning of the included file.
pages: can be a :class:`PageRange<pypdf.pagerange.PageRange>`
or a ``(start, stop[, step])`` tuple
or a list of pages to be processed
to merge only the specified range of pages from the source
document into the output document.
import_outline: You may prevent the source document's
outline (collection of outline items, previously referred to as
'bookmarks') from being imported by specifying this as ``False``.
excluded_fields: provide the list of fields/keys to be ignored
if ``/Annots`` is part of the list, the annotation will be ignored
if ``/B`` is part of the list, the articles will be ignored
Raises:
TypeError: The pages attribute is not configured properly
"""
if isinstance(fileobj, PdfDocCommon):
reader = fileobj
else:
stream, encryption_obj = self._create_stream(fileobj)
# Create a new PdfReader instance using the stream
# (either file or BytesIO or StringIO) created above
reader = PdfReader(stream, strict=False) # type: ignore[arg-type]
if excluded_fields is None:
excluded_fields = ()
# Find the range of pages to merge.
if pages is None:
pages = list(range(len(reader.pages)))
elif isinstance(pages, PageRange):
pages = list(range(*pages.indices(len(reader.pages))))
elif isinstance(pages, list):
pass # keep unchanged
elif isinstance(pages, tuple) and len(pages) <= 3:
pages = list(range(*pages))
elif not isinstance(pages, tuple):
raise TypeError(
'"pages" must be a tuple of (start, stop[, step]) or a list'
)
srcpages = {}
for page in pages:
if isinstance(page, PageObject):
pg = page
else:
pg = reader.pages[page]
assert pg.indirect_reference is not None
if position is None:
# numbers in the exclude list identifies that the exclusion is
# only applicable to 1st level of cloning
srcpages[pg.indirect_reference.idnum] = self.add_page(
pg, list(excluded_fields) + [1, "/B", 1, "/Annots"] # type: ignore
)
else:
srcpages[pg.indirect_reference.idnum] = self.insert_page(
pg, position, list(excluded_fields) + [1, "/B", 1, "/Annots"] # type: ignore
)
position += 1
srcpages[pg.indirect_reference.idnum].original_page = pg
reader._named_destinations = (
reader.named_destinations
) # need for the outline processing below
arr: Any
def _process_named_dests(dest: Any) -> None:
arr = dest.dest_array
if "/Names" in self._root_object and dest["/Title"] in cast(
List[Any],
cast(
DictionaryObject,
cast(DictionaryObject, self._root_object["/Names"])["/Dests"],
)["/Names"],
):
# already exists : should not duplicate it
pass
elif isinstance(dest["/Page"], NullObject):
pass
elif isinstance(dest["/Page"], int):
# the page reference is a page number normally not a PDF Reference
# page numbers as int are normally accepted only in external goto
try:
p = reader.pages[dest["/Page"]]
except IndexError:
return
assert p.indirect_reference is not None
try:
arr[NumberObject(0)] = NumberObject(
srcpages[p.indirect_reference.idnum].page_number
)
self.add_named_destination_array(dest["/Title"], arr)
except KeyError:
pass
elif dest["/Page"].indirect_reference.idnum in srcpages:
arr[NumberObject(0)] = srcpages[
dest["/Page"].indirect_reference.idnum
].indirect_reference
self.add_named_destination_array(dest["/Title"], arr)
for dest in reader._named_destinations.values():
_process_named_dests(dest)
outline_item_typ: TreeObject
if outline_item is not None:
outline_item_typ = cast(
"TreeObject",
self.add_outline_item(
TextStringObject(outline_item),
next(iter(srcpages.values())).indirect_reference,
fit=PAGE_FIT,
).get_object(),
)
else:
outline_item_typ = self.get_outline_root()
_ro = reader.root_object
if import_outline and CO.OUTLINES in _ro:
outline = self._get_filtered_outline(
_ro.get(CO.OUTLINES, None), srcpages, reader
)
self._insert_filtered_outline(
outline, outline_item_typ, None
) # TODO : use before parameter
if "/Annots" not in excluded_fields:
for pag in srcpages.values():
lst = self._insert_filtered_annotations(
pag.original_page.get("/Annots", ()), pag, srcpages, reader
)
if len(lst) > 0:
pag[NameObject("/Annots")] = lst
self.clean_page(pag)
if "/AcroForm" in _ro and _ro["/AcroForm"] is not None:
if "/AcroForm" not in self._root_object:
self._root_object[NameObject("/AcroForm")] = self._add_object(
cast(
DictionaryObject,
reader.root_object["/AcroForm"],
).clone(self, False, ("/Fields",))
)
arr = ArrayObject()
else:
arr = cast(
ArrayObject,
cast(DictionaryObject, self._root_object["/AcroForm"])["/Fields"],
)
trslat = self._id_translated[id(reader)]
try:
for f in reader.root_object["/AcroForm"]["/Fields"]: # type: ignore
try:
ind = IndirectObject(trslat[f.idnum], 0, self)
if ind not in arr:
arr.append(ind)
except KeyError:
# for trslat[] which mean the field has not be copied
# through the page
pass
except KeyError: # for /Acroform or /Fields are not existing
arr = self._add_object(ArrayObject())
cast(DictionaryObject, self._root_object["/AcroForm"])[
NameObject("/Fields")
] = arr
if "/B" not in excluded_fields:
self.add_filtered_articles("", srcpages, reader)
def _add_articles_thread(
self,
thread: DictionaryObject, # thread entry from the reader's array of threads
pages: Dict[int, PageObject],
reader: PdfReader,
) -> IndirectObject:
"""
Clone the thread with only the applicable articles.
Args:
thread:
pages:
reader:
Returns:
The added thread as an indirect reference
"""
nthread = thread.clone(
self, force_duplicate=True, ignore_fields=("/F",)
) # use of clone to keep link between reader and writer
self.threads.append(nthread.indirect_reference)
first_article = cast("DictionaryObject", thread["/F"])
current_article: Optional[DictionaryObject] = first_article
new_article: Optional[DictionaryObject] = None
while current_article is not None:
pag = self._get_cloned_page(
cast("PageObject", current_article["/P"]), pages, reader
)
if pag is not None:
if new_article is None:
new_article = cast(
"DictionaryObject",
self._add_object(DictionaryObject()).get_object(),
)
new_first = new_article
nthread[NameObject("/F")] = new_article.indirect_reference
else:
new_article2 = cast(
"DictionaryObject",
self._add_object(
DictionaryObject(
{NameObject("/V"): new_article.indirect_reference}
)
).get_object(),
)
new_article[NameObject("/N")] = new_article2.indirect_reference
new_article = new_article2
new_article[NameObject("/P")] = pag
new_article[NameObject("/T")] = nthread.indirect_reference
new_article[NameObject("/R")] = current_article["/R"]
pag_obj = cast("PageObject", pag.get_object())
if "/B" not in pag_obj:
pag_obj[NameObject("/B")] = ArrayObject()
cast("ArrayObject", pag_obj["/B"]).append(
new_article.indirect_reference
)
current_article = cast("DictionaryObject", current_article["/N"])
if current_article == first_article:
new_article[NameObject("/N")] = new_first.indirect_reference # type: ignore
new_first[NameObject("/V")] = new_article.indirect_reference # type: ignore
current_article = None
assert nthread.indirect_reference is not None
return nthread.indirect_reference
def add_filtered_articles(
self,
fltr: Union[
Pattern[Any], str
], # thread entry from the reader's array of threads
pages: Dict[int, PageObject],
reader: PdfReader,
) -> None:
"""
Add articles matching the defined criteria.
Args:
fltr:
pages:
reader:
"""
if isinstance(fltr, str):
fltr = re.compile(fltr)
elif not isinstance(fltr, Pattern):
fltr = re.compile("")
for p in pages.values():
pp = p.original_page
for a in pp.get("/B", ()):
thr = a.get_object().get("/T")
if thr is None:
continue
else:
thr = thr.get_object()
if thr.indirect_reference.idnum not in self._id_translated[
id(reader)
] and fltr.search((thr.get("/I", {})).get("/Title", "")):
self._add_articles_thread(thr, pages, reader)
def _get_cloned_page(
self,
page: Union[None, IndirectObject, PageObject, NullObject],
pages: Dict[int, PageObject],
reader: PdfReader,
) -> Optional[IndirectObject]:
if isinstance(page, NullObject):
return None
elif isinstance(page, DictionaryObject) and page.get("/Type", "") == "/Page":
_i = page.indirect_reference
elif isinstance(page, IndirectObject):
_i = page
try:
return pages[_i.idnum].indirect_reference # type: ignore
except Exception:
return None
def _insert_filtered_annotations(
self,
annots: Union[IndirectObject, List[DictionaryObject]],
page: PageObject,
pages: Dict[int, PageObject],
reader: PdfReader,
) -> List[Destination]:
outlist = ArrayObject()
if isinstance(annots, IndirectObject):
annots = cast("List[Any]", annots.get_object())
for an in annots:
ano = cast("DictionaryObject", an.get_object())
if (
ano["/Subtype"] != "/Link"
or "/A" not in ano
or cast("DictionaryObject", ano["/A"])["/S"] != "/GoTo"
or "/Dest" in ano
):
if "/Dest" not in ano:
outlist.append(self._add_object(ano.clone(self)))
else:
d = ano["/Dest"]
if isinstance(d, str):
# it is a named dest
if str(d) in self.get_named_dest_root():
outlist.append(ano.clone(self).indirect_reference)
else:
d = cast("ArrayObject", d)
p = self._get_cloned_page(d[0], pages, reader)
if p is not None:
anc = ano.clone(self, ignore_fields=("/Dest",))
anc[NameObject("/Dest")] = ArrayObject([p] + d[1:])
outlist.append(self._add_object(anc))
else:
d = cast("DictionaryObject", ano["/A"])["/D"]
if isinstance(d, str):
# it is a named dest
if str(d) in self.get_named_dest_root():
outlist.append(ano.clone(self).indirect_reference)
else:
d = cast("ArrayObject", d)
p = self._get_cloned_page(d[0], pages, reader)
if p is not None:
anc = ano.clone(self, ignore_fields=("/D",))
cast("DictionaryObject", anc["/A"])[
NameObject("/D")
] = ArrayObject([p] + d[1:])
outlist.append(self._add_object(anc))
return outlist
def _get_filtered_outline(
self,
node: Any,
pages: Dict[int, PageObject],
reader: PdfReader,
) -> List[Destination]:
"""
Extract outline item entries that are part of the specified page set.
Args:
node:
pages:
reader:
Returns:
A list of destination objects.
"""
new_outline = []
if node is None:
node = NullObject()
node = node.get_object()
if is_null_or_none(node):
node = DictionaryObject()
if node.get("/Type", "") == "/Outlines" or "/Title" not in node:
node = node.get("/First", None)
if node is not None:
node = node.get_object()
new_outline += self._get_filtered_outline(node, pages, reader)
else:
v: Union[None, IndirectObject, NullObject]
while node is not None:
node = node.get_object()
o = cast("Destination", reader._build_outline_item(node))
v = self._get_cloned_page(cast("PageObject", o["/Page"]), pages, reader)
if v is None:
v = NullObject()
o[NameObject("/Page")] = v
if "/First" in node:
o._filtered_children = self._get_filtered_outline(
node["/First"], pages, reader
)
else:
o._filtered_children = []
if (
not isinstance(o["/Page"], NullObject)
or len(o._filtered_children) > 0
):
new_outline.append(o)
node = node.get("/Next", None)
return new_outline
def _clone_outline(self, dest: Destination) -> TreeObject:
n_ol = TreeObject()
self._add_object(n_ol)
n_ol[NameObject("/Title")] = TextStringObject(dest["/Title"])
if not isinstance(dest["/Page"], NullObject):
if dest.node is not None and "/A" in dest.node:
n_ol[NameObject("/A")] = dest.node["/A"].clone(self)
else:
n_ol[NameObject("/Dest")] = dest.dest_array
# TODO: /SE
if dest.node is not None:
n_ol[NameObject("/F")] = NumberObject(dest.node.get("/F", 0))
n_ol[NameObject("/C")] = ArrayObject(
dest.node.get(
"/C", [FloatObject(0.0), FloatObject(0.0), FloatObject(0.0)]
)
)
return n_ol
def _insert_filtered_outline(
self,
outlines: List[Destination],
parent: Union[TreeObject, IndirectObject],
before: Union[None, TreeObject, IndirectObject] = None,
) -> None:
for dest in outlines:
# TODO : can be improved to keep A and SE entries (ignored for the moment)
# with np=self.add_outline_item_destination(dest,parent,before)
if dest.get("/Type", "") == "/Outlines" or "/Title" not in dest:
np = parent
else:
np = self._clone_outline(dest)
cast(TreeObject, parent.get_object()).insert_child(np, before, self)
self._insert_filtered_outline(dest._filtered_children, np, None)
def close(self) -> None:
"""Implemented for API harmonization."""
return
def find_outline_item(
self,
outline_item: Dict[str, Any],
root: Optional[OutlineType] = None,
) -> Optional[List[int]]:
if root is None:
o = self.get_outline_root()
else:
o = cast("TreeObject", root)
i = 0
while o is not None:
if (
o.indirect_reference == outline_item
or o.get("/Title", None) == outline_item
):
return [i]
elif "/First" in o:
res = self.find_outline_item(
outline_item, cast(OutlineType, o["/First"])
)
if res:
return ([i] if "/Title" in o else []) + res
if "/Next" in o:
i += 1
o = cast(TreeObject, o["/Next"])
else:
return None
def find_bookmark(
self,
outline_item: Dict[str, Any],
root: Optional[OutlineType] = None,
) -> None: # deprecated
"""
.. deprecated:: 2.9.0
Use :meth:`find_outline_item` instead.
"""
deprecation_with_replacement("find_bookmark", "find_outline_item", "5.0.0")
def reset_translation(
self, reader: Union[None, PdfReader, IndirectObject] = None
) -> None:
"""
Reset the translation table between reader and the writer object.
Late cloning will create new independent objects.
Args:
reader: PdfReader or IndirectObject referencing a PdfReader object.
if set to None or omitted, all tables will be reset.
"""
if reader is None:
self._id_translated = {}
elif isinstance(reader, PdfReader):
try:
del self._id_translated[id(reader)]
except Exception:
pass
elif isinstance(reader, IndirectObject):
try:
del self._id_translated[id(reader.pdf)]
except Exception:
pass
else:
raise Exception("invalid parameter {reader}")
def set_page_label(
self,
page_index_from: int,
page_index_to: int,
style: Optional[PageLabelStyle] = None,
prefix: Optional[str] = None,
start: Optional[int] = 0,
) -> None:
"""
Set a page label to a range of pages.
Page indexes must be given starting from 0.
Labels must have a style, a prefix or both.
If a range is not assigned any page label, a decimal label starting from 1 is applied.
Args:
page_index_from: page index of the beginning of the range starting from 0
page_index_to: page index of the beginning of the range starting from 0
style: The numbering style to be used for the numeric portion of each page label:
* ``/D`` Decimal Arabic numerals
* ``/R`` Uppercase Roman numerals
* ``/r`` Lowercase Roman numerals
* ``/A`` Uppercase letters (A to Z for the first 26 pages,
AA to ZZ for the next 26, and so on)
* ``/a`` Lowercase letters (a to z for the first 26 pages,
aa to zz for the next 26, and so on)
prefix: The label prefix for page labels in this range.
start: The value of the numeric portion for the first page label
in the range.
Subsequent pages are numbered sequentially from this value,
which must be greater than or equal to 1.
Default value: 1.
"""
if style is None and prefix is None:
raise ValueError("At least one of style and prefix must be given")
if page_index_from < 0:
raise ValueError("page_index_from must be greater or equal than 0")
if page_index_to < page_index_from:
raise ValueError(
"page_index_to must be greater or equal than page_index_from"
)
if page_index_to >= len(self.pages):
raise ValueError("page_index_to exceeds number of pages")
if start is not None and start != 0 and start < 1:
raise ValueError("If given, start must be greater or equal than one")
self._set_page_label(page_index_from, page_index_to, style, prefix, start)
def _set_page_label(
self,
page_index_from: int,
page_index_to: int,
style: Optional[PageLabelStyle] = None,
prefix: Optional[str] = None,
start: Optional[int] = 0,
) -> None:
"""
Set a page label to a range of pages.
Page indexes must be given starting from 0.
Labels must have a style, a prefix or both.
If a range is not assigned any page label a decimal label starting from 1 is applied.
Args:
page_index_from: page index of the beginning of the range starting from 0
page_index_to: page index of the beginning of the range starting from 0
style: The numbering style to be used for the numeric portion of each page label:
/D Decimal Arabic numerals
/R Uppercase Roman numerals
/r Lowercase Roman numerals
/A Uppercase letters (A to Z for the first 26 pages,
AA to ZZ for the next 26, and so on)
/a Lowercase letters (a to z for the first 26 pages,
aa to zz for the next 26, and so on)
prefix: The label prefix for page labels in this range.
start: The value of the numeric portion for the first page label
in the range.
Subsequent pages are numbered sequentially from this value,
which must be greater than or equal to 1. Default value: 1.
"""
default_page_label = DictionaryObject()
default_page_label[NameObject("/S")] = NameObject("/D")
new_page_label = DictionaryObject()
if style is not None:
new_page_label[NameObject("/S")] = NameObject(style)
if prefix is not None:
new_page_label[NameObject("/P")] = TextStringObject(prefix)
if start != 0:
new_page_label[NameObject("/St")] = NumberObject(start)
if NameObject(CatalogDictionary.PAGE_LABELS) not in self._root_object:
nums = ArrayObject()
nums_insert(NumberObject(0), default_page_label, nums)
page_labels = TreeObject()
page_labels[NameObject("/Nums")] = nums
self._root_object[NameObject(CatalogDictionary.PAGE_LABELS)] = page_labels
page_labels = cast(
TreeObject, self._root_object[NameObject(CatalogDictionary.PAGE_LABELS)]
)
nums = cast(ArrayObject, page_labels[NameObject("/Nums")])
nums_insert(NumberObject(page_index_from), new_page_label, nums)
nums_clear_range(NumberObject(page_index_from), page_index_to, nums)
next_label_pos, *_ = nums_next(NumberObject(page_index_from), nums)
if next_label_pos != page_index_to + 1 and page_index_to + 1 < len(self.pages):
nums_insert(NumberObject(page_index_to + 1), default_page_label, nums)
page_labels[NameObject("/Nums")] = nums
self._root_object[NameObject(CatalogDictionary.PAGE_LABELS)] = page_labels
def _pdf_objectify(obj: Union[Dict[str, Any], str, int, List[Any]]) -> PdfObject:
if isinstance(obj, PdfObject):
return obj
if isinstance(obj, dict):
to_add = DictionaryObject()
for key, value in obj.items():
name_key = NameObject(key)
cast_value = _pdf_objectify(value)
to_add[name_key] = cast_value
return to_add
elif isinstance(obj, str):
if obj.startswith("/"):
return NameObject(obj)
else:
return TextStringObject(obj)
elif isinstance(obj, (int, float)):
return FloatObject(obj)
elif isinstance(obj, list):
return ArrayObject(_pdf_objectify(i) for i in obj)
else:
raise NotImplementedError(
f"{type(obj)=} could not be cast to a PdfObject"
)
def _create_outline_item(
action_ref: Union[None, IndirectObject],
title: str,
color: Union[Tuple[float, float, float], str, None],
italic: bool,
bold: bool,
) -> TreeObject:
outline_item = TreeObject()
if action_ref is not None:
outline_item[NameObject("/A")] = action_ref
outline_item.update(
{
NameObject("/Title"): create_string_object(title),
}
)
if color:
if isinstance(color, str):
color = hex_to_rgb(color)
outline_item.update(
{NameObject("/C"): ArrayObject([FloatObject(c) for c in color])}
)
if italic or bold:
format_flag = 0
if italic:
format_flag += 1
if bold:
format_flag += 2
outline_item.update({NameObject("/F"): NumberObject(format_flag)})
return outline_item
def generate_appearance_stream(
txt: str,
sel: List[str],
da: str,
font_full_rev: Dict[str, bytes],
rct: RectangleObject,
font_height: float,
y_offset: float,
) -> bytes:
ap_stream = f"q\n/Tx BMC \nq\n1 1 {rct.width - 1} {rct.height - 1} re\nW\nBT\n{da}\n".encode()
for line_number, line in enumerate(txt.replace("\n", "\r").split("\r")):
if line in sel:
# may be improved but cannot find how to get fill working => replaced with lined box
ap_stream += (
f"1 {y_offset - (line_number * font_height * 1.4) - 1} {rct.width - 2} {font_height + 2} re\n"
f"0.5 0.5 0.5 rg s\n{da}\n"
).encode()
if line_number == 0:
ap_stream += f"2 {y_offset} Td\n".encode()
else:
# Td is a relative translation
ap_stream += f"0 {- font_height * 1.4} Td\n".encode()
enc_line: List[bytes] = [
font_full_rev.get(c, c.encode("utf-16-be")) for c in line
]
if any(len(c) >= 2 for c in enc_line):
ap_stream += b"<" + (b"".join(enc_line)).hex().encode() + b"> Tj\n"
else:
ap_stream += b"(" + b"".join(enc_line) + b") Tj\n"
ap_stream += b"ET\nQ\nEMC\nQ\n"
return ap_stream
Reference in New Issue View Git Blame Copy Permalink