o eO @sddlZddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z m Z mZmZddlmZmZmZmZddlmZddlmZdd lmZd Zd Zd Zd Zd dddddddZGdddZ GdddZ!GdddZ"dededefddZ#e$e!j%j&'Z(e$e"j&'Z)dedede fdd Z*dedede+fd!d"Z,efd#ed$ede+fd%d&Z-eefded'ed(ede+fd)d*Z.d-ded$edefd+d,Z/dS).N) itemgetter) AnyDict GeneratorListMatchOptionalPatternTupleUnion)T_numT_obj T_obj_iter T_obj_list)cluster_objects)to_list)objects_to_bboxg@ ffffifflfiflst)uffuffiufflufiuflustuſtc@seZdZdZdeeeeefddfddZ   dd e ed e d e d e de eeff d dZ     ddeeeefde de d e d e d e dee eeffddZ dde d e dee eeffddZdS)TextMapz A TextMap maps each unicode character in the text to an individual `char` object (or, in the case of layout-implied whitespace, `None`). tuplesreturnNcCs ||_dttd||_dS)Nr)rjoinmapr as_stringselfrr&H/var/www/html/venv/lib/python3.10/site-packages/pdfplumber/utils/text.py__init__#szTextMap.__init__rTm main_group return_groups return_charsc Csn|j||||}dd|D}t|\}}} } ||||| | d} |r/|| d<|r5|| d<| S)NcSsg|] \}}|dur|qSNr&).0textcr&r&r' /sz)TextMap.match_to_dict..)r/x0topx1bottomgroupschars)rstartendrgroupr6) r%r)r*r+r,subsetr7r2r3r4r5resultr&r&r' match_to_dict's zTextMap.match_to_dictpatternregexcasec st|tr|dur td|durtd|}n|dur!t|}|dur(tjnd}t||}t|j} t fdd| } fdd| DS) NFzACannot pass a compiled search pattern *and* regex=False together.z@Cannot pass a compiled search pattern *and* case=False together.rcst|Sr-)boolr:strip)r))r*r&r'`sz TextMap.search..csg|] }j|dqS))r+r,r*)r=)r.r)r*r,r+r%r&r'r1asz"TextMap.search..) isinstancer ValueErrorreescapeIcompilefinditerr#filter) r%r>r?r@r+r,r*compiledflagsgenfilteredr&rDr'searchBs&   zTextMap.searchrBcCs |rd}nd}|j|d|ddS)af `strip` is analogous to Python's `str.strip()` method, and returns `text` attributes without their surrounding whitespace. Only relevant when the relevant TextMap is created with `layout` = True Setting `return_chars` to False will exclude the individual character objects from the returned text-line dicts. z *([^\n]+?) *(\n|$)z([^\n]+)rF)r*r,r+)rQ)r%rBr,patr&r&r'extract_text_linesks  zTextMap.extract_text_lines)rTT)TTTTr)TT)__name__ __module__ __qualname____doc__rr strrrr(rintrArrr=r r rQrSr&r&r&r'rsZ"   *rc@seZdZdZdeeeefddfddZddddde e dde ddd f d e d e d e d edede de de de de de de de defddZdS)WordMapz& A WordMap maps words->chars. rrNcCs ||_dSr-)rr$r&r&r'r(s zWordMap.__init__FrTlayout layout_width layout_heightlayout_width_charslayout_height_chars x_density y_densityx_shifty_shift y_tolerance use_text_flow presortedexpand_ligaturesc#Csg}t|js t|S| rtni}|r;|r|rtdntt||}|r-|r,tdntt||}dg|}ng}d}| sC| rF|jnt|jddd}|dd}|d|d }tt |d d| | pg| d D]\}}|r|ddd|| |nd}t t|dkt||}t |D]}t|r|d dd kr||7}| dq||7}d}| s| r|nt|ddd}|D]F\}}|r|d||nd}t t d|t||}|dg|7}||7}|D]}||d|d} | D] }!| |!|f|d7}qqq|r|dg||7}qk|r@||d}"t |"D]}|dkr,||7}| dq!|d dkr@|dd }t|S)a Given a list of (word, chars) tuples (i.e., a WordMap), return a list of (char-text, char) tuples (i.e., a TextMap) that can be used to mimic the structural layout of the text on the page(s), using the following approach: - Sort the words by (doctop, x0) if not already sorted. - Calculate the initial doctop for the starting page. - Cluster the words by doctop (taking `y_tolerance` into account), and iterate through them. - For each cluster, calculate the distance between that doctop and the initial doctop, in points, minus `y_shift`. Divide that distance by `y_density` to calculate the minimum number of newlines that should come before this cluster. Append that number of newlines *minus* the number of newlines already appended, with a minimum of one. - Then for each cluster, iterate through each word in it. Divide each word's x0, minus `x_shift`, by `x_density` to calculate the minimum number of characters that should come before this cluster. Append that number of spaces *minus* the number of characters and spaces already appended, with a minimum of one. Then append the word's text. - At the termination of each line, add more spaces if necessary to mimic `layout_width`. - Finally, add newlines to the end if necessary to mimic to `layout_height`. Note: This approach currently works best for horizontal, left-to-right text, but will display all words regardless of orientation. There is room for improvement in better supporting right-to-left text, as well as vertical text. z;`layout_width` and `layout_width_chars` cannot both be set.z=`layout_height` and `layout_height_chars` cannot both be set.) NrcSt|ddSNrdoctopfloatxr&r&r'rCz$WordMap.to_textmap..keyrkr3cSrirjrlrnr&r&r'rCrp)preserve_order )ruNcSri)Nrr2rlrnr&r&r'rCrpr2rr/N)lenrr LIGATURESrFrYroundsorted enumeratermaxrangeappendminget)#r%r[r\r]r^r_r`rarbrcrdrerfrg_textmap expansions blank_line num_newlineswords_sorted_doctop first_word doctop_startiwsy_distnum_newlines_prependline_lenline_words_sorted_x0wordr7x_distnum_spaces_prependr0lettersletternum_newlines_appendr&r&r' to_textmaps3                zWordMap.to_textmap)rTrUrVrWrr rrr(DEFAULT_X_DENSITYDEFAULT_Y_DENSITYDEFAULT_Y_TOLERANCErAr rYrrr&r&r&r'rZsZ     rZc@seZdZeedddddddf dededededed ed eee d e ee fd efd dZ de de fddZde de defddZdedee ddffddZdedee ddffddZdedeee e fddffddZdedefddZde de fd d!ZdS)" WordExtractorFTN x_tolerancerdkeep_blank_charsrehorizontal_ltr vertical_ttb extra_attrssplit_at_punctuationrgc Csf||_||_||_||_||_||_|durgn||_|dur"tjn|p%d|_ | r.t |_ dSi|_ dS)NTr ) rrdrrerrrstring punctuationrrwr) r%rrdrrerrrrrgr&r&r'r(%s zWordExtractor.__init__ ordered_charsrc st|\}}}}|dd|dd}|dd}|r jr%njr%dnd}dfdd |D||||||||d } jD] } |d| | | <qA| S) Nrrkr3uprightrrtr c3s&|]}j|d|dVqdSr/N)rrr.r0r%r&r' Js z,WordExtractor.merge_chars..)r/r2r4r3rkr5r direction)rrrr!r) r%rr2r3r4r5 doctop_adjrrrrrr&rr' merge_charsBs$  zWordExtractor.merge_chars prev_char curr_charc Cs|dr2|j}|j}|d}|d}|jr"|d}|d}|d} n=|d }|d }|d } n-|j}|j}|d}|d}|jrP|d}|d}|d} n|d }|d }|d } t| |kpo| ||kpo|||kS)aThis method takes several factors into account to determine if `curr_char` represents the beginning of a new word: - Whether the text is "upright" (i.e., non-rotated) - Whether the user has specified that horizontal text runs left-to-right (default) or right-to-left, as represented by self.horizontal_ltr - Whether the user has specified that vertical text the text runs top-to-bottom (default) or bottom-to-top, as represented by self.vertical_ttb - The x0, top, x1, and bottom attributes of prev_char and curr_char - The self.x_tolerance and self.y_tolerance settings. Note: In this case, x/y refer to those directions for non-rotated text. For vertical text, they are flipped. A more accurate terminology might be "*intra*line character distance tolerance" and "*inter*line character distance tolerance" An important note: The *intra*line distance is measured from the *end* of the previous character to the *beginning* of the current character, while the *inter*line distance is measured from the *top* of the previous character to the *top* of the next character. The reasons for this are partly repository-historical, and partly logical, as successive text lines' bounding boxes often overlap slightly (and we don't want that overlap to be interpreted as the two lines being the same line). The upright-ness of the character determines the attributes to compare, while horizontal_ltr/vertical_ttb determine the direction of the comparison. rr3r2r4r5)rrdrrrA) r%rrroyaycyaxbxcxr&r&r'char_begins_new_word[s:'          z"WordExtractor.char_begins_new_wordc#sgdttdttddfffdd }|D]@}|d}|js,|r,|dEdHq||jvr@||EdH|dEdHqrR|d|rR||EdHq|qr_VdSdS)Nnew_charrc3s&rV|durgdS|gdSr-r&)r current_wordr&r'start_next_wordsz:WordExtractor.iter_chars_to_words..start_next_wordr/rt) rrrrrisspacerrr})r%rrcharr/r&rr'iter_chars_to_wordss(     z!WordExtractor.iter_chars_to_wordsr7c csdtdtfdd}tt||dD]@}|dd}|rdnd}t|t||j}|D]&}|r1dnd}t|t|d } |rA|jsLn|jsLt | EdHq+| EdHq+qdS) NrorcSst|d SNr)rYrnr&r&r' upright_keysz2WordExtractor.iter_sort_chars..upright_keyrrrkr2rq) rrYrlistrrdryrrreversed) r%r7rupright_clusterr cluster_key subclustersscsort_keyto_yieldr&r&r'iter_sort_charss      zWordExtractor.iter_sort_charsccsf|jr|n||}tdg|jR}t||}|D]\}}||D] }|||fVq%qdSr)rerrr itertoolsgroupbyrr)r%r7r grouping_key grouped_charskeyvals char_group word_charsr&r&r'iter_extract_tupless  z!WordExtractor.iter_extract_tuplescCstt||Sr-)rZrrr%r7r&r&r'extract_wordmapszWordExtractor.extract_wordmapcCstdd||DS)Ncss|]\}}|VqdSr-r&)r.rrr&r&r'rz.WordExtractor.extract_words..)rrrr&r&r' extract_wordsszWordExtractor.extract_words)rTrUrVDEFAULT_X_TOLERANCErr rArrrXr r(rrrrrrrrr rrZrrr&r&r&r'r$sd    K  #  rr7kwargsrcKstdi||S)Nr&)rr)r7rr&r&r'rsrc sVdditdifddtD}||}|jdifddtD}|S)NrfTci|] }|vr||qSr&r&r.krr&r' z$chars_to_textmap..crr&r&rrr&r'rrr&)updaterWORD_EXTRACTOR_KWARGSrrTEXTMAP_KWARGS)r7r extractorwordmaptextmapr&rr'chars_to_textmaps rc st|}t|dkr dSdrt|fijSdt}td ifddtD}||}t |t d|}d d d |DS) Nrr r[rdcrr&r&rrr&r'rrz extract_text..rkrucss$|] }ddd|DVqdS)rhcss|]}|dVqdSrr&)r.rr&r&r'rrz)extract_text...N)r!)r.liner&r&r'rs"zextract_text..r&) rrvrrr#rrrrrrr!)r7rrdrwordslinesr&rr' extract_text s    r line_chars tolerancecCsVd}d}t|tddD]}|dur|d||kr|d7}|d}||d7}q |S)Nr r2rqrhr4r/)ryr)rrcolllast_x1rr&r&r' collate_linesrrrdcs(t|td|}dfdd|DS)Nrkruc3s|]}t|VqdSr-)rrrr&r'r2sz&extract_text_simple..)rrr!)r7rrd clusteredr&rr'extract_text_simple,srcsRtddddtdddtdttd d fffd d }||}t||jd S) u Removes duplicate chars — those sharing the same text, fontname, size, and positioning (within `tolerance`) as other characters in the set. fontnamesizerr/rkr2r7rNc3slt|d}tj|dD]%\}}tt|tdD]}t|tdD] }t|ddVq&qqdS)Nrqrkr2r)ryrrrrr)r7 sorted_charsgrp grp_chars y_cluster x_clusterrrpos_keyrr&r'yield_unique_chars=s  z(dedupe_chars..yield_unique_charsrq)rrrrryindex)r7rrdedupedr&rr' dedupe_chars5s  $ r)r)0inspectrrGroperatorrtypingrrrrrrr r r _typingr rrr clusteringrgenericrgeometryrrrrrrwrrZrr signaturer parameterskeysrrrrXrrrrr&r&r&r'st ,    c%Q