U 3gw@slddlZddlmZmZmZddlmZddlmZddl m Z m Z m Z m Z mZmZmZmZmZddlmZddlmZdd lmZdd lmZmZmZmZdd lmZdd l m!Z!dd l m"Z#ddl m$Z$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3ddl4m5Z5m6Z6ddlm7Z7m8Z8m9Z9m:Z:m;Z;dZGddde>Z?dS)N)BytesIOFileIOIOBase)Path) TracebackType) AnyDictIterableListOptionalTupleTypeUnioncast) Encryption) PageObject) PdfReader) StrByteTypedeprecation_bookmarkdeprecation_with_replacementstr_) PdfWriter)GoToActionArguments)PagesAttributes) TypArgumentsTypFitArguments) PAGE_FIT ArrayObject DestinationDictionaryObjectFit FloatObjectIndirectObject NameObject NullObject NumberObject OutlineItemTextStringObject TreeObject) PageRange PageRangeSpec)FitType LayoutType OutlineType 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|_||_dSN)r3r2 out_pagedatar4)selfr2r3r4r92/tmp/pip-unpacked-wheel-ifm06wwb/PyPDF2/_merger.py__init__Qsz_MergedPage.__init__)__name__ __module__ __qualname____doc__rrintr;r9r9r9r:r1Nsr1c @s,eZdZdZedddYeeeefdddd Z dd d d Z e e e e e e edd ddZeddddZe eeeeefe ee eee eddddZeeeefeee efdddZedddd[eeeefe eedeeeefeeeefeefeddddZeeefddddZdd dd Zeeefdd!d"d#Z eeefdd!d$d%Z!e"dd&d'd(Z#e"dd&d)d*Z$e%dd+d,d-Z&e%dd+d.d/Z'eeeeeeffeeeefeeeefeefeeeefd0d1d2Z(ee)eeeefeeeefeefe)d3d4d5Z*dd d6d7Z+eddd\e e,e-e e.dd8d9d:Z/edd;ee-e0fe1dd<d=d>Z2ee1dd?d@dAZ3eddd]ee1e e,e-ddBdCdDZ4edd;d^eeefe e)e eedEdFdGZ5edd;d_eeefe e)e eedEdHdIZ6ddddde7dfee eede.e8fe ee9e9e9feee:e ee8dJ dKdLZ;d`eeede.e8fe ee9e9e9feeedaeeede.e8fe ee9e9e9feee` (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``. :param fileobj: Output file. Can be a filename or any kind of file-like object. outline)Z bookmarksFN)strictfileobjr5cCs6g|_g|_t|_g|_g|_d|_||_||_dS)Nr) inputspagesroutputrB named_destsid_countrErD)r8rDrEr9r9r:r;iszPdfMerger.__init__)r5cCs|Sr6r9)r8r9r9r: __enter__vszPdfMerger.__enter__)exc_typeexc tracebackr5cCs|jr||j|dS)z*Write to the fileobj and close the merger.N)rEwriteclose)r8rLrMrNr9r9r:__exit__zs zPdfMerger.__exit__ outline_itemimport_outline)bookmarkZimport_bookmarksT) page_numberrErRrGrSpositionr5cCs|dk r>|dkr6|}d}d}t|d|dtntd|dkrNtd|dkr^td||\} } t| |jd } |j| | f| dk r| | _ |dkrd t | j f}n:t |t r|t | j }nt |trnt |tstd g} g} |r | j} || | |} |r@tt|t|jt}|j|| g7_n|j| 7_| j}|| ||}|j|7_t|D]:}| j |}|j}|jd 7_t|| |}| |qx|| | | | |j ||<dS) a Merge the pages from the given file into the output file at the specified page number. :param int page_number: 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 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. :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. Can also be a list of pages to merge. :param bool 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``. NrVrUzG is deprecated as an argument and will be removed in PyPDF2=4.0.0. Use  insteadzCThe argument position of merge is deprecated. Use page_number only.page_number may not be Nonezfileobj may not be None)rDrz0"pages" must be a tuple of (start, stop[, step])r)!warningswarnDeprecationWarning ValueError_create_streamrrDrFappend _encryptionlenrG isinstancer*indiceslisttuple TypeErrorrB _trim_outliner'r(r&rJr!fitZnamed_destinations _trim_destsrIranger1_associate_dests_to_pages!_associate_outline_items_to_pages)r8rUrErRrGrSrVold_termnew_termstreamencryption_objreaderZsrcpagesrBZoutline_item_typdestsZ trimmed_destsipager4mpr9r9r:mergesl#        zPdfMerger.merge)rEr5cCsd}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.) rastrrrrr_rntellrwrrxhasattrNotImplementedError)r8rErornZ orig_tellZ filecontentr9r9r:r]s$       zPdfMerger._create_stream)rErRrGrSr5cCs|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 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. :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. Can also be a list of pages to append. :param bool 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)rur`rG)r8rErRrGrSr9r9r:r^s!zPdfMerger.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. :param fileobj: Output file. Can be a filename or any kind of file-like object. N)rH RuntimeErrorERR_CLOSED_WRITERrGZadd_pager2rrryrZ_pages get_objectZ get_referencePAZKIDSr7 _write_dests_write_outlinerOrP)r8rErsZ pages_objZmy_fileZ ret_fileobjr9r9r:rOAs  zPdfMerger.writecCs.g|_|jD]\}}|q g|_d|_dS)zHShut all file descriptors (input and output) and clear all memory usage.N)rGrFrPrH)r8fo_readerr9r9r:rPas  zPdfMerger.close)infosr5cCs"|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)rHr~r add_metadatar8rr9r9r:rjs zPdfMerger.add_metadatacCstdd||dS)zW .. deprecated:: 1.28.0 Use :meth:`add_metadata` instead. addMetadatarN)rrrr9r9r:rvs zPdfMerger.addMetadata)layoutr5cCstdd||dS)zZ .. deprecated:: 1.28.0 Use :meth:`set_page_layout` instead. setPageLayoutset_page_layoutN)rrr8rr9r9r:rs 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)rHr~rZ_set_page_layoutrr9r9r:rs zPdfMerger.set_page_layout)moder5cCstddd||dS)zX .. deprecated:: 1.28.0 Use :meth:`set_page_mode` instead. setPageMode set_page_mode3.0.0N)rrr8rr9r9r:rs zPdfMerger.setPageModecCs"|jdkrtt|j|dS)aP Set the page mode. :param str 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)rHr~rrrr9r9r:rs zPdfMerger.set_page_mode)pdfrqrGr5c Csg}t|tr|n tt|}|D]f\}}|D]X}|j||dkr2|d|td<t|t|dks|t| |q&q2q&|S)zHRemove named destinations that are not a part of the specified page set./Page/Title) rarcriitemsrGrr$rAssertionErrorr^) r8rrqrGZ new_destslstkeyobjjr9r9r:rhs zPdfMerger._trim_dests)rrBrGr5c 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)zJRemove outline item entries that are not a part of the specified page set.TrFrN) rarcri enumeraterfr^rGrr$) r8rrBrGZ new_outlineZprev_header_addedrrrrRsubrr9r9r:rfs,    zPdfMerger._trim_outlinecCsv|jdkrtt|jD]X}d}d|kr\t|jD](\}}|j|dkr2|j|td<q\q2|dk r|j |qdS)Nr) rHr~rrIrrGr4r7r$Zadd_named_destination_object)r8 named_destpagenorsr9r9r:rs  zPdfMerger._write_dests)rBparentr5cCs|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) rHr~rrBrrarcrrrGr4_write_outline_item_on_pageZadd_outline_item_dict)r8rBrZ last_addedrRZpage_norsr9r9r:rs&     zPdfMerger._write_outline)rT)rRrsr5cCstt|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)rryr&r4r$rZFIT_HrZTOPZFIT_BHZFIT_VLEFTZFIT_BVZXYZZFIT_RZBOTTOMRIGHTgetrdrar%r^r"r rSDr)r8rRrsZoi_typeargsZ fit2arg_keysZarg_keyr9r9r:r"sD   z%PdfMerger._write_outline_item_on_page)rGr5cCsv|jD]j}d}|d}t|tr"q|D]}||jkr&|j}q&|dkr`td|ddt||td<qdS)NrzUnresolved named destination 'r')rIrar&rr2r4r\r$)r8rGrrnprsr9r9r:rjEs  z#PdfMerger._associate_dests_to_pages)rGrBr5cCs|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) rBrrarcrkr&rr2r4r$)r8rGrBrRrZoutline_item_pagepr9r9r:rkWs    z+PdfMerger._associate_outline_items_to_pages)rRrootr5cCsz|dkr|j}t|D]^\}}t|trH|||}|rt|g|Sq||ksjttttf|d|kr|gSqdS)Nr)rBrrarcfind_outline_itemrrr)r8rRrrrZoi_enumresr9r9r:rps   zPdfMerger.find_outline_itemcCs |||S)zZ .. deprecated:: 2.9.0 Use :meth:`find_outline_item` instead. )r)r8rRrr9r9r: find_bookmarks zPdfMerger.find_bookmark) titlerUrcolorbolditalicrgpagenumr5c Cs|dk r|dk rtd|dk rDd} d} t| d| dt|}|dkrTtd|j} | dkrjtt| |||d||||S)al Add an outline item (commonly referred to as a "Bookmark") to this PDF file. :param str title: Title to use for this outline item. :param int page_number: Page number this outline item will point to. :param parent: A reference to a parent outline item to create nested outline items. :param tuple color: Color of the outline item's font as a red, green, blue tuple from 0.0 to 1.0 :param bool bold: Outline item font is bold :param bool italic: Outline item font is italic :param Fit fit: The fit of the destination page. NzMThe argument pagenum of add_outline_item is deprecated. Use page_number only.rrUH is deprecated as an argument and will be removed in PyPDF2==4.0.0. Use rWrX)r\rYrZr[rHr~radd_outline_item) r8rrUrrrrrgrrlrmwriterr9r9r:rs6zPdfMerger.add_outline_item/Fit) rrrrrrrgrr5c Gs*tddd|||||||t||dS)zZ .. deprecated:: 1.28.0 Use :meth:`add_outline_item` instead. addBookmarkrrZfit_typeZfit_argsrrr! r8rrrrrrrgrr9r9r:rs  zPdfMerger.addBookmarkc Gs*tddd|||||||t||dS)zY .. deprecated:: 2.9.0 Use :meth:`add_outline_item` instead. rrrrrrr9r9r: add_bookmarks  zPdfMerger.add_bookmark)rrr5cCstddd|||S)z_ .. deprecated:: 1.28.0 Use :meth:`add_named_destination` instead. addNamedDestinationadd_named_destinationr)rr)r8rrr9r9r:rs zPdfMerger.addNamedDestination)rrUrr5cCs|dk r|dk rtd|dk rDd}d}t|d|dt|}|dkrTtdtt|t|tjdd }|j |dS) z Add a destination to the output. :param str title: Title to use :param int page_number: Page number this destination points at. NzRThe argument pagenum of add_named_destination is deprecated. Use page_number only.rrUrrWrXi:)top) r\rYrZr[rr(r&r!Zfit_horizontallyrIr^)r8rrUrrlrmdestr9r9r:r s(  zPdfMerger.add_named_destination)FrC)NNNNTN)NNT)NN)N)N)N)NNFFr)NNFFr)NN)Br<r=r>r?rboolrrrr;rKr r BaseExceptionrrQr@rryr+rur rrr]r*r r^rOrPrrrrr-rrr/rrrhr.rfrr r'r)rrr1rrjrkrrrr#floatr!rr,r0rrrrr9r9r9r:rAXs@     v  #  ""      " "   "       ;     rAcs&eZdZeeddfdd ZZS) PdfFileMergerN)rkwargsr5cs:tdddd|kr(t|dkr(d|d<tj||dS)NrrArrDrT)rr`superr;)r8rr __class__r9r:r;0s zPdfFileMerger.__init__)r<r=r>rr; __classcell__r9r9rr:r/sr)@rYiorrrpathlibrtypesrtypingrrr r r r r rrr_rZ_pagerrr_utilsrrrrZ_writerr constantsrrrrrZgenericrrrr r!r"r#r$r%r&r'r(r)Z pageranger*r+r,r-r.r/r0rr1rArr9r9r9r:s0  ,      < \