U dS@sddlTddlmZmZddlmZmZddlmZddl m Z e dkrZddl m Z e Z ndd lmZdd lmZeZ Gd d d eZGd ddeZGdddeZdS))*)isStringstr_) PdfFileReader PdfFileWriter) PageRange) version_info)r)StringIO)BytesIO)FileIOc@seZdZdZddZdS) _MergedPagez _MergedPage is used internally by PdfFileMerger to collect necessary information on each page that is being merged. cCs||_||_d|_||_dSN)srcpagedata out_pagedataid)selfrrrr1/tmp/pip-unpacked-wheel-eeeohlmn/PyPDF3/merger.py__init__1sz_MergedPage.__init__N)__name__ __module__ __qualname____doc__rrrrrr,src@seZdZdZd&ddZd'ddZd(dd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZd)ddZddZd*ddZd+d d!Zd,d"d#Zd$d%ZdS)- PdfFileMergera Initializes a PdfFileMerger object. PdfFileMerger 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 ``True``. TcCs0g|_g|_t|_g|_g|_d|_||_dS)Nr)inputspagesroutput bookmarks named_destsid_countstrict)rr#rrrrFszPdfFileMerger.__init__NcCsd}d}t|r t|d}d}nt|drTt|drT|d|}t|}d}nRt|tr|j } |jdt|j}|j| |}t|dr|j }d}t||j d } |dk r|| _ |dkrd| f}n.t|t r|| }nt|tstd g} |r*tt|t|jtd }g} |rJ| } || | |} |rd|j|| g7_n|j| 7_| j} || | |} |j| 7_t|D]:}| |}|j}|jd 7_t|| |}| |q| | |!| | |j"||<|j#|| |fdS) a Merges 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 :ref:`Page Range ` 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``. FNrbTseekreadr_decryption_key)r#z0"pages" must be a tuple of (start, stop[, step])z/Fitr)$rfilehasattrr%r&StreamIO isinstancerstreamtellr'r#Z getNumPagesrindicestuple TypeErrorBookmarkTextStringObject NumberObjectr" NameObjectZ getOutlines _trim_outliner ZnamedDestinations _trim_destsr!rangegetPagerappend_associate_dests_to_pages_associate_bookmarks_to_pagesrr)rpositionfileobjbookmarkrimport_bookmarksmy_fileZdecryption_keyZ filecontentZ orig_tellpdfrZsrcpagesoutlinedestsiZpgrmprrrmergeOsf              zPdfFileMerger.mergecCs|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 :ref:`Page Range ` 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)rFlenr)rr=r>rr?rrrr9szPdfFileMerger.appendcCsd}t|rt|d}d}|jD]4}|j|j|j|jjdd|_ q | | |j ||r~| dS)z Writes 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. FwbT/KidsN)rr(rrZaddPagerZ getReference_pages getObjectr _write_dests_write_bookmarkswriteclose)rr=r@pagerrrrOs  $ zPdfFileMerger.writecCs4g|_|jD]\}}}|r |q g|_d|_dS)zd Shuts all file descriptors (input and output) and clears all memory usage. N)rrrPr)rforAZminerrrrPs  zPdfFileMerger.closecCs|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 addMetadata)rZinfosrrrrSszPdfFileMerger.addMetadatacCs|j|dS)a Set the page layout :param str layout: The page layout to be used Valid layouts are: /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 setPageLayout)rZlayoutrrrrTszPdfFileMerger.setPageLayoutcCs|j|dS)a Set the page mode. :param str mode: The page mode to use. Valid modes are: /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 setPageMode)rmoderrrrUszPdfFileMerger.setPageModec Csg}d}t|D]j\}}t|D]X}|||dkr$|d|td<t|t|dksnt||qq$q|S)zg Removes any named destinations that are not a part of the specified page set. T/Page/Title) listitemsr7r8rLr4rAssertionErrorr9) rpdfrCrZ new_destsprev_header_addedkojrrrr6#s  zPdfFileMerger._trim_destsc Csg}d}t|D]\}}t|trV||||}|r|sJ|||d||qd}t|D]D} || |dkrb|d|td<||d}qqbq|S)zm Removes any outline/bookmark entries that are not a part of the specified page set. TrFrW) enumerater+rYr5r9r7r8rLr4) rr\rBrZ new_outliner]rDr_subr`rrrr53s"    zPdfFileMerger._trim_outlinecCsv|j}|D]f}d}d}d|kr\t|jD]2\}}|j|dkr(|j|td<|}|j}q\q(|dkr |j|q dS)NrW) r!rarrrr4rrZaddNamedDestinationObject)rrCvpagenor\rDprrrrMKszPdfFileMerger._write_destsc Cs|dkr|j}d}|D]j}t|tr4|||qd}d}d|kr`t|jD] \}}|j|dkrPt|jt|dg} |ddks|ddkrd|krt|dt s| t |dn| t d|d=nN|ddks|ddkr8d |kr t|d t s | t |d n| t d|d =n|dd krd |krtt|d t st| t |d n| t dd|krt|dt s| t |dn| t dd |krt|d t s| t |d n| t d|d=|d =|d =n|dd kr&d |krLt|d t sL| t |d n| t dd |krt|d t s| t |d n| t dd|krt|dt s| t |dn| t dd|krt|dt s| t |dn| t d|d =|d=|d =|d=t tdtdtdt | i|td<|}|j}q`qP|dkr|d=|d=|j||}qdS)NrWz/Type/FitHz/FitBHz/Toprz/FitVz/FitBVz/Leftz/XYZz/Zoomz/FitRz/Bottomz/Right/S/GoTo/D/A)r r+rYrNrarrr3r4Z NullObjectr9Z FloatObjectDictionaryObject ArrayObjectrrZaddBookmarkDict) rr parentZ last_addedbrdr\rDreargsrrrrN[sn      ( zPdfFileMerger._write_bookmarkscCsv|jD]j}d}|d}t|tr"q|D]}||jkr&|j}q&|dkr^t||td<qtd|dfqdS)NrWz!Unresolved named destination '%s'rX)r!r+r3rLrrr4 ValueError)rrZndrdnprerrrr:s  z'PdfFileMerger._associate_dests_to_pagescCs|dkr|j}|D]}t|tr.|||qd}|d}t|trFq|D]}||jkrJ|j}qJ|dkrt||td<qt d|dfqdS)NrWzUnresolved bookmark '%s'rX) r r+rYr;r3rLrrr4rp)rrr rnrdZbprerrrr;s    z+PdfFileMerger._associate_bookmarks_to_pagescCsl|dkr|j}t|D]P\}}t|trH|||}|rf|g|Sq||ks\|d|kr|gSqdS)NrX)r rar+rY findBookmark)rr>rootrDrnresrrrrrs   zPdfFileMerger.findBookmarkc Cs|dkrt|jdg}nt|tr*|}n ||}tt|t|tdtd}|dkrh|j |nd|j}|ddD] }||}qz|dd}|t|krt||tr|| |n| ||g|S)a 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. Nrrf:rJ) rGr r+rYrrr1r2r3r4r9insert) rtitlepagenumrmZilocdestZbmparentrDZnposrrr addBookmarks     zPdfFileMerger.addBookmarkcCs.tt|t|tdtd}|j|dS)z Add a destination to the output. :param str title: Title to use :param int pagenum: Page number this destination points at. rfruN)Z Destinationr2r3r4r!r9)rrwrxryrrraddNamedDestinationsz!PdfFileMerger.addNamedDestination)T)NNT)NNT)NN)N)N)N)rrrrrrFr9rOrPrSrTrUr6r5rMrNr:r;rrrzr{rrrrr8s$ h    J   rc@s.eZdZd ddZddZddZdd ZdS) OutlinesObjectNcCs t|||_||_||_dSr)rYrtreer\rm)rr\r}rmrrrrs zOutlinesObject.__init__cCs||}||=|j|dSr)r} removeChild)rindexobjrrrremoveszOutlinesObject.removecCs|j|jjd|}t}|tdt|tdtdgtdtdi|j|}t }|td|tdt |i|j||j |dS) NrIrirfrurgrhrjrX) r\rLrKrkupdater4rlr3Z _addObjectZ TreeObjectZcreateStringObjectr}ZaddChild)rrwrxZpageRefactionZ actionRefr>rrradds$  zOutlinesObject.addcCs2dd|jDD]}|j||qdS)NcSsg|]}|qSrr).0xrrr 'sz,OutlinesObject.removeAll..)r}childrenr~pop)rchildrrr removeAll&s zOutlinesObject.removeAll)N)rrrrrrrrrrrr|s r|N)Zgenericutilsrrr\rrZ pagerangersysr cStringIOr r*ior r r(objectrrrYr|rrrrs       R