U Bb]@s<ddlmZmZmZddlmZmZmZmZm Z m Z m Z m Z ddl mZddlmZddlmZddlmZmZmZddlmZdd lmZdd lmZdd lmZmZdd lm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*dd l+m,Z,m-Z-ddl.m/Z/m0Z0m1Z1m2Z2m3Z3dZ4GdddZ5GdddZ6Gddde6Z7dS))BytesIOFileIOIOBase)AnyDictIterableListOptionalTupleUnioncast) Encryption) PageObject) PdfReader) StrByteTypedeprecate_with_replacementstr_) PdfWriter)GoToActionArguments)PagesAttributes) TypArgumentsTypFitArguments) ArrayObjectBookmark DestinationDictionaryObject FloatObjectIndirectObject NameObject NullObject NumberObjectTextStringObject TreeObject) PageRange PageRangeSpec)FitType LayoutType OutlinesType PagemodeType ZoomArgTypez=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|_||_dS)N)r-r, out_pagedatar.)selfr,r-r.r22/tmp/pip-unpacked-wheel-weujb6sz/PyPDF2/_merger.py__init__=sz_MergedPage.__init__)__name__ __module__ __qualname____doc__rrintr4r2r2r2r3r+:sr+c @seZdZdZdGeddddZdHeeee fe e e e eddd d Z eee feeee efd d d ZdIeee fe e edeeeefeeeeffeddddZedd ddZddddZee efddddZee efddddZeddddZeddddZedd d!d"Zedd d#d$Ze ee ee effeeeefeeeeffe ee efd%d&d'Z!e e"eeeefeeeeffe"d(d)d*Z#ddd+d,Z$dJe e%e&e e'dd-d.d/Z(ee&e)fe*dd0d1d2Z+e e*dd3d4d5Z,dKe e*e e%e&dd6d7d8Z-dLee efe e"e e ed9d:d;Z.dMe eede'e/fe ee0e0e0feee1e2e/d= d>d?Z3dNe eede'e/fe ee0e0e0feee1e2e/d= d@dAZ4e eddBdCdDZ5e eddBdEdFZ6dS)O PdfMergera Initialize a ``PdfMerger`` object. ``PdfMerger`` merges multiple PDFs into a single PDF. It can concatenate, slice, insert, or any combination of the above. See the functions :meth:`merge()` (or :meth:`append()`) and :meth:`write()` for usage information. :param bool strict: Determines whether user should be warned of all problems and also causes some correctable problems to be fatal. Defaults to ``False``. FN)strictr/cCs0g|_g|_t|_g|_g|_d|_||_dS)Nr)inputspagesroutput bookmarks named_destsid_countr;)r1r;r2r2r3r4SszPdfMerger.__init__T)positionfileobjbookmarkr=import_bookmarksr/cCsr||\}}}t||jd} |j|| |f|dk r>|| _|dkrVdt| jf}n.t|t rr| t| j}nt|t st dg} g} |r| j } || | |} |rtt|t|jttj} |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. :param int position: The *page number* to insert this file. File will be inserted after the given number. :param 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. :param str bookmark: Optionally, you may specify a bookmark to be applied at the beginning of the included file by supplying the text of the bookmark. :param 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. :param bool import_bookmarks: You may prevent the source document's bookmarks from being imported by specifying this as ``False``. )r;Nrz0"pages" must be a tuple of (start, stop[, step])r )_create_streamrr;r<append _encryptionlenr= isinstancer$indicestuple TypeErrorZoutlines _trim_outlinerr"r!rArrZFITr?Znamed_destinations _trim_destsr@ranger+_associate_dests_to_pages_associate_bookmarks_to_pages)r1rBrCrDr=rEstreammy_fileencryption_objreaderZsrcpagesoutlineZ bookmark_typdestsZ trimmed_destsipager.mpr2r2r3merge\sH       zPdfMerger.merge)rCr/cCsd}d}t|tr"t|d}d}nt|trn|jr8|j}|j}|jdt|j }|j|d}n8t |drt |dr|d| }t|}d}n|}|||fS)NFrbTrseekread) rJstrrrrHrStellr^rr_hasattr)r1rCrTrUrSZ orig_tellZ filecontentr2r2r3rFs(       zPdfMerger._create_stream)rCrDr=rEr/cCs|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. :param 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. :param str bookmark: Optionally, you may specify a bookmark to be applied at the beginning of the included file by supplying the text of the bookmark. :param 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. :param bool import_bookmarks: You may prevent the source document's bookmarks from being imported by specifying this as ``False``. N)r\rIr=)r1rCrDr=rEr2r2r3rGszPdfMerger.appendcCs|jdkrttd}t|tr.t|d}d}|jD]H}|j|jt t tt f|jj }|j|tjd |_q4|||j||r|dS)z Write all data that has been merged to the given output file. :param fileobj: Output file. Can be a filename or any kind of file-like object. NFwbT)r> RuntimeErrorERR_CLOSED_WRITERrJr`rr=Zadd_pager,r rrZ_pages get_objectZ get_referencePAZKIDSr0 _write_dests_write_bookmarkswriteclose)r1rCrTrZZ pages_objr2r2r3rks"     zPdfMerger.write)r/cCs4g|_|jD]\}}}|r |q g|_d|_dS)zHShut all file descriptors (input and output) and clear all memory usage.N)r=r<rlr>)r1fo_readerZminer2r2r3rls  zPdfMerger.close)infosr/cCs"|jdkrtt|j|dS)z Add custom metadata to the output. :param dict infos: a Python dictionary where each key is a field and each value is your new metadata. Example: ``{u'/Title': u'My title'}`` N)r>rerf add_metadatar1ror2r2r3rp$s zPdfMerger.add_metadatacCstdd||dS)zW .. deprecated:: 1.28.0 Use :meth:`add_metadata` instead. addMetadatarpN)rrprqr2r2r3rr0s zPdfMerger.addMetadata)layoutr/cCstdd||dS)zZ .. deprecated:: 1.28.0 Use :meth:`set_page_layout` instead. setPageLayoutset_page_layoutN)rrur1rsr2r2r3rt9s zPdfMerger.setPageLayoutcCs"|jdkrtt|j|dS)a Set the page layout. :param str 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)r>rerfZ_set_page_layoutrvr2r2r3ruBs zPdfMerger.set_page_layout)moder/cCstdd||dS)zX .. deprecated:: 1.28.0 Use :meth:`set_page_mode` instead. setPageMode set_page_modeN)rryr1rwr2r2r3rx^s zPdfMerger.setPageModecCs"|jdkrtt|j|dS)aR Set the page mode. :param str mode: The page mode to use. .. list-table:: Valid ``mode`` arguments :widths: 50 200 * - /UseNone - Do not show outlines or thumbnails panels * - /UseOutlines - Show outlines (aka bookmarks) panel * - /UseThumbs - Show page thumbnails panel * - /FullScreen - Fullscreen view * - /UseOC - Show Optional Content Group (OCG) panel * - /UseAttachments - Show attachments panel N)r>rerfryrzr2r2r3rygs zPdfMerger.set_page_mode)pdfrXr=r/cCs|g}|D]j\}}t|D]X}|j||dkr|d|td<t|t|dksft||q qq |S)zHRemove named destinations that are not a part of the specified page set./Page/Title)itemsrPr=rgrrAssertionErrorrG)r1r{rXr=Z new_destskeyobjjr2r2r3rOs  zPdfMerger._trim_dests)r{rWr=r/c Csg}d}t|D]\}}t|trV||||}|r|sJ|||d||qd}t|D]R} |ddkrtqb|j| |dkrb|d|td<||d}qqbq|S)zNRemove outline/bookmark entries that are not a part of the specified page set.Tr Fr|N) enumeraterJlistrNrGrPr=rgr) r1r{rWr=Z new_outlineZprev_header_addedrYosubrr2r2r3rNs&     zPdfMerger._trim_outlinecCsv|jdkrtt|jD]X}d}d|kr\t|jD](\}}|j|dkr2|j|td<q\q2|dk r|j |qdS)Nr|) r>rerfr@rr=r.r0rZadd_named_destination_object)r1Z named_destpagenorZr2r2r3ris  zPdfMerger._write_dests)r?parentr/cCs|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) r>rerfr?rrJrrjrr=r._write_bookmark_on_pageZadd_bookmark_dict)r1r?rZ last_addedrDZpage_norZr2r2r3rjs&     zPdfMerger._write_bookmarks)rDrZr/cCstt|d}t|jt|g}tjtjftj tjftj tj ftj tj ftj tj tjdftjtj tjtjtjfi}||tD]B}||krt||ts|t||n|td||=qtttjtdttjt|i|td<dS)Nrz/Zoomrz/GoToz/A)r r`r!r.rrFIT_HrZTOPZFIT_BHZFIT_VLEFTZFIT_BVZXYZZFIT_RZBOTTOMRIGHTgetrLrJr rGrrrSDr)r1rDrZZbm_typeargsZ fit2arg_keysZarg_keyr2r2r3rs>  z!PdfMerger._write_bookmark_on_page)r=r/cCsx|jD]l}d}|d}t|tr"q|D]}||jkr&|j}q&|dk r^t||td<qtd|ddqdS)Nr|zUnresolved named destination 'r}')r@rJr!rgr,r.r ValueError)r1r=Zndrnppr2r2r3rQs  z#PdfMerger._associate_dests_to_pages)r=r?r/cCs|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|) r?rrJrrRr!rgr,r.r)r1r=r?brZbprr2r2r3rRs    z'PdfMerger._associate_bookmarks_to_pages)rDrootr/cCsl|dkr|j}t|D]P\}}t|trH|||}|rf|g|Sq||ks\|d|kr|gSqdS)Nr})r?rrJr find_bookmark)r1rDrrYrresr2r2r3r s   zPdfMerger.find_bookmark/Fit) titlepagenumrcolorbolditalicfitrr/c Gs&tdd|j|||||||f|S)zV .. deprecated:: 1.28.0 Use :meth:`add_bookmark` instead. addBookmark add_bookmark)rr) r1rrrrrrrrr2r2r3r5s zPdfMerger.addBookmarkc Gs2|j} | dkrtt| j|||||||f|S)aG Add a bookmark to this PDF file. :param str title: Title to use for this bookmark. :param int pagenum: Page number this bookmark will point to. :param parent: A reference to a parent bookmark to create nested bookmarks. :param tuple color: Color of the bookmark as a red, green, blue tuple from 0.0 to 1.0 :param bool bold: Bookmark is bold :param bool italic: Bookmark is italic :param str fit: The fit of the destination page. See :meth:`addLink()` for details. N)r>rerfr) r1rrrrrrrrwriterr2r2r3rIszPdfMerger.add_bookmark)rrr/cCstdd|||S)z_ .. deprecated:: 1.28.0 Use :meth:`add_named_destination` instead. addNamedDestinationadd_named_destination)rr)r1rrr2r2r3ris zPdfMerger.addNamedDestinationcCs0tt|t|ttjtd}|j|dS)z Add a destination to the output. :param str title: Title to use :param int pagenum: Page number this destination points at. i:N)rr"r!rrrr@rG)r1rrdestr2r2r3rqszPdfMerger.add_named_destination)F)NNT)NNT)NN)N)N)NNFFr)NNFFr)7r5r6r7r8boolr4r9r rrr r`r%r\r rrrFr$rGrkrlrrrprrr'rtrur)rxryrrOr(rNrirrr#rjrr+rrQrRrrfloatr&r*rrrrr2r2r2r3r:Ds   W  )   $                   r:cs&eZdZeeddfdd ZZS) PdfFileMergerN)rkwargsr/cs8tddd|kr&t|dkr&d|d<tj||dS)NrZPdfMerger;r T)rrIsuperr4)r1rr __class__r2r3r4s zPdfFileMerger.__init__)r5r6r7rr4 __classcell__r2r2rr3rsrN)8iorrrtypingrrrrr r r r rHrZ_pagerrnr_utilsrrrZ_writerr constantsrrrhrrZgenericrrrrrrrr r!r"r#Z pageranger$r%typesr&r'r(r)r*rfr+r:rr2r2r2r3s((      4  A