o `^hw@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*eZdZdZdedededdfddZdS) _MergedPagez@Collect necessary information on each page that is being merged.pagedatasrcidreturnNcCs||_||_d|_||_dSN)r3r2 out_pagedatar4)selfr2r3r4r9L/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/PyPDF2/_merger.py__init__Qs z_MergedPage.__init__)__name__ __module__ __qualname____doc__rrintr;r9r9r9r:r1Nsr1c@seZdZdZedd d]dedeeefdd fd d Z d^d d Z de e e de e de edd fddZeddd  d_de edeeeefde ede edede edd fddZdeeeefdeee effddZeddd d`deeeefde eded eeeefeeeefeefdedd f ddZdeeefdd fd d!Zdad"d#Zd$eeefdd fd%d&Z d$eeefdd fd'd(Z!d)e"dd fd*d+Z#d)e"dd fd,d-Z$d.e%dd fd/d0Z&d.e%dd fd1d2Z'd3ed4eeeeeffdeeeefeeeefeefdeeeeffd5d6Z(d3ede)deeeefeeeefeefde)fd7d8Z*dad9d:Z+edd dbde e,e-d;e e.dd fddee-e0fd?e1dd fd@dAZ2dee1dd fdBdCZ3edd dcdee1de e,e-dd fdDdEZ4edd> dcdeeefdFe e)de eefdGdHZ5edd> dcdeeefdFe e)de eefdIdJZ6d d d dde7d fdKede ed;ed e.e8fdLe ee9e9e9fdMedNedOe:dPe ede8fdQdRZ;   SdddKedPed;ed e.e8fdLe ee9e9e9fdMedNedOe   SdddKedPed;ed e.e8fdLe ee9e9e9fdMedNedOe` (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) bookmarksFstrictfileobjr5NcCs6g|_g|_t|_g|_g|_d|_||_||_dS)Nr) inputspagesroutputrB named_destsid_countrFrE)r8rErFr9r9r:r;is zPdfMerger.__init__cCs|Sr6r9)r8r9r9r: __enter__vszPdfMerger.__enter__exc_typeexc tracebackcCs|jr ||j|dS)z*Write to the fileobj and close the merger.N)rFwriteclose)r8rMrNrOr9r9r:__exit__zs  zPdfMerger.__exit__ outline_itemimport_outline)bookmarkimport_bookmarksT page_numberrHpositioncCs|dur|dur|}d}d}t|d|dtntd|dur'td|dur/td||\} } t| |jd } |j| | f| durL| | _ |durXd t | j f}nt |t rf|t | j }nt |trln t |tsutd g} g} |r| j} || | |} |rtt|t|jt}|j|| g7_n|j| 7_| j}|| ||}|j|7_t|D]}| j |}|j}|jd 7_t|| |}| |q|| | | | |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``. NrXrWzG 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)rErz0"pages" must be a tuple of (start, stop[, step])r)!warningswarnDeprecationWarning ValueError_create_streamrrErGappend _encryptionlenrH isinstancer*indiceslisttuple TypeErrorrB _trim_outliner'r(r&rKr!fitnamed_destinations _trim_destsrJranger1_associate_dests_to_pages!_associate_outline_items_to_pages)r8rWrFrSrHrTrXold_termnew_termstreamencryption_objreadersrcpagesrBoutline_item_typdests trimmed_destsipager4mpr9r9r:mergesp#         zPdfMerger.mergecCsd}t|ttfrt|d}||fSt|tr9|jr|j}|j}|jdt |j }|j|||fSt |drTt |drT|d| }t |}||fSt d)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.) rcstrrrrrarqtellr}rr~hasattrNotImplementedError)r8rFrrrq orig_tell filecontentr9r9r:r_s(        zPdfMerger._create_streamcCs|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)r{rbrH)r8rFrSrHrTr9r9r:r`s!zPdfMerger.appendcCs|jdur tt|jD]$}|j|jtttt f|jj }|j |t jd |_q |||j|\}}|rI|dSdS)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)rI RuntimeErrorERR_CLOSED_WRITERrHadd_pager2rrrr_pages get_object get_referencePAKIDSr7 _write_dests_write_outlinerPrQ)r8rFry pages_objmy_file ret_fileobjr9r9r:rPAs   zPdfMerger.writecCs.g|_|jD]\}}|qg|_d|_dS)zHShut all file descriptors (input and output) and clear all memory usage.N)rHrGrQrI)r8fo_readerr9r9r:rQas   zPdfMerger.closeinfoscC"|jdur tt|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)rIrr add_metadatar8rr9r9r:rjs zPdfMerger.add_metadatacCtdd||dS)zW .. deprecated:: 1.28.0 Use :meth:`add_metadata` instead. addMetadatarN)rrrr9r9r:rv zPdfMerger.addMetadatalayoutcCr)zZ .. deprecated:: 1.28.0 Use :meth:`set_page_layout` instead. setPageLayoutset_page_layoutN)rrr8rr9r9r:rrzPdfMerger.setPageLayoutcCr)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)rIrr_set_page_layoutrr9r9r:rs zPdfMerger.set_page_layoutmodecCstddd||dS)zX .. deprecated:: 1.28.0 Use :meth:`set_page_mode` instead. setPageMode set_page_mode3.0.0N)rrr8rr9r9r:rs zPdfMerger.setPageModecCr)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)rIrrrrr9r9r:rs zPdfMerger.set_page_modepdfrvc Csg}t|tr |ntt|}|D]3\}}|D],}|j||dkrE|d|td<t|t|dks>J||nqq|S)zHRemove named destinations that are not a part of the specified page set./Page/Title) rcrerlitemsrHrr$rr`) r8rrvrH new_destslstkeyobjjr9r9r:rks zPdfMerger._trim_destsc Csg}d}t|tr |ntt|}t|D]Q\}}t|tr8||||} | r7|s2|||d|| qd}|D])} |ddurEq<|j| |dkre|d|td<||d}nq}t|tr*|||qd}d|vrIt|jD]\}}|j |dkrH| ||nq5|durZ|d=|d=|j ||}qdS)N hint for mypyr/Type) rIrrrBrcrerrrHr4_write_outline_item_on_pageadd_outline_item_dict)r8rBr last_addedrSpage_noryr9r9r:rs,     zPdfMerger._write_outline)rUrycCstt|d}t|jt|g}tjtjftj tjftj tj ftj tj ftj tj tjdftjtj tjtjtjfi}||tD]!}||vrWt||tsW|t||n|td||=q@tttjtdttjt|i|td<dS)Nrz/Zoomrz/GoToz/A)rrr&r4r$rFIT_HrTOPFIT_BHFIT_VLEFTFIT_BVXYZFIT_RBOTTOMRIGHTgetrfrcr%r`r"r rSDr)r8rSryoi_typeargs fit2arg_keysarg_keyr9r9r:r"s4     z%PdfMerger._write_outline_item_on_pagecCsv|jD]5}d}|d}t|trq|D]}||jkr!|j}q|dur0td|ddt||td<qdS)NrzUnresolved named destination 'r')rJrcr&rr2r4r^r$)r8rHrrnpryr9r9r:rmEs  z#PdfMerger._associate_dests_to_pagescCs|dur|j}|dusJd|D]7}t|tr|||qd}|d}t|tr+q|D]}||jkr;|j}q-|durHt||td<qdS)Nrr) rBrcrernr&rr2r4r$)r8rHrBrSroutline_item_pagepr9r9r:rnWs&   z+PdfMerger._associate_outline_items_to_pagesrootcCsz|dur|j}t|D]/\}}t|tr$|||}|r#|g|Sq ||ks5ttttf|d|kr:|gSq dS)Nr)rBrrcrefind_outline_itemrrr)r8rSrrxoi_enumresr9r9r:rps   zPdfMerger.find_outline_itemcCs |||S)zZ .. deprecated:: 2.9.0 Use :meth:`find_outline_item` instead. )r)r8rSrr9r9r: find_bookmarks zPdfMerger.find_bookmarktitlecolorbolditalicripagenumc Cs|dur |dur td|dur"d} d} t| d| dt|}|dur*td|j} | dur5tt| |||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.rrWH is deprecated as an argument and will be removed in PyPDF2==4.0.0. Use rYrZ)r^r[r\r]rIrradd_outline_item) r8rrWrrrrrirrorpwriterr9r9r:rs:zPdfMerger.add_outline_item/Fitrc G*tddd|||||||t||dS)zZ .. deprecated:: 1.28.0 Use :meth:`add_outline_item` instead. addBookmarkrrfit_typefit_argsrrr! r8rrrrrrrirr9r9r:r  zPdfMerger.addBookmarkc Gr)zY .. deprecated:: 2.9.0 Use :meth:`add_outline_item` instead. rrrrrrr9r9r: add_bookmarkrzPdfMerger.add_bookmarkcCstddd|||S)z_ .. deprecated:: 1.28.0 Use :meth:`add_named_destination` instead. addNamedDestinationadd_named_destinationr)rr)r8rrr9r9r:rs zPdfMerger.addNamedDestinationcCs|dur |dur td|dur"d}d}t|d|dt|}|dur*tdtt|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.rrWrrYrZi:)top) r^r[r\r]rr(r&r!fit_horizontallyrJr`)r8rrWrrorpdestr9r9r:r s,  zPdfMerger.add_named_destination)FrD)r5rA)NNNNTN)NNT)r5N)NNr6)NNFFr)Br<r=r>r?rboolrrrr;rLr r BaseExceptionrrRr@rrr+r{r rrr_r*r r`rPrQrrrrr-rrr/rrrkr.rhrr r'r)rrr1rrmrnrrrr#floatr!rr,r0rrrrr9r9r9r:rAXs      u  # " "      " "     "           ;       rAcs*eZdZdededdffdd ZZS) PdfFileMergerrkwargsr5Ncs>tdddd|vrt|dkrd|d<tj|i|dS)NrrArrErT)rrbsuperr;)r8rr __class__r9r:r;0s zPdfFileMerger.__init__)r<r=r>rr; __classcell__r9r9rr:r/s"r)@r[iorrrpathlibrtypesrtypingrrr r r r r rrrar_pagerrr_utilsrrrr_writerr constantsrrrrrgenericrrrr r!r"r#r$r%r&r'r(r) pageranger*r+r,r-r.r/r0rr1rArr9r9r9r:s2  ,     < \