U M{j;@sddlZddlZddlZddlZddlZddlmZmZmZm Z ddl Z ddl m Z m Z mZddZddZd2d d Zd d ZddZd3ddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Z d+d,Z!d-d.Z"d/d0Z#e$d1kre#dS)4N)DictListSetTuple)TEXT_INHIBIT_SPACESTEXT_PRESERVE_LIGATURESTEXT_PRESERVE_WHITESPACEcCsd|ddS)Nz %s K-)center)xr L/var/www/html/bwave/python/venv/lib/python3.8/site-packages/fitz/__main__.pyrcCs|d}|d}|dkr"||Sdd}t||}t||}|j|jkrt|j|jkrfdkrtnn |jdkstd||ft|d}||St|}||jd}}||S)zReturn image for a given XREF.rcSs"|jjdkr|Sttj|}|S)N) colorspacenfitzPixmapcsRGB)pixZtpixr r rgetimage s zrecoverpix..getimagez&Warning: unsupported /SMask %i for %i:N) Z extract_imagerrZirectalpharprintZ set_alphaZsamples)docitemr srZpix1pix2rr r r recoverpixs    2  r FTcCst|}|js"|dkr"tdd}|js0|S|rn||}|sLtd|dkr~t|dkrfddndntd |j|S) z!Open and authenticate a document.Tz$this command supports PDF files onlyzauthentication unsuccessfulzauthenticated as %sowneruserz'%s' requires a password) ropenZis_pdfsysexitZ needs_passZ authenticatername)filenamepasswordshowpdfrrcr r r open_file=s    r.cCsJtdd|Dd}|D]"\}}d|||f}t|q"dS)zPrint a Python dictionary.cSsg|] }t|qSr )len).0kr r r Rszprint_dict..rz%s: %sN)maxkeysitemsrjustr)rlr1vmsgr r r print_dictPs  r:cCstd|||}t|||r~|}z(|dd}||}|drVd}Wnd}YnXtd|tdtdd S) zPrint an object given by XREF number. Simulate the PDF source in "pretty" format. For a stream also print its size. z%i 0 objz/Lengthrz0 Runknownzstream ...%s bytesZ endstreamZendobjN)rZ xref_objectZxref_is_streamsplitindexendswith)rxrefZxref_strtempidxsizer r r print_xrefYs      rCpagec Csrt|d}|d|dd}|d}g}t|D]4\}}|d}|rt|} d| krl|krnn|t|q6td||fq6z"|d\} } t| } t| } Wntd||fYnXd| kr|kr nnd| kr|ksntd||f| | kr2|| q6| | krT|t t | | d7}q6|t t | | dd 7}q6|S) aKTransform a page / xref specification into a list of integers. Args ---- rlist: (str) the specification limit: maximum number, i.e. number of pages, number of objects what: a string to be used in error messages Returns ------- A list of integers representing the specification. rN ,zbad %s specification at item %ir z%bad %s range specification at item %ir!) strreplacer< enumerate isdecimalintappendr&r'listrange) ZrlistlimitwhatrEZ rlist_arrZout_listseqrrii1i2r r rget_listps6   4   rWc Cst|j|jd}tj|jd}d}|dkr:|d}d}t|d}|j}td|j|j | d|||d|d f|j }|d kr| }td ||d krd ndf| }|d krtd|t|jrttd|}t||t|jrttdt|jt|jr^ttdt|j| dd}|D]}t||tqF|jrttdt|j|j d} | D]4} | d}||}td| t||tq|jrttdt|t|dS)NTiZKBiMBrz7'%s', pages: %i, objects: %i, %g %s, %s, encryption: %sformat encryptionrz5document contains %i root form fields and is %ssignedznot rGz#document contains %i embedded filesz PDF catalogz PDF metadatazobject informationr?)rRzpage informationzPage %i:z PDF trailer)r.inputr*ospathgetsizeroundmetadatar page_countZ xref_lengthZ is_form_pdfZ get_sigflagsZ embfile_countcatalogmycenterZ pdf_catalogrCr:ZxrefsrWpagesZ page_xreftrailerZ pdf_trailerclose) argsrrBflagmetarrr?Zxreflpagelpnor r rr+sz                  r+c Cst|j|jdd}|j}d|}|js`|j|j|j|j |j |j |j |j ||j|j|jd dSt|j|jd}t}|D]}|d}|j|||dq~|j|j|j|j |j |j |j |j ||j|j|jd ||dS)NTr,Zkeepnonezrc4-40zrc4-128zaes-128zaes-256) garbagedeflateprettycleanasciilinearrZZowner_pwZuser_pwZ permissionsrZ from_pageZto_page)r.r\r*rZr=resaveoutputrpcompressrrsanitizertrur#r$Z permissionrWrbrr% insert_pdfrg)rhrrZZencryptreZoutdocrlrr r rrssR  rsc Cs|j}t}|D]}|d}t|dkr4|dnd}t|d|dd}d|dd}|rtd|dd|jd}nt d|jd}|D]} |j || d| ddq| q|j |j d dd | dS) z&Join pages from several PDF documents.rHrNrTrmr"rvr)rprq)r\rr%r<r/r.joinrWrbrPr{rgrwrx) rhZdoc_listrZsrc_itemZsrc_listr*srcreZ page_listrTr r rdoc_join s   r~c CsJt|j|jdd}|s6|jr,|j|jkr6tdt|j|j}|j rTt |j nt }t | }|r~||kstdn|}|std|t | @}|rtdt ||D]J}| |}||}|j|||d|d|d d td ||j fq||jr6|j|jkr6|j|jd d n||dS)z!Copy embedded files between PDFs.Trmcannot save PDF incrementallyz%not all names are contained in sourceznothing to copyz2following names already exist in receiving PDF: %sr) ufilenamedescr)rrzcopied entry '%s' from '%s'r[rpN)r.r\r*can_save_incrementallyrxr&r'sourceZ pwdsourcer(set embfile_namesrI embfile_info embfile_get embfile_addrrgrwsaveIncr) rhrr}namesZ src_namesZ intersectrinfoZbuffr r r embedded_copy"sF       rcCst|j|jdd}|s6|jr,|j|jkr6tdz||jWn$t k rjtd|jYnX|jr~|j|jkr| n|j |jdd| dS)zDelete an embedded file entry.Trmrno such embedded file '%s'rrN) r.r\r*rrxr&r' embfile_delr( ValueErrorZ save_incrrwrg)rhrr r r embedded_delJs   rcCst|j|jdd}z||j}||j}Wn$tk rRtd|jYnX|j r`|j n|d}t |d}| || t d|j|f| dS)z&Retrieve contents of an embedded file.Trmrr)wbzsaved entry '%s' as '%s'N)r.r\r*rr(rrr&r'rxr%writergr)rhrstreamdr)rxr r r embedded_get]s   rcCst|j|jdd}|s:|jdks0|j|jkr:tdz ||jtd|jWn YnXt j |j rt j |j std|j t |j d}|j }|}|js|}n|j}|j|j||||d|jr|j|jkr|n|j|jd d |dS) zInsert a new embedded file.TrmNrzentry '%s' already existszno such file '%s'rbrr[r)r.r\r*rrxr&r'rr(r]r^existsisfiler%readrrrrwrgrhrrr)rrr r r embedded_addms>    rcCs>t|j|jdd}|s:|jdks0|j|jkr:tdz||jWntd|jYnX|j dk rt j |j rt j |j rt |j d}nd}|jr|j}nd}|jr|j}n|jr|j}nd}|jr|j}nd}|j|j||||d|jdks|j|jkr"|n|j|jdd |dS) z0Update contents or metadata of an embedded file.TrmNrrrrr[r)r.r\r*rrxr&r'rr(r^r]rrr%rr)rrZ embfile_updrrwrgrr r r embedded_updsR     rcCst|j|jdd}|}|jdk r|j|kr@td|jnJttdt|t|dkr`dndftt | |jtdS|std |jdSt|dkrd |jt|f}n d |j}t|t|D]2}|j st|q| |}t | |tq| dS) zList embedded files.TrmNrz!printing 1 of %i embedded file%s:rrrGz'%s' contains no embedded filesz-'%s' contains the following %i embedded filesz)'%s' contains the following embedded file) r.r\r*rr(r&r'rr/r:rdetailrg)rhrrr9r(_r r r embedded_lists>     rc Csf|js|jstdt|j|jdd}|jrBt|j|j d}nt d|j d}|j sht j t j}n,|j }t j |rt j |std|t}t}|D]z}|jrL||d}|D]}|d} | |kr|| || \} } } } | dks| sqt j || dd d | d | }t|d }|| |d } q|jr||d}|D]}|d} | |krd|| t||}t|tkr|d } |d}t j |d| | f}t|d }|||nt%dt&||f|jrZt%dt&||f|d S)z)Extract images and / or fonts from a PDF.z"neither fonts nor images requestedTrmrz"output directory %s does not existrzn/arFr .rNextimagez img-%i.%sz img-%i.pngrzsaved %i fonts to '%s'zsaved %i images to '%s')'ZfontsZimagesr&r'r.r\r*rerWrbrPrxr]r^abspathcurdirrisdirrZget_page_fontsaddZ extract_fontr|rJr%rrgZget_page_imagesr typedictrrrrrrwrr/)rhrreZout_dirZ font_xrefsZ image_xrefsrlZitemlistrr?ZfontnamerrbufferoutnameoutfilerZimgdatarr r rextract_objectssp              rc CsX|rdntdg}|jd|d}|s6|s2||dS||jddd||dS)N textflagsutf8 surrogatepasserrors)bytesget_textrencode) rDtextoutGRIDfontsize noformfeed skip_emptyreoprr r r page_simple's  rc Csz|rdntdg}|jd|d}|gkr:|s6||dS|jddd|D]} || djd d d qN||dS) NrrblocksrcSs|d|dfS)Nr[rr )br r rr:rz page_blocksort..keyrrrr)rrrsortr) rDrrrrrrrrrr r rpage_blocksort3s  rc!s>|rdntdg}ttttddd}tttddd} tttjd fd d } ttd d ddd} |jd|dd} | | |\} }}}}| gkr|s| |dS| ||}| j dddi}| D]8}|\}}}}|||}| |g}| ||||<qt |}| ||}i}|D]f}||}t|}|dkrZd||<q0dd|D}| t|}||kr|}|d||<q0||d|d|t|d}|d}| d|D]\}||kr| d||7}q| ||||||} | | djdd d!||}q| |dS)"Nrr)valuesvaluereturncSs0t||}|r||dStd||fdS)zFind the right row coordinate. Args: values: (list) y-coordinates of rows. value: (int) lookup for this value (y-origin of char). Returns: y-ccordinate of appropriate line for value. rzLine for %g not found in %sN)bisect bisect_right RuntimeError)rrrTr r rfind_line_indexEs  z$page_layout..find_line_index)rowsrcSsJt|}||dg}|ddD]}||d|kr&||q&|S)Nrrr!)rOrrN)rrZnrowshr r r curate_rowsTs  z page_layout..curate_rows)rrDc st}|jj}|jj}|}|}d}g}|D]} | dD]|} | ddkrLq8| d\} } } }|dks8| |jjkrrq8|| }||kr|}| dD]$}|dkrq|dD]}|d\} }} }| | }|d \}}tt|}|||d }||kr |d kr |}|| kr| }|dkr|gkr|d \}}}}||kr|td krb||}n,|dkrvtd}n|dkrtd}n|}||||f|d <q|||||fqqq8q*|||||fS)Nrlinesdir)rrZbboxspansrBcharsorigincrFr!rTr7) rrectwidthheightrMr`rchrrN)rrDrZ page_widthZ page_height rowheightleftrightrblocklineZx0Zy0x1y1rspanrrcwidthoxoychZold_chold_oxZold_oyZ old_cwidthligr joinligaturer rprocess_blocks]sZ            z#page_layout..process_blocks)rrcSst|dkrtdS|dkr tdS|dkr0tdS|dkr@tdS|d krPtd S|d kr`td S|d krptdS|S)zReturn ligature character for a given pair / triple of characters. Args: lig: (str) 2/3 characters, e.g. "ff" Returns: Ligature, e.g. "ff" -> chr(0xFB00) ffrfiifliffirfflrftisti)r)rr r rrs z!page_layout..joinligaturecSsd}d}d}d}|tjkr&td||D]}|\} } } } | |} | | } || krd| || dkrdq*| dkr~|| | dkr~q*| }| ||kr|| 7}| }| }q*| dkrq*t| |t|}| |kr|dkr|d|7}|| 7}| }| }q*|S)aProduce the text of one output line. Args: left: (float) left most coordinate used on page slot: (float) avg width of one character in any font in use. minslot: (float) min width for the characters in this line. chars: (list[tuple]) characters of this line. Returns: text: (str) text string for this line rGrz%program error: minslot too small = %gg?rFg?r)rZEPSILONrrMr/rstrip)rslotZminslotlcharsrZold_charZold_x1rrcharrrrrdeltar r r make_textlines:      z"page_layout..make_textlineZrawdictrrcSs|dS)Nrr )rr r rrrzpage_layout..rr"rcSsg|] }|dqS)r[r )r0rr r rr2 szpage_layout..rr!g333333? rrr)rrrMrrrZPagerIrrrgetrNrOr4r/ statisticsZmedianr)!rDrrrrrrrrrrrrrrrrrrrrryrr4rZminslotsr1ZccountwidthsZ this_slotZrowposrr rr page_layoutAs` 46            $     rc Cst|j|jdd}t|j|jd}|j}|dkrLtj |j \}}|d}t |d}t t B}|jrl|t N}|jrz|t N}|jr|tN}tttd}|D]4} || d} ||j| ||j|j|j|j|dq|dS)NFrmrz.txtrsimplerlayoutr)r.r\r*rWrerbrxr]r^splitextr(r%rrZ convert_whiteZ noligaturesZ extra_spacesrrrrmodegridrrrrg) rhrrkrxr)rrrfuncrlrDr r rgettext"s<   r c Cstjdtdd}|jddd}|jdtdd }|jd td d |jd dd|jdddd|jdddd|jdddd|jdtdd |jdtdd |jtd|jdtdd }|jd td d |jdtd d |jd dd|jd!d"d#d$d%|jd&td'd |jd(td)d |jd*t d+t d,d-d.|jd/dd0d1d2|jd3dd0d4d2|jd5dd0d6d2|jd7t d8d9d:|jd;dd0dd2|jdd?d|jt d|jd@tdAdBdC}|jd dDdEdF|jdGdHdIdJ|jt d|jdKtdLd }|jd td d |jdMddNd|jdOddPd|jdGdQd|jd dd|jdtdRd |jt d|jdStdTd }|jd d d|jdUdVd|jdWddXd|jd dd|jtd|jdYtdZd }|jd d d|jd dd|jdGd[d|jdUdHd\dJ|jd]dHd^dJ|jd_d`d|jtd|jdatdbd }|jd d d|jd dd|jdGd[d|jdUdHdcdJ|jtd|jddtdedfdC}|jd d d|jdUdHdgdJ|jd dd|jdGdhd|jd]did|jdjdkd|jdldmd|jd_dnd|jtd|jdotdpd } | jd td d | jdUdHdgdJ| jd dd| jdGdqd| jtd|jdrtdsd } | jd tdtd | jd dud| jdGdvd| jdwdHdxdJ| jdydzd| jdUdDd{dF| jtd|jd|td}d } | jd td~d | jd dd| jdtdddd.| jdtddd| jdddd0d| jdddd0d| jdddd0d| jdddd0d| jdddd0d| jdGdd| jdtddd| jdtddd| jtd|} t| ds |n | | dS)zDefine command configurations.rzBasic PyMuPDF Functions)prog descriptionZ Subcommandsz/Enter 'command -h' for subcommand specific help)titlehelpr+zdisplay PDF information)r r\z PDF filename)rrz -passwordr*)rz-catalog store_truezshow PDF catalog)actionrz-trailerzshow PDF trailerz -metadatazshow PDF metadataz-xrefsz&show selected objects, format: 1,5-7,Nz-pagesz'show selected pages, format: 1,5-7,50-N)r rsz.optimize PDF, or create sub-PDF if pages givenrxzoutput PDF filenamez -encryptionzencryption methodrnro)rchoicesdefaultz-ownerzowner passwordz-userz user passwordz-garbagezgarbage collection levelr)rrrrz -compressFzcompress (deflate) output)rrrz-asciizASCII encode binary dataz-linearzformat for fast web displayz -permissionr!zinteger with permission levels)rrrz -sanitizezsanitize / clean contentsz-prettyzprettify PDF structurez/output selected pages pages, format: 1,5-7,50-Nr|zjoin PDF documentsz3specify each input as 'filename[,password[,pages]]')r epilog*zinput filenames)nargsrz-outputTzoutput filename)requiredrextractz extract images and fonts to diskz-imageszextract imagesz-fontsz extract fontsz-folder to receive output, defaults to currentz-consider these pages only, format: 1,5-7,50-Nz embed-infozlist embedded filesz-namezif given, report only this onez-detailzdetail informationz embed-addzadd embedded filez-output PDF filename, incremental save if nonezname of new entryz-pathzpath to data for new entryz-desczdescription of new entryz embed-delzdelete embedded filezname of entry to deletez embed-updzupdate embedded filez*except '-name' all parameters are optionalz name of entryz-Output PDF filename, incremental save if nonezpath to new data for entryz -filenameznew filename to store in entryz -ufilenamez&new unicode filename to store in entryz!new description to store in entryz embed-extractzextract embedded file to diskz'output filename, default is stored namez embed-copyz copy embedded files between PDFszPDF to receive embedded fileszpassword of inputz2output PDF, incremental save to 'input' if omittedz-sourcezcopy embedded files from herez -pwdsourcezpassword of 'source' PDFzrestrict copy to these entriesr z(extract text in various formatting modeszinput document filenamezpassword for input documentz-modez-mode: simple, block sort, or layout (default)rrz select pages, format: 1,5-7,50-Nz1-N)rrrz -noligaturesz*expand ligature characters (default False))rrrz-convert-whitez6convert whitespace characters to white (default False)z -extra-spacesz%fill gaps with spaces (default False)z -noformfeedz-write linefeeds, no formfeeds (default False)z -skip-emptyz+suppress pages with no text (default False)z3store text in this file (default inputfilename.txt)z-gridz+merge lines if closer than this (default 2)r"z -fontsizez4only include text with a larger fontsize (default 3)r[r N)argparseArgumentParserrdadd_subparsers add_parser add_argumentrI set_defaultsr+rMrPrsr~rrrrrrrfloatr  parse_argshasattr print_helpr ) parserZsubpsZps_showZps_cleanZps_joinZ ps_extractZ ps_embed_addZ ps_embed_delZ ps_embed_updZps_embed_extractZ ps_embed_copyZ ps_gettextrhr r rmainEs"             r$__main__)FT)rD)%rrr]r&rtypingrrrrrZ fitz.fitzrrrrdr r.r:rCrWr+rsr~rrrrrrrrrrr r$__name__r r r rsB$   /?/(!1$D b#,