o jqP@s dZddlZddlZddlZddlmZmZddlmZddl m Z m Z m Z ddl Z ddlmZddlmZdd lmZeeee jd Zegd ks`gd ekr^gd krdednedejejddGdddZGdddeZGdddeZ dS)zPDF to Docx Converter.N)Pool cpu_count) perf_counter)AnyStrIOUnion)DocumentPage)Pages.)r r)r )r rzJ1.19.0 <= PyMuPDF <= 1.23.8, or PyMuPDF>=1.23.16 is required for pdf2docx.z[%(levelname)s] %(message)s)levelformatc @sheZdZdZ d:dededefddZedd Zed d Z d d Z eddZ d;de de de fddZd;de de de fddZddZddZd ConverteraThe ``PDF`` to ``docx`` converter. * Read PDF file with ``PyMuPDF`` to get raw layout data page by page, including text, image, drawing and its properties, e.g. boundary box, font, size, image width, height. * Analyze layout in document level, e.g. page header, footer and margin. * Parse page layout to docx structure, e.g. paragraph and its properties like indentation, spacing, text alignment; table and its properties like border, shading, merging. * Finally, generate docx with ``python-docx``. Npdf_filepasswordstreamcCsP||_t|pd|_|s|std|rtj|d|_nt||_t|_dS)zInitialize fitz object with given pdf file path. Args: pdf_file (str): pdf file path. stream (bytes): pdf file in memory. password (str): Password for encrypted pdf. Default to None if not encrypted. z(Either pdf_file or stream must be given.)rN) filename_pdfstrr ValueErrorfitzr _fitz_docr _pages)selfrrrr S/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/pdf2docx/converter.py__init__&s   zConverter.__init__cC|jSN)rrr r r!fitz_docAzConverter.fitz_doccCr#r$)rr%r r r!pagesDr'zConverter.pagescCs|jdSr$)rcloser%r r r!r)HszConverter.closecCsiddddddddddd d d d d ddddddd dd dddddddd ddiddddd d!d"d!d#dd$d%d&d'd(dd)dd*dd+dd,dd-dd.dd/dd0dS)1zDefault parsing parameters.debugFZocrrignore_page_errorTmulti_processingrZmin_section_heightg4@Zconnected_border_toleranceg?Zmax_border_widthg@Zmin_border_clearanceg@Zfloat_image_ignorable_gapg@Zpage_margin_factor_topZpage_margin_factor_bottomZshape_min_dimensionZmax_line_spacing_ratiog?Zline_overlap_thresholdg?Zline_break_width_ratioZline_break_free_space_ratiog?Zline_separate_thresholdZnew_paragraph_free_space_ratiog333333?Zlines_left_aligned_thresholdg?Zlines_right_aligned_thresholdZlines_center_aligned_thresholdZclip_image_res_ratiog@Zmin_svg_gap_dxg.@Zmin_svg_gap_dyZ min_svg_wZ min_svg_hZextract_stream_tableZparse_lattice_tableZparse_stream_tableZdelete_end_line_hyphenraw_exceptionsZlist_not_tabler r%r r r!default_settingsKs      !zConverter.default_settingsrstartendr(cKs&||||jdi|jdi|S)a2Parse pages in three steps: * open PDF file with ``PyMuPDF`` * analyze whole document, e.g. page section, header/footer and margin * parse specified pages, e.g. paragraph, image and table Args: start (int, optional): First page to process. Defaults to 0, the first page. end (int, optional): Last page to process. Defaults to None, the last page. pages (list, optional): Range of page indexes to parse. Defaults to None. kwargs (dict, optional): Configuration parameters. Nr ) load_pagesparse_document parse_pages)rr/r0r(kwargsr r r!parsevs zConverter.parsecCst|d|jjr#|jstd|jd|j|js#tdt |j}|j ddt |D| ||||}|D]}d|j |_q?|S)aStep 1 of converting process: open PDF file with ``PyMuPDF``, especially for password encrypted file. Args: start (int, optional): First page to process. Defaults to 0, the first page. end (int, optional): Last page to process. Defaults to None, the last page. pages (list, optional): Range of page indexes to parse. Defaults to None. z[1/4] Opening document...zRequire password for r zIncorrect password.cSg|]}t|ddqST)id skip_parsingr .0ir r r! z(Converter.load_pages..F)logginginfo _color_outputrZ needs_passrConversionExceptionrZ authenticatelenrresetrange _page_indexesr9)rr/r0r(num page_indexesr<r r r!r1s  zConverter.load_pagescKs*t|d|jj|jfi||S)zjStep 2 of converting process: analyze whole document, e.g. page section, header/footer and margin.z[2/4] Analyzing document...)r?r@rArr5r&)rr4r r r!r2szConverter.parse_documentc Kst|ddd|jD}t|}t|ddD]L\}}|jd}td|||z |jdi|Wqtyf}z%|drB|dsR|d rRt d ||n t d |d |WYd }~qd }~ww|S)zKStep 3 of converting process: parse pages, e.g. paragraph, image and table.z[3/4] Parsing pages...cSsg|]}|js|qSr )r9r;pager r r!r=sz)Converter.parse_pages..r r/(%d/%d) Page %dr-r*r+z,Ignore page %d due to parsing page error: %szError when parsing page : Nr ) r?r@rArrC enumerater8r5 ExceptionerrorrB)rr4r( num_pagesr<rJpider r r!r3s$  zConverter.parse_pagesc Ks8t|dttdd|j}|std|s:|jr6|jdtd d}t j |r5t |ntdt }t|}t|d d D]M\}}|jsOqG|jd }td |||z||WqGty} z%|d rp|d s|drtd|| n td|d| WYd} ~ qGd} ~ ww||dS)zStep 4 of converting process: create docx file with converted pages. Args: filename_or_stream (str, file-like): docx file to write. kwargs (dict, optional): Configuration parameters. z[4/4] Creating pages...cSr#r$) finalized)rJr r r!sz%Converter.make_docx..z)No parsed pages. Please parse page first.rz.pdfz.docxz?Please specify a docx file name or a file-like object to write.r rKrLr-r*r+z+Ignore page %d due to making page error: %szError when make page rMN)r?r@rAlistfilterrrBrrCospathexistsremoverrNrTr8 make_docxrOrPMakedocxExceptionsave) rZfilename_or_streamr4Z parsed_pagesZ docx_filerQr<rJrRrSr r r!r\s<    zConverter.make_docxcCs(tj|jt|jdd|jDdS)z"Store parsed pages in dict format.cSsg|] }|jr|qSr )rTstorerIr r r!r=sz#Converter.store..)filenamepage_cntr()rXrYbasenamerrCrr%r r r!r_s zConverter.storedatacCs^|js|dd}|jddt|D|dgD]}|dd}|j||qdS) z"Restore pages from parsed results.radcSr6r7r r:r r r!r=r>z%Converter.restore..r(r8N)rgetrDrErestore)rrcrGZraw_pageidxr r r!rgs  zConverter.restorer`cCsLt|ddd}|tj|ddWddS1swYdS)z*Write parsed pages to specified JSON file.wzutf-8)encoding)indentN)openwritejsondumpsr_)rr`fr r r! serialize s"zConverter.serializecCsBt|d }t|}Wdn1swY||dS)z+Load parsed pages from specified JSON file.rN)rmroloadrg)rr`rqrcr r r! deserializes  zConverter.deserializer< docx_filename debug_pdf layout_filecKsxtj|j\}}|stj|d|}|stj|d}|dt|d|j|fd|gi|| |dS)aParse, create and plot single page for debug purpose. Args: i (int): Page index to convert. docx_filename (str): docx filename to write to. debug_pdf (str): New pdf file storing layout information. Default to add prefix ``debug_``. layout_file (str): New json file storing parsed layout data. Default to ``layout.json``. Zdebug_z layout.jsonT)r*Z debug_docZdebug_filenamer(N) rXrYsplitrjoinupdaterrconvertrr)rr<rvrwrxr4rYr`r r r! debug_pages zConverter.debug_pagecKst}td|j|j}|||r|drtd|dr,|j|||fi|n|j|||fi|j |fi|tdt|dS)aConvert specified PDF pages to docx file. Args: docx_filename (str, file-like, optional): docx file to write. Defaults to None. start (int, optional): First page to process. Defaults to 0, the first page. end (int, optional): Last page to process. Defaults to None, the last page. pages (list, optional): Range of page indexes. Defaults to None. kwargs (dict, optional): Configuration parameters. Defaults to None. Refer to :py:meth:`~pdf2docx.converter.Converter.default_settings` for detail of configuration parameters. .. note:: Change extension from ``pdf`` to ``docx`` if ``docx_file`` is None. .. note:: * ``start`` and ``end`` is counted from zero if ``--zero_based_index=True`` (by default). * Start from the first page if ``start`` is omitted. * End with the last page if ``end`` is omitted. .. note:: ``pages`` has a higher priority than ``start`` and ``end``. ``start`` and ``end`` works only if ``pages`` is omitted. .. note:: Multi-processing works only for continuous pages specified by ``start`` and ``end`` only. zStart to convert %sr,zPMulti-processing works for continuous pages specified by "start" and "end" only.zTerminated in %.2fs.N) rr?r@rr.r{rB_convert_with_multi_processingr5r\)rrvr/r0r(r4t0settingsr r r!r|8s  $zConverter.convertcKsV|j}|||j|||fi|g}|jD]}|jr(||jdi|q|S)aExtract table contents from specified PDF pages. Args: start (int, optional): First page to process. Defaults to 0, the first page. end (int, optional): Last page to process. Defaults to None, the last page. pages (list, optional): Range of page indexes. Defaults to None. kwargs (dict, optional): Configuration parameters. Defaults to None. Returns: list: A list of parsed table content. Nr )r.r{r5rrTextendextract_tables)rr/r0r(r4rZtablesrJr r r!rhs  zConverter.extract_tablesc  sdr tdtntdfddtD}t}|j|dtD]}d|d}tj|sAq0 |t |q0j |fidS) zParse and create pages based on page indexes with multi-processing. Reference: https://pymupdf.readthedocs.io/en/latest/faq.html#multiprocessing rr(c s0g|]}|jjd|dfqS)-.json)rrr:cpur0r4prefixrr/r r!r=s z.r rrN) minrrErmap_parse_pages_per_cpurXrYrZrur[r\) rrvr/r0r4Zvectorspoolr<r`r rr!r~s   z(Converter._convert_with_multi_processingcs|\}}}}}}}}t||} | |pt| j}t||t} t| |} | |} | t|| k} | d|t| |d}t|| | }fddt||D}| jD]}d|_qX|D]}d| j|_q`| j di|j di| || dS) aRender a page range of a document. Args: vector (list): A list containing required parameters. * 0 : segment number for current process * 1 : count of CPUs * 2,3: whole pages range to process * 4 : pdf filename * 5 : password for encrypted pdf * 6 : configuration parameters * 7 : json filename storing parsed results r rcsg|]}|qSr r r:Z all_indexesr r!r=z2Converter._parse_pages_per_cpu..TFNr ) rr1rCr&rEintrr(r9r2r3rrr))ZvectorrhrsrSZ pdf_filenamerr4Z json_filenameZcvrQmnZseg_sizeZseg_fromZseg_torHrJr<r rr!rs,    zConverter._parse_pages_per_cpucCs@|r dd|D}|S|p|}tt|t|}t||}|S)zParsing arguments.cSsg|]}t|qSr )r)r;xr r r!r=rz+Converter._page_indexes..)slicerrE)r/r0r(Zpdf_lenZindexesrr r r!rFs zConverter._page_indexescCs d|dS)Nzzr )msgr r r!rAs zConverter._color_output)NNN)rNNr$)NrNN)#__name__ __module__ __qualname____doc__rbytesr"propertyr&r(r)r.rrVr5r1r2r3r\r_dictrgrrrur}rrrr|rr~ staticmethodrrFrAr r r r!rsH     * 1   (0 .  rc@ eZdZdS)rBNrrrr r r r!rBrBc@r)r]Nrr r r r!r]rr])!rror?rXmultiprocessingrrtimertypingrrrrZdocxrZ page.Pager Z page.Pagesr rVrrZ VersionBindryv SystemExit basicConfigINFOrrOrBr]r r r r!s4    &E