NggVdZddlZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl m ZddlmZddlmZddlmZddlmZddlmZdd lmZejeZGd d ejZ Gd d ejZ!GddejZ"dZ#e ddZ$dS)) PdfDocument PdfFormEnv PdfXObjectPdfOutlineItemN)Path) namedtuple) PDFIUM_INFO) PdfiumError)PdfPage) PdfObject) PdfAttachmentc\eZdZdZd'fd ZfdZedZedZ dZ d Z d Z d Z ed Zd(d ZdZdZdZdejfdZejfdZdZdZdZd)dZdZdZdZdZ dZ!d(dZ"dZ#d*dZ$dZ%d Z&d!Z'd"Z( d+d%Z)e*j+ddddfd&Z+xZ,S),ra Document helper class. Parameters: input_data (str | pathlib.Path | bytes | ctypes.Array | typing.BinaryIO | FPDF_DOCUMENT): The input PDF given as file path, bytes, ctypes array, byte buffer, or raw PDFium document handle. A byte buffer is defined as an object that implements ``seek() tell() read() readinto()``. password (str | None): A password to unlock the PDF, if encrypted. Otherwise, None or an empty string may be passed. If a password is given but the PDF is not encrypted, it will be ignored (as of PDFium 5418). autoclose (bool): Whether byte buffer input should be automatically closed on finalization. Raises: PdfiumError: Raised if the document failed to load. The exception message is annotated with the reason reported by PDFium. FileNotFoundError: Raised if an invalid or non-existent file path was given. Hint: * :func:`len` may be called to get a document's number of pages. * Looping over a document will yield its pages from beginning to end. * Pages may be loaded using list index access. * The ``del`` keyword and list index access may be used to delete pages. Attributes: raw (FPDF_DOCUMENT): The underlying PDFium document handle. formenv (PdfFormEnv | None): Form env, if the document has forms and :meth:`.init_forms` was called. NFct|trt|}t|trI|}|st |||_||_||_ g|_ g|_ d|_ t|jtjr |j|_nIt!|j|j|j \|_}}|xj |z c_ |xj |z c_ t#t&j|j |j dSN) isinstancestrr expanduserresolveis_fileFileNotFoundError_input _password _autoclose _data_holder _data_closerformenvpdfium_c FPDF_DOCUMENTraw _open_pdfsuper__init__r _close_impl)selfinputpassword autocloseto_holdto_close __class__s W/var/www/html/ai-engine/env/lib/python3.11/site-packages/pypdfium2/_helpers/document.pyr"zPdfDocument.__init__9s2 eS ! ! KKE eT " " /$$&&..00E==?? /'... !# dk8#9 : : *{DHH*3DKQUQ`*a*a 'DHgx    (      )   0$2CTEVWWWWWc t|jtr"tt |j}nt|jt r&dt t|jd}nYt|jtj r&dt t|jd}nt|j}t ddd|dS)Nzz*>AAAGG  X%; < < (B3r$++?+?BBBGG4;''G''""$$SbS);;;;;;r,cdSrr$s r+parentzPdfDocument.parentastr,ctj||D]}t||D]}|||dSr)rFPDF_CloseDocumentr3closeclear)r data_holder data_closerdatas r+r#zPdfDocument._close_implfsw#C(((  D tHHHH  D JJLLLLr,c*tj|Sr)rFPDF_GetPageCountr8s r+__len__zPdfDocument.__len__qs)$///r,c#\Ktt|D] }||V dSr)rangelenr$is r+__iter__zPdfDocument.__iter__ts<D ##  Aq'MMMM  r,c,||Sr)get_pagerGs r+ __getitem__zPdfDocument.__getitem__xs}}Qr,c0||dSr)del_pagerGs r+ __delitem__zPdfDocument.__delitem__{s ar,c>tj}||S)zJ Returns: PdfDocument: A new, empty document. )rFPDF_CreateNewDocument)clsnew_pdfs r+newzPdfDocument.news 133s7||r,c|}|tjks|jrdSdtjvr/tjdkrtjdkrtd|sbdtjvr?tj d}tj d d tj | }ntj d }tj ||}|std |d t||||_||j|tjtjfvrdtjvrgtj|}|sOtj}t*dt.j|dSdSt*ddSdS)a Initialize a form env, if the document has forms. If already initialized, nothing will be done. See the :attr:`formenv` attribute. Attention: If form rendering is desired, this method shall be called immediately after document construction, before getting document length or page handles. Parameters: config (FPDF_FORMFILLINFO | None): Custom form config interface to use (optional). NV8 sourcebuildi-z@V8 enabled pdfium-binaries builds <= 5677 crash on init_forms().XFA)versionF)rZ xfa_disabled m_pJsPlatformz*Initializing form env failed for document .zFPDF_LoadXFA() failed with zinit_forms() called on XFA pdf, but this pdfium binary was compiled without XFA support. Run `PDFIUM_PLATFORM=auto-v8 pip install -v pypdfium2 --no-binary pypdfium2` to get a build with XFA support.) get_formtyper FORMTYPE_NONErr flagsoriginbuild RuntimeErrorIPDF_JSPLATFORMFPDF_FORMFILLINFOctypespointerFPDFDOC_InitFormFillEnvironmentr r_add_kidFORMTYPE_XFA_FULLFORMTYPE_XFA_FOREGROUND FPDF_LoadXFAFPDF_GetLastErrorloggerwarningpdfium_i XFAErrorToStrget)r$configformtype js_platformrokerrs r+ init_formszPdfDocument.init_formss$$&& x- - - - F ;$ $ $);})L)LQ\QbfjQjQjabb b ? )))&6qAAA !3AEagaop{a|a|}}}!3A>>>6tVDD TR4RRRSS S!#vt44  dl### 2H4TU U U )))*400d"466CNN#bAWA[A[\_A`A`#b#bcccccddD V Ur,c*tj|S)z Returns: int: PDFium form type that applies to the document (:attr:`FORMTYPE_*`). :attr:`FORMTYPE_NONE` if the document has no forms. )rFPDF_GetFormTyper8s r+r_zPdfDocument.get_formtypes (...r,c*tj|S)zV Returns: int: Page displaying mode (:attr:`PAGEMODE_*`). )rFPDFDoc_GetPageModer8s r+ get_pagemodezPdfDocument.get_pagemodes +D111r,cDttj|S)zm Returns: bool: Whether the document is tagged (cf. PDF 1.7, 10.7 "Tagged PDF"). )boolrFPDFCatalog_IsTaggedr8s r+ is_taggedzPdfDocument.is_taggeds X2488:::r,ct|ttfrt|dd}}n-t j|dr|d}}nt d|d |t j||f}|tj |ntj g||R}|std |r| dSdS#|r| wwxYw) a  Save the document at its current state. Parameters: dest (str | pathlib.Path | io.BytesIO): File path or byte buffer the document shall be written to. version (int | None): The PDF version to use, given as an integer (14 for 1.4, 15 for 1.5, ...). If None (the default), PDFium will set a version automatically. flags (int): PDFium saving flags (defaults to :attr:`FPDF_NO_INCREMENTAL`). wbTwFzCannot save to ''NzFailed to save document.) rrropenrq is_buffer ValueError get_bufwriterrFPDF_SaveAsCopyFPDF_SaveWithVersionr r<)r$destrZrabuffer need_closesaveargsrws r+savezPdfDocument.saves dS$K ( ( 9!%dD!1!14JFF  c * * 9!%uJFF777788 8 h4V<!"<=== >    z   s A CCctj||dd}tj|}tj|||||jd|dz S)a Parameters: type (int): The identifier type to retrieve (:attr:`FILEIDTYPE_*`), either permanent or changing. If the file was updated incrementally, the permanent identifier stays the same, while the changing identifier is re-calculated. Returns: bytes: Unique file identifier from the PDF's trailer dictionary. See PDF 1.7, Section 14.4 "File Identifiers". Nrr[)rFPDF_GetFileIdentifierrgcreate_string_bufferr)r$typen_bytesrs r+get_identifierzPdfDocument.get_identifiersV1$dAFF,W55'dFGDDDz*719*%%r,chtj}tj||}|sdS|jS)z Returns: int | None: The PDF version of the document (14 for 1.4, 15 for 1.5, ...), or None if the document is new or its version could not be determined. N)rgc_intrFPDF_GetFileVersionvalue)r$rZrws r+ get_versionzPdfDocument.get_versions6 ,..  )$ 8 8 4}r,c|dzd}tj||dd}tj|}tj|||||jd|dz dS)z Returns: str: Value of the given key in the PDF's metadata dictionary. If the key is not contained, an empty string will be returned. utf-8Nrr[ utf-16-le)encoderFPDF_GetMetaTextrgrrdecode)r$keyenc_keyrrs r+get_metadata_valuezPdfDocument.get_metadata_values{ <''00+D'4CC,W55!$AAAz*719*%,,[999r,)TitleAuthorSubjectKeywordsCreatorProducer CreationDateModDatecnfdjD}|rd|D}|S)z Get the document's metadata as dictionary. Parameters: skip_empty (bool): If True, skip items whose value is an empty string. Returns: dict: PDF metadata. c<i|]}||Sr7)r).0kr$s r+ z1PdfDocument.get_metadata_dict..,s)NNNaAt..q11NNNr,ci|] \}}||| Sr7r7)rrvs r+rz1PdfDocument.get_metadata_dict...s#???AQ?1???r,) METADATA_KEYSitems)r$ skip_emptymetadatas` r+get_metadata_dictzPdfDocument.get_metadata_dict"sMONNN4;MNNN  @??)9)9???Hr,c*tj|S)zY Returns: int: The number of embedded files in the document. )rFPDFDoc_GetAttachmentCountr8s r+count_attachmentszPdfDocument.count_attachments2s 24888r,cvtj||}|std|dt||S)z] Returns: PdfAttachment: The attachment at *index* (zero-based). z"Failed to get attachment at index r^)rFPDFDoc_GetAttachmentr r )r$indexraw_attachments r+get_attachmentzPdfDocument.get_attachment:sI "7eDD MK5KKKLL L^T222r,c|dzd}tj|tj}tj||}|st d|dt||S)aN Add a new attachment to the document. It may appear at an arbitrary index (as of PDFium 5418). Parameters: name (str): The name the attachment shall have. Usually a file name with extension. Returns: PdfAttachment: Handle to the new, empty attachment. rrz!Failed to create new attachment 'z'.)rrgcastrFPDF_WIDESTRINGFPDFDoc_AddAttachmentr r )r$nameenc_name enc_name_ptrrs r+new_attachmentzPdfDocument.new_attachmentEst6M))+66{8X-EFF !7lKK LJ$JJJKK K^T222r,cZtj||}|std|ddS)a Unlink the attachment at *index* (zero-based). It will be hidden from the viewer, but is still present in the file (as of PDFium 5418). Following attachments shift one slot to the left in the array representation used by PDFium's API. Handles to the attachment in question received from :meth:`.get_attachment` must not be accessed anymore after this method has been called. z%Failed to delete attachment at index r^N)rFPDFDoc_DeleteAttachmentr )r$rrws r+del_attachmentzPdfDocument.del_attachmentWsE .tU ; ; PNeNNNOO O P Pr,c tj||}|stdt|||j}|jr5tj||j|j|n|||S)a Returns: PdfPage: The page at *index* (zero-based). Note: This calls ``FORM_OnAfterLoadPage()`` if the document has an active form env. The form env must not be closed before the page is closed! zFailed to load page.)r FPDF_LoadPager r rFORM_OnAfterLoadPagerj)r$rraw_pagepages r+rKzPdfDocument.get_pagefs)$66 6455 5xt|44 <  )$ = = = L ! !$ ' ' ' ' MM$    r,c|t|}tj||||}t||d}|||S)a Insert a new, empty page into the document. Parameters: width (float): Target page width (horizontal size). height (float): Target page height (vertical size). index (int | None): Suggested zero-based index at which the page shall be inserted. If None or larger that the document's current last index, the page will be appended to the end. Returns: PdfPage: The newly created page. N)rFr FPDFPage_Newr rj)r$widthheightrrrs r+new_pagezPdfDocument.new_page}sR =IIE(ueVDDxt,, d r,c0tj||dS)z: Remove the page at *index* (zero-based). N)rFPDFPage_Delete)r$rs r+rNzPdfDocument.del_pages  u-----r,cF|t|}t|tr+tj|||d|}n>d}d}|r t|}t j|z|}tj|||||}|stddS)a. Import pages from a foreign document. Parameters: pdf (PdfDocument): The document from which to import pages. pages (list[int] | str | None): The pages to include. It may either be a list of zero-based page indices, or a string of one-based page numbers and ranges. If None, all pages will be included. index (int): Zero-based index at which to insert the given pages. If None, they are appended to the end of the document. NasciirzFailed to import pages.) rFrrrFPDF_ImportPagesrrgrFPDF_ImportPagesByIndexr )r$pdfpagesrrw page_countc_pagess r+ import_pageszPdfDocument.import_pagess =IIE eS ! ! Y*4ell76K6KUSSBBJG > ZZ !<*4u=1$WjRWXXB 9788 8 9 9r,ctj}tj|||}|std|j|jfS)z Returns: (float, float): Width and height in PDF canvas units of the page at *index* (zero-based). z!Failed to get page size by index.)rFS_SIZEFFPDF_GetPageSizeByIndexFr rr)r$rsizerws r+ get_page_sizezPdfDocument.get_page_sizesM  ""  .tUD A A CABB B DK((r,ctj||dd}tj|}tj|||||jd|dz dS)z Returns: str: Label of the page at *index* (zero-based). (A page label is essentially an alias that may be displayed instead of the page number.) Nrr[r)rFPDF_GetPageLabelrgrrr)r$rrrs r+get_page_labelzPdfDocument.get_page_labelsd ,T5$BB,W55"4@@@z*719*%,,[999r,ctj|||}|td|dt||}|||S)a_ Capture a page as XObject and attach it to a document's resources. Parameters: index (int): Zero-based index of the page. dest_pdf (PdfDocument): Target document to which the XObject shall be added. Returns: PdfXObject: The page as XObject. Nz Failed to capture page at index z as FPDF_XOBJECT.rr)rFPDF_NewXObjectFromPager rrj)r$rdest_pdf raw_xobjectxobjects r+page_as_xobjectzPdfDocument.page_as_xobjectsa6xuMM  YYYYZZ Z(;;; gr,c |tj|dd}tj|}tj||||jd|dz d}tj|}|dkrdn |dkrdnd}t|}tj||} tj || } | dkrd} tj } tj dz} tj | | | } t| d| j} t||||| | | S) Nrr[rTFr/)leveltitle is_closedn_kids page_index view_modeview_pos)rFPDFBookmark_GetTitlergrrrFPDFBookmark_GetCountabsFPDFBookmark_GetDestFPDFDest_GetDestPageIndexc_ulongFS_FLOATFPDFDest_GetViewlistrr)r$bookmarkrrrrcountrrrrn_paramsrrs r+ _get_bookmarkzPdfDocument._get_bookmarksA04CC,W55&xAAA :GAI:&--k::.x88!AIIDD5A::445 U,T8<<7dCC   J>##%),,-dHhGG >>/8>/2!#!    r,rc#K|t}tj||}|rtj|j}||vrt ddS||| ||V||dz kr"| |||dz|Ed{Vtj ||}|dSdS)a Iterate through the bookmarks in the document's table of contents. Parameters: max_depth (int): Maximum recursion depth to consider. Yields: :class:`.PdfOutlineItem`: Bookmark information. NzPA circular bookmark reference was detected whilst parsing the table of contents.) max_depthr9rseen) setrFPDFBookmark_GetFirstChildrg addressofcontentsrorpaddrget_tocFPDFBookmark_GetNextSibling)r$r r9rr raddresss r+rzPdfDocument.get_toc s"" <55D6tVDD L&x'899G$qrrr!!!$$Xu55 5 5 5y{""<< )%!AI ( ;D(KKH% L L L L Lr,c+Ktjdt|s&dtt |D}|D]B}|||fi|} |r"|| | fV5|| VCdS)a .. deprecated:: 4.19 This method will be removed with the next major release due to serious issues rooted in the original API design. Use :meth:`.PdfPage.render()` instead. *Note that the CLI provides parallel rendering using a proper caller-side process pool with inline saving in rendering jobs.* .. versionchanged:: 4.25 Removed the original process pool implementation and turned this into a wrapper for linear rendering, due to the serious conceptual issues and possible memory load escalation, especially with expensive receiving code (e.g. PNG encoding) or long documents. See the changelog for more info zThe document-level pdf.render() API is deprecated and uncored due to serious issues in the original concept. Use page.render() and a caller-side loop or process pool instead.)categorycg|]}|Sr7r7)rrHs r+ z&PdfDocument.render..Ms888!A888r,N)warningswarnDeprecationWarningrErFget_info) r$ converterrenderer page_indices pass_info n_processes mk_formconfigkwargsrHbitmaps r+renderzPdfDocument.render7s&  GRd e e e e 988uSYY'7'7888L ( (AXd1g0000F ( y((&//*;*;<<<<<i''''''  ( (r,)NFr)F)NN)rNrN)-__name__ __module__ __qualname____doc__r"r4propertyr9 staticmethodr#rCrIrLrO classmethodrTryr_r~rrFPDF_NO_INCREMENTALrFILEIDTYPE_PERMANENTrrrrrrrrrrKrrNrrrrrrr r% __classcell__r*s@r+rrs<XXXXXX8 < < < < <X\000   [----b///222;;;"&X-I>#+"?&&&&"    : : :qM     999333333$ P P P.0...9999> ) ) ) : : :,   H (L(L(L(L\~ ((((((((r,rcNeZdZdZfdZedZedZxZ S)raM Form environment helper class. Attributes: raw (FPDF_FORMHANDLE): The underlying PDFium form env handle. config (FPDF_FORMFILLINFO): Accompanying form configuration interface, to be kept alive. pdf (PdfDocument): Parent document this form env belongs to. c|||c|_|_|_tt j|j|jdSr)rrtrr!r"rr#)r$rrtrr*s r+r"zPdfFormEnv.__init__csB*-vs'$+tx /dhGGGGGr,c|jSrrr8s r+r9zPdfFormEnv.parentg xr,cZtj|t|d|_dSr)rFPDFDOC_ExitFormFillEnvironmentr3r)rrtrs r+r#zPdfFormEnv._close_implks)0555 6  r,) r&r'r(r)r"r*r9r+r#r/r0s@r+rrVs}  HHHHHX\r,rc>eZdZdZfdZedZdZxZS)rz XObject helper class. Attributes: raw (FPDF_XOBJECT): The underlying PDFium XObject handle. pdf (PdfDocument): Reference to the document this XObject belongs to. c|||c|_|_ttjdSr)rrr!r"rFPDF_CloseXObject)r$rrr*s r+r"zPdfXObject.__init__{s4 #$( 344444r,c|jSrr4r8s r+r9zPdfXObject.parentr5r,cVtj|}t||jS)a Returns: PdfObject: An independent page object representation of the XObject. If multiple page objects are created from one XObject, they share resources. Page objects created from an XObject remain valid after the XObject is closed. r)rFPDF_NewFormObjectFromXObjectr r)r$ raw_pageobjs r+ as_pageobjectzPdfXObject.as_pageobjects3Q>U>UV^>_>_cccddd  !!r,rz:level title is_closed n_kids page_index view_mode view_pos)%__all__osrglogginginspectrpathlibr collectionsrmultiprocessingmp pypdfium2.rawrrpypdfium2.internalinternalrqpypdfium2.versionr pypdfium2._helpers.miscr pypdfium2._helpers.pager pypdfium2._helpers.pageobjectsr pypdfium2._helpers.attachmentr getLoggerr&ro AutoCloseablerrrr rr7r,r+r^s H """""" %%%%%%))))))//////++++++444444777777  8 $ $y(y(y(y(y(8)y(y(y(x((8     ((   >"""6,.jkkr,