U hr2eH9 @s@ddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddl m!Z!ddl m"Z"ddl m#Z#ddl$m%Z%dZ&eddk Z'e'r:ddl(m)Z)eddk Z*e*rTddl+Z,dZ-e.dZ/dZ0dddd d!d"d#d$d%d&h Z1d'e2fd(e2fd)e2fd*e3fd+e4fd,e3fd-Z5e#d.Z6e#d/Z7Gd0d1d1Z8Gd2d3d3e9Z:Gd4d5d5e;ZGd:d;d;e9Z?dd?ZAd@dAZBdBdCZCdDdEZDdFdGZEedHdIZFdqdKdLZGdrdNdOZHdsdQdRZIdSdTZJdUdVZKdWdXZLeBdtdYdZZMeBd[d\ZNddMde8jOdfd]d^ZPdud`daZQdvdbdcZRddMde8jOdfdddeZSdwdfdgZTddMde8jOddfdhdiZUdjdMde8jOdfdkdlZVdmdnZWeXdokr= 3.05 requiredr*r-r/r'r(r,aszTSVNotSupported.__init__r1r'r'r/r(r:`sr:cseZdZfddZZS)ALTONotSupportedcstddS)Nz6ALTO output not supported. Tesseract >= 4.1.0 requiredr*r-r/r'r(r,hszALTONotSupported.__init__r1r'r'r/r(r;gsr;cCs`|zBz|dWn.tk r4tdYntk rFYnXW5|||_XdS)N) terminatekill returncodewait TypeErrorr Exception)processcoder'r'r(r>ns  r>ccszb|s|dVWLdSz|j|d\}}|VWn(tjk r^t|dtdYnXW5|j|j|jXdS)Nr<)timeoutzTesseract process timeout) stdinclosestdoutstderr communicate subprocessTimeoutExpiredr> RuntimeError)procseconds_ error_stringr'r'r(timeout_manager{s    rScs tfdd_S)Ncsjkr||_jSr4)_result)r7kwargsfuncwrapperr'r(rXs  zrun_once..wrapper)rrT)rWr'rVr(run_oncesrYcCs"ddd|tDS)N css|] }|VqdSr4r'.0liner'r'r( szget_errors..)joindecodeDEFAULT_ENCODING splitlinesstrip)rRr'r'r( get_errorss  rdc Cs\t|r|dn|D]@}z t|Wqtk rT}z|jtkrDW5d}~XYqXqdS)z5Tries to remove temp files by filename wildcard path.*N)rr OSErrorerrnor) temp_namefilenameer'r'r(cleanups   rkcCstrt|trt|}t|tjs,td|js6dn|j}|tkrLtdd|krt t |j d}| |d| d|}||_||fS)NzUnsupported image objectrzUnsupported image format/typeA)rmrm)rr)numpy_installed isinstancerrZ fromarrayrAformatSUPPORTED_FORMATSZgetbandsnewRGB_MODEsizeZpasteZ getchannel)image extensionZ backgroundr'r'r(prepares   rwc csztdddv}t|trD|jttt|fVW5QRWNdSt|\}}|jdt |}|j ||j d|j|fVW5QRXW5t|jXdS)NZtess_F)prefixdelete_input)rp) rknamerrostrrr r rwr saverp)rufrvZinput_file_namer'r'r(r}s  r}TcCsftjtjdtd}ttdrHt|d<|djtjO_tj|d_|rXtj|d<n tj |d<|S)N)rGrJ startupinfoenv STARTUPINFOrrI) rLPIPErhasattrrZdwFlagsZSTARTF_USESHOWWINDOWZSW_HIDEZ wShowWindowDEVNULL)Zinclude_stdoutrUr'r'r(subprocess_argss     rc Csg}tjds*|dkr*|ddt|f7}|t||f7}|dk rL|d|f7}|r^|t|7}|rt|dkrt||ztj |ft }Wn6t k r} z| j t krntW5d} ~ XYnXt||} |jrt|jt| W5QRXdS)Nwin32rnicez-n-l>boxtsvosdxml)sysplatform startswithr|r9shlexsplitappendrLPopenrrfrgrr8rSr?r3rd) input_filenameoutput_filename_baservlangconfigrrEcmd_argsrOrjrRr'r'r( run_tesseracts&      rFc Cst|\}}|||||||d} tf| | dt|} t| dL} |rp| W5QRW5QRS| tW5QRW5QRSQRXW5QRXdS)N)rrrvrrrrErrb)r}rr openreadr`ra) rurvrrrrEZ return_bytesrhrrUriZ output_filer'r'r(run_and_get_output s    rc si}fdd|dD}t|dkr0|S|d}t|}t|d|kr`|dd|dkrp||7}t|D]|\}}t||<|D]d} t| |krq||krztt| |} Wqt k r| |} YqXn| |} ||| qqx|S)Ncsg|]}|qSr'r)r\rowcell_delimiterr'r( *sz file_to_dict.. rrFr) rcrlenpopr enumeratelistintfloat ValueError) rrZ str_col_idxresultZrowsheaderlengthiheadrvalr'rr( file_to_dict(s.    rcCsB|tkr|S|tkr>zt|WdStk r<YdSXdS)NTF)risdigitrr)r_typer'r'r(is_validKsrcCsdddd|dDDS)NcSsXi|]P}t|dkrt|dt|ddrt|ddt|dd|dqS)rr<r)rrOSD_KEYS)r\kvr'r'r( Zs  zosd_to_dict..css|]}|dVqdS): Nrr[r'r'r(r^\szosd_to_dict..rr)rr'r'r( osd_to_dictYsrcCstdg}|r|t|7}ztj|tjtjd}Wntk rLtYnX|j dkr^tg}|j r|j t t D] }|}t|rz||qz|S)Nz --list-langs)rIrJ)rr<)r9rrrLrunrSTDOUTrfr8r?rIr`rar rc LANG_PATTERNmatchr)rrr languagesr]rr'r'r( get_languagesas(     rc Csztjtdgtjttjd}Wntk r8tYnX|t }| t j dd d^}}| d^}}zt|}|tkstWn(ttfk rtd|dYnX|S) z9 Returns Version object of the Tesseract version z --version)rJrrG NrZ-zInvalid tesseract version: "")rL check_outputr9rrrrfr8r`ralstripr printable partitionrTESSERACT_MIN_VERSIONAssertionErrorr SystemExit)outputZ raw_versionZ str_versionrQversionr'r'r(get_tesseract_version~s$   rcsD|d||||gtjfddtjfddtjfddi|S)zS Returns the result of a Tesseract OCR run on the provided image to string txtcstdgSNTrr'r7r'r(z!image_to_string..cs dtiS)Ntextrr'rr'r(rrcstSr4rr'rr'r(rr)rr#r%r&rurrr output_typerEr'rr(image_to_strings    rpdfcCs0|dkrtd|||||||dg}t|S)zU Returns the result of a Tesseract OCR run on the provided image to pdf/hocr >rhocrzUnsupported extension: T)rr)rurrrrvrEr7r'r'r(image_to_pdf_or_hocrs rcCs8ttkrtd|}|d||||dg}t|S)zU Returns the result of a Tesseract OCR run on the provided image to ALTO XML z-c tessedit_create_alto=1 rT)rTESSERACT_ALTO_VERSIONr;rcr)rurrrrEr7r'r'r(image_to_alto_xmls rcsR|d}|d||||gtjfddtjfddtjfddi|S)zR Returns string containing recognized characters and their box boundaries z batch.nochop makeboxrcstdgSrrr'rr'r(rrz image_to_boxes..cstdtddS)Nz char left bottom right top page rZrrrr'rr'r(rs cstSr4rr'rr'r(rrrcrr#r%r&rr'rr(image_to_boxess    rc CsRts ttdd}z||Wnttfk r:YnXtjtt |f|S)N )quotingsep) pandas_installedr)rupdaterArpdZread_csvrr)r7rrUr'r'r(get_pandas_outputs rc srttkrtd|}|d||||gtjfddtjfddtjfddtjfddi|S)zt Returns string containing box boundaries, confidences, and other information. Requires Tesseract 3.05+ z-c tessedit_create_tsv=1 rcstdgSrrr'rr'r(rrzimage_to_data..cstdgSr)rr'r7 pandas_configr'r(rscsttddS)NrrFrr'rr'r(rrcstSr4rr'rr'r(rr) rrr:rcrr#r$r%r&)rurrrrrErr'rr( image_to_datas     rrcsRd|}|d||||gtjfddtjfddtjfddi|S)zN Returns string containing the orientation and script detection (OSD) z--psm 0 rcstdgSrrr'rr'r(r)rzimage_to_osd..cs ttSr4)rrr'rr'r(r*rcstSr4rr'rr'r(r+rrrr'rr( image_to_osds    rc Csttjdkr tjdd}}nFttjdkrTtjddkrTtjdtjd}}ntdtjddSz*t|}tt||dW5QRXWntk r}z"tt |d tjdWYdSd}~XYnHt k r}z(tt |j d |tjdWYdSd}~XYnXdS) Nrr<rz(Usage: pytesseract [-l lang] input_file )file)rrr) rrargvprintrJrrrr8r|rftyper )ririmgrjr'r'r(main/s r__main__)N)T)rrr)rNrrrF)r)Nrrrr)Nrrr)N)ZrerrrLr contextlibrcsvrrgr functoolsrglobriorosrr r r os.pathr r rpkgutilrtempfilertimerZpackaging.versionrrrZPILrr9rnrrrrrracompilerrsrqrrr|rrrrEnvironmentErrorr)rNr3r8r:r;r>rSrYrdrkrwr}rrrrrrrrr&rrrrrrrrr exitr'r'r'r(s                                ( #        !