a g@sddlmZddlZgdZgdZgdZgdZeeeegZe ddd Z gd fe e ed d d Z de e dddZde edddZdS))ListN)uผปแอิืทมใฝu!ฟหกดเ้่าสวงu'ๆไำพะัีรนยบลฃu ๅ/_ภถุึคตจขช)u()ฉฮฺ์?ฒฬฦuฤฆฏโฌ็๋ษศซ.u ๐"ฎฑธํ๊ณฯญฐ,u"+๑๒๓๔ู฿๕๖๗๘๙)z 1234567890-=z qwertyuiop[]\z asdfghjkl;'z zxcvbnm,./)z !@#$%^&*()_+z QWERTYUIOP{}|z ASDFGHJKL:"z ZXCVBNM<>?)charcCsNdD]D}tt|D]2\}}||vr||d|d||fSqqdS)Nr) enumerateALL_CHARACTERSindex)r language_ixixrowr B/usr/local/lib/python3.9/dist-packages/pythainlp/tools/misspell.pysearch_location_of_character*sr)rrrrrr)rrrr)locrkernelc Cs|\}}}}g}|D]r\}} |||| } } d| krBdkrnqd| krntt||d| krnq|||| | |fq|S)Nrr)lenrappend) rrrr is_shiftr posvalid_neighboursZkrksZ_row_posr r rfind_neighbour_locations1s  & rF)rverbosec Cs<t|}|durdSt||}g}dgd|gdgd}|D]\}}}} }zt||d|| }||||d| |df} | dkrd} nH| dkrd } n:| d d krd| d} n | d krd } n| d krd} nWq@||| <Wq@tyYq@Yq@ty4} ztd || WYd} ~ q@d} ~ 00q@|S)Nu▐rrrrrrrrzSomething wrong with: )rrrr IndexError Exceptionprint) rr rrcharsZprinting_locationsr rr rrr er r rfind_misspell_candidatesCs:     r)皙?)sentenceratiocCsttt||t}tjjt||dd}t|}|D].}t||}|durTq:tj|}|||<q:d |S)u Simulate some misspellings of the input sentence. The number of misspelled locations is governed by ratio. :params str sentence: sentence to be misspelled :params float ratio: number of misspells per 100 chars. Defaults to 0.5. :return: sentence containing some misspelled words :rtype: str :Example: :: from pythainlp.tools.misspell import misspell sentence = "ภาษาไทยปรากฏครั้งแรกในพุทธศักราช 1826" misspell(sent, ratio=0.1) # output: ภาษาไทยปรากฏครั้งแรกในกุทธศักราช 1727 F)sizereplaceN) npfloorrZastypeintrandomchoicelistr)join)r+r,Z num_misspellsZ positionsZ misspelledrZpotential_candidates candidater r rmisspellis    r8)F)r*)typingrnumpyr0ZTHAI_CHARACTERS_WITHOUT_SHIFTZTHAI_CHARACTERS_WITH_SHIFTZ ENGLISH_CHARACTERS_WITHOUT_SHIFTZENGLISH_CHARACTERS_WITH_SHIFTrstrrtuplerboolr)floatr8r r r rs"    &