U Nf_@s@ddlmZmZmZddlmZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZddlmZddlmZddlmZdd lmZmZmZdd lmZdd lmZm Z m!Z!dd lm"Z#dd l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1ddl2m3Z3m4Z4ddlm5Z5m6Z6m7Z7dZ8GdddZ9GdddZ:dS))BytesIOFileIOIOBase)Path) TracebackType) AnyDictIterableListOptionalTupleTypeUnioncast) Encryption) PageObject) PdfReader) StrByteTypedeprecate_with_replacementstr_) PdfWriter)GoToActionArguments TypArgumentsTypFitArguments)PagesAttributes) PAGE_FIT ArrayObject DestinationDictionaryObjectFit FloatObjectIndirectObject NameObject NullObject NumberObject OutlineItemTextStringObject TreeObject) PageRange PageRangeSpec) LayoutType OutlineType PagemodeTypez=close() was called and thus the writer cannot be used anymorec@s$eZdZdZeeeddddZdS) _MergedPagez@Collect necessary information on each page that is being merged.N)pagedatasrcidreturncCs||_||_d|_||_dSN)r0r/ out_pagedatar1)selfr/r0r1r61/tmp/pip-unpacked-wheel-zevpxvmc/pypdf/_merger.py__init__Nsz_MergedPage.__init__)__name__ __module__ __qualname____doc__rrintr8r6r6r6r7r.Ksr.c @seZdZdZdEeeeefddddZddd d Z e e e e e e e dd d d ZdFeeeeefe ee eeddddZeeeefeee efdddZdGeeeefe eedeeeefeeeefeefeddddZeeefddddZddddZeeefddddZe ddd d!Z!e"dd"d#d$Z#e$e e"dd%d&Z%e%j&e"dd"d'd&Z%eeeeeeffeeeefeeeefeefeeeefd(d)d*Z'ee(eeeefeeeefeefe(d+d,d-Z)ddd.d/Z*dHe e+e,e e-dd0d1d2Z.ee,e/fe0dd3d4d5Z1ee0dd6d7d8Z2dIee0e e+e,dd9d:d;Z3dJeeefe e(e eed<d=d>Z4dddde5feeede-e6fe ee7e7e7feee8e6d?d@dAZ9eeddBdCdDZ:dS)K PdfMergerzD Use :class:`PdfWriter` instead. .. deprecated:: 5.0.0 FN)strictfileobjr2cCsBtdddg|_g|_t|_g|_g|_d|_||_||_ dS)Nr>r5.0.0r) rinputspagesroutputoutline named_destsid_countrAr@)r5r@rAr6r6r7r8\s zPdfMerger.__init__)r2cCstddd|S)Nr>rrB)rr5r6r6r7 __enter__is zPdfMerger.__enter__)exc_typeexc tracebackr2cCs|jr||j|dS)z*Write to the fileobj and close the merger.N)rAwriteclose)r5rKrLrMr6r6r7__exit__ns zPdfMerger.__exit__T) page_numberrA outline_itemrDimport_outliner2cCsx||\}}t||jd}|j||f|dk r:||_|dkrRdt|jf}n:t|t rn| t|j}nt|t rznt|t st dg} g} |r|j} ||| |} |rtt|t|jt} |j| | g7_n|j| 7_|j} ||| |} |j| 7_t|D]:}|j|}|j}|jd7_t|||}| |q|| || | |j||<dS)a Merge the pages from the given file into the output file at the specified page number. Args: page_number: 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 an outline item (previously referred to as a 'bookmark') to be applied at the beginning of the included file by supplying the text of the outline item. pages: can be a :class:`PageRange` or a ``(start, stop[, step])`` tuple to merge only the specified range of pages from the source document into the output document. Can also be a list of pages to merge. 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``. )r@Nrz0"pages" must be a tuple of (start, stop[, step])r)_create_streamrr@rCappend _encryptionlenrD isinstancer)indiceslisttuple TypeErrorrF _trim_outliner&r'r%rHr fitZnamed_destinations _trim_destsrGranger._associate_dests_to_pages!_associate_outline_items_to_pages)r5rQrArRrDrSstreamencryption_objreaderZsrcpagesrFZoutline_item_typdestsZ trimmed_destsipager1mpr6r6r7mergeysL        zPdfMerger.merge)rAr2cCsd}t|ttfrt|d}nt|trf|jr4|j}|j}|jdt |j }|j|n8t |drt |dr|d| }t |}nt d||fS)NrbrseekreadzPdfMerger.merge 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.) rXstrrrrrVrctellrlrrmhasattrNotImplementedError)r5rArdrcZ orig_tellZ file_contentr6r6r7rTs$       zPdfMerger._create_stream)rArRrDrSr2cCs|t|j||||dS)a Identical to the :meth:`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 an outline item (previously referred to as a 'bookmark') to be applied at the beginning of the included file by supplying the text of the outline item. pages: can be a :class:`PageRange` or a ``(start, stop[, step])`` tuple to merge only the specified range of pages from the source document into the output document. Can also be a list of pages to append. 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``. N)rjrWrD)r5rArRrDrSr6r6r7rUszPdfMerger.appendcCs|jdkrtt|jD]H}|j|jtttt f|jj }|j |t jd |_q|||j|\}}|r|dS)z Write all data that has been merged to the given output file. Args: fileobj: Output file. Can be a filename or any kind of file-like object. N)rE RuntimeErrorERR_CLOSED_WRITERrDZadd_pager/rrrnrZ_pages get_objectZ get_referencePAZKIDSr4 _write_dests_write_outlinerNrO)r5rArhZ pages_objZmy_fileZ ret_fileobjr6r6r7rNs  zPdfMerger.writecCs.g|_|jD]\}}|q g|_d|_dS)zHShut all file descriptors (input and output) and clear all memory usage.N)rDrCrOrE)r5Zfile_descriptor_readerr6r6r7rO2s  zPdfMerger.close)infosr2cCs"|jdkrtt|j|dS)z Add custom metadata to the output. Args: infos: a Python dictionary where each key is a field and each value is your new metadata. An example is ``{'/Title': 'My title'}`` N)rErsrt add_metadata)r5rzr6r6r7r{;s zPdfMerger.add_metadata)layoutr2cCs"|jdkrtt|j|dS)a 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 N)rErsrtZ_set_page_layout)r5r|r6r6r7set_page_layoutHs zPdfMerger.set_page_layout)moder2cCs ||_dSaW Set the page mode. Args: mode: The page mode to use. .. list-table:: Valid ``mode`` arguments :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 N) page_moder5r~r6r6r7 set_page_modeeszPdfMerger.set_page_modecCs|jdkrtt|jjSrrErsrtrrIr6r6r7r~s zPdfMerger.page_modecCs|jdkrtt||j_dSr3rrr6r6r7rs )pdfrfrDr2c Csg}t|tr|n tt|}|D]f\}}|D]X}|j||dkr2|d|td<t|t|dks|t| |q&q2q&|S)z Remove named destinations that are not a part of the specified page set. Args: pdf: dests: pages: /Page/Title) rXrZr`itemsrDrur#rAssertionErrorrU) r5rrfrDZ new_destslstkeyobjjr6r6r7r_s zPdfMerger._trim_dests)rrFrDr2c Csg}d}t|tr|n tt|}t|D]\}}t|trp||||} | r|sd|||d|| q*d}|D]R} |ddkrqx|j| |dkrx|d|td<||d}q*qxq*|S)z Remove outline item entries that are not a part of the specified page set. Args: pdf: outline: pages: Returns: An outline type TrFrN) rXrZr` enumerater]rUrDrur#) r5rrFrDZ new_outlineZprev_header_addedrrgrRsubrr6r6r7r]s,    zPdfMerger._trim_outlinecCsv|jdkrtt|jD]X}d}d|kr\t|jD](\}}|j|dkr2|j|td<q\q2|dk r|j |qdS)Nr) rErsrtrGrrDr1r4r#Zadd_named_destination_object)r5 named_dest page_indexrhr6r6r7rws  zPdfMerger._write_dests)rFparentr2cCs|jdkrtt|dkr |j}|dk s0tdd}|D]|}t|trT|||q8d}d|krt|j D]&\}}|j |dkrj| ||qqj|dk r8|d=|d=|j ||}q8dS)N hint for mypyr/Type) rErsrtrFrrXrZrxrrDr1_write_outline_item_on_pageZadd_outline_item_dict)r5rFrZ last_addedrRZpage_norhr6r6r7rxs&     zPdfMerger._write_outline)rRrhr2cCstt|d}t|jt|g}tjtjftj tjftj tj ftj tj ftj tj tjdftjtj tjtjtjfi}||dD]B}||krt||ts|t||n|td||=q~tttjtdttjt|i|td<dS)Nrz/Zoomr6rz/GoToz/A)rrnr%r1r#rZFIT_HrZTOPZFIT_BHZFIT_VLEFTZFIT_BVZXYZZFIT_RZBOTTOMRIGHTgetrXr$rUr!rrSDr)r5rRrhZoi_typeargsZ fit2arg_keysZarg_keyr6r6r7r sD   z%PdfMerger._write_outline_item_on_page)rDr2cCsv|jD]j}d}|d}t|tr"q|D]}||jkr&|j}q&|dkr`td|ddt||td<qdS)NrzUnresolved named destination 'r')rGrXr%rur/r1 ValueErrorr#)r5rDrrnprhr6r6r7ra-s  z#PdfMerger._associate_dests_to_pages)rDrFr2cCs|dkr|j}|dk std|D]n}t|tr>|||q"d}|d}t|trVq"|D]}||jkrZ|j}qZ|dk r"t||t d<q"dS)Nrr) rFrrXrZrbr%rur/r1r#)r5rDrFrRrZoutline_item_pagepr6r6r7rb?s    z+PdfMerger._associate_outline_items_to_pages)rRrootr2cCsz|dkr|j}t|D]^\}}t|trH|||}|rt|g|Sq||ksjttttf|d|kr|gSqdS)Nr)rFrrXrZfind_outline_itemrrr)r5rRrrgZoi_enumresr6r6r7rWs   zPdfMerger.find_outline_item)titlerQrcolorbolditalicr^r2c Cs.|j}|dkrtt||||d||||S)aQ Add an outline item (commonly referred to as a "Bookmark") to this 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. color: Color of the outline item's font as a red, green, blue tuple from 0.0 to 1.0 bold: Outline item font is bold italic: Outline item font is italic fit: The fit of the destination page. N)rErsrtadd_outline_item) r5rrQrrrrr^writerr6r6r7roszPdfMerger.add_outline_item)rrQr2cCs,tt|t|tjdd}|j|dS)z Add a destination to the output. Args: title: Title to use page_number: Page number this destination points at. i:)topN)rr'r%r Zfit_horizontallyrGrU)r5rrQdestr6r6r7add_named_destinations  zPdfMerger.add_named_destination)Fr?)NNT)NNT)NN)N)N);r9r:r;r<boolrrrr8rJr r BaseExceptionrrPr=rrnr*rjr rrrTr)r rUrNrOrrr{r+r}r-rpropertyrsetterr_r,r]rwr r&r(rxrr.rrarbrrr"floatr rrr6r6r6r7r>Us      X  & "   " " )    "      (r>N);iorrrpathlibrtypesrtypingrrr r r r r rrrVrZ_pagerryr_utilsrrrZ_writerr constantsrrrrrvZgenericrrrrr r!r"r#r$r%r&r'r(Z pageranger)r*r+r,r-rtr.r>r6r6r6r7s   ,     <