U
    Á½„bD+  ã                   @   s   d dl Z d dlmZ d dlmZmZ d dlmZ d dlm	Z	m
Z
mZmZ ddlmZ ddlmZmZmZmZ dd	lmZ dd
lmZ ddlmZmZmZmZmZ ee
e dœdd„Zee
e dœdd„Z eƒ ee
e dœdd„ƒZ!eƒ ee
e dœdd„ƒZ"eed�eee#e#f dœdd„ƒZ$d,e
e e#e
e dœdd„Z%ee
e e&dœdd „Z'ee
e d!œd"d#„Z(e
e ed$œd%d&„Z)ed'd�d-ee&ee ed)œd*d+„ƒZ*dS ).é    N)ÚIncrementalDecoder)ÚCounterÚOrderedDict)Ú	lru_cache)ÚDictÚListÚOptionalÚTupleé   )ÚFREQUENCIES)ÚKO_NAMESÚLANGUAGE_SUPPORTED_COUNTÚTOO_SMALL_SEQUENCEÚZH_NAMES)Ú is_suspiciously_successive_range)ÚCoherenceMatches)Úis_accentuatedÚis_latinÚis_multi_byte_encodingÚis_unicode_range_secondaryÚunicode_range)Ú	iana_nameÚreturnc                    s¶   t | ƒrtdƒ‚t d | ¡¡j}|dd�}i ‰d‰ tddƒD ]^}| t|gƒ¡}|r>t	|ƒ}|dkrhq>t
|ƒd	kr>|ˆkr„dˆ|< ˆ|  d
7  < ˆ d
7 ‰ q>t‡ ‡fdd„ˆD ƒƒS )zF
    Return associated unicode ranges in a single byte code page.
    z.Function not supported on multi-byte code pagezencodings.{}Úignore)Úerrorsr   é@   éÿ   NFr
   c                    s    g | ]}ˆ| ˆ  d kr|‘qS )g333333Ã?© )Ú.0Úcharacter_range©Úcharacter_countZseen_rangesr   ú9/tmp/pip-unpacked-wheel-2ta4nrol/charset_normalizer/cd.pyÚ
<listcomp>1   s   þz*encoding_unicode_range.<locals>.<listcomp>)r   ÚIOErrorÚ	importlibÚimport_moduleÚformatr   ÚrangeÚdecodeÚbytesr   r   Úsorted)r   ÚdecoderÚpÚiÚchunkr   r   r    r"   Úencoding_unicode_range   s,    

þÿr0   )Úprimary_ranger   c                 C   s>   g }t  ¡ D ],\}}|D ]}t|ƒ| kr| |¡  qqq|S )z>
    Return inferred languages used with a unicode range.
    )r   Úitemsr   Úappend)r1   Ú	languagesÚlanguageÚ
charactersÚ	characterr   r   r"   Úunicode_range_languages9   s    
r8   c                 C   s<   t | ƒ}d}|D ]}d|kr|} q&q|dkr4dgS t|ƒS )zœ
    Single-byte encoding language association. Some code page are heavily linked to particular language(s).
    This function does the correspondence.
    NZLatinúLatin Based)r0   r8   )r   Zunicode_rangesr1   Zspecified_ranger   r   r"   Úencoding_languagesH   s    r:   c                 C   sb   |   d¡s&|   d¡s&|   d¡s&| dkr,dgS |   d¡s>| tkrFddgS |   d	¡sX| tkr^d
gS g S )z›
    Multi-byte encoding language association. Some code page are heavily linked to particular language(s).
    This function does the correspondence.
    Zshift_Ú
iso2022_jpZeuc_jÚcp932ÚJapaneseÚgbÚChinesezClassical ChineseÚ
iso2022_krÚKorean)Ú
startswithr   r   )r   r   r   r"   Úmb_encoding_languages\   s    ÿþýürC   )Úmaxsize)r5   r   c                 C   sB   d}d}t |  D ](}|s$t|ƒr$d}|rt|ƒdkrd}q||fS )zg
    Determine main aspects from a supported language if it contains accents and if is pure Latin.
    FT)r   r   r   )r5   Útarget_have_accentsÚtarget_pure_latinr7   r   r   r"   Úget_target_featuresq   s    rG   F)r6   Úignore_non_latinr   c                    s¬   g }t dd„ ˆ D ƒƒ}t ¡ D ]l\}}t|ƒ\}}|r@|dkr@q|dkrN|rNqt|ƒ}t‡ fdd„|D ƒƒ}	|	| }
|
dkr| ||
f¡ qt|dd„ d	d
�}dd„ |D ƒS )zE
    Return associated languages associated to given characters.
    c                 s   s   | ]}t |ƒV  qd S ©N)r   )r   r7   r   r   r"   Ú	<genexpr>Š   s     z%alphabet_languages.<locals>.<genexpr>Fc                    s   g | ]}|ˆ kr|‘qS r   r   )r   Úc©r6   r   r"   r#   ™   s      z&alphabet_languages.<locals>.<listcomp>gš™™™™™É?c                 S   s   | d S ©Nr
   r   ©Úxr   r   r"   Ú<lambda>¡   ó    z$alphabet_languages.<locals>.<lambda>T©ÚkeyÚreversec                 S   s   g | ]}|d  ‘qS )r   r   )r   Zcompatible_languager   r   r"   r#   £   s     )Úanyr   r2   rG   Úlenr3   r+   )r6   rH   r4   Zsource_have_accentsr5   Zlanguage_charactersrE   rF   r!   Zcharacter_match_countÚratior   rL   r"   Úalphabet_languages‚   s"    ÿrX   )r5   Úordered_charactersr   c                    s0  | t krtd | ¡ƒ‚d}|D �]}|t |  kr2qt |  dt |   |¡… }t |  t |   |¡d… }|d| |¡… ‰|| |¡d… ‰ ‡fdd„|D ƒ d¡}‡ fdd„|D ƒ d¡}t|ƒdkrØ|dkrØ|d	7 }qt|ƒdkrö|dkrö|d	7 }q|t|ƒ d
k�s|t|ƒ d
kr|d	7 }qq|t|ƒ S )aN  
    Determine if a ordered characters list (by occurrence from most appearance to rarest) match a particular language.
    The result is a ratio between 0. (absolutely no correspondence) and 1. (near perfect fit).
    Beware that is function is not strict on the match in order to ease the detection. (Meaning close match is 1.)
    z{} not availabler   Nc                    s   g | ]}|ˆ k‘qS r   r   ©r   Úe)Úcharacters_beforer   r"   r#   Å   s    z1characters_popularity_compare.<locals>.<listcomp>Tc                    s   g | ]}|ˆ k‘qS r   r   rZ   )Úcharacters_afterr   r"   r#   Ê   s    é   r
   gš™™™™™Ù?)r   Ú
ValueErrorr'   ÚindexÚcountrV   )r5   rY   Zcharacter_approved_countr7   Zcharacters_before_sourceZcharacters_after_sourceZbefore_match_countZafter_match_countr   )r]   r\   r"   Úcharacters_popularity_compare¦   sV    
 ÿÿ ÿÿ
ÿý
ÿýÿþrb   )Údecoded_sequencer   c                 C   s–   t ƒ }| D ]~}| ¡ dkrq
t|ƒ}|dkr.q
d}|D ]}t||ƒdkr6|} qRq6|dkr^|}||krt| ¡ ||< q
||  | ¡ 7  < q
t| ¡ ƒS )a  
    Given a decoded text sequence, return a list of str. Unicode range / alphabet separation.
    Ex. a text containing English/Latin with a bit a Hebrew will return two items in the resulting list;
    One containing the latin letters and the other hebrew.
    FN)r   Úisalphar   r   ÚlowerÚlistÚvalues)rc   Zlayersr7   r   Zlayer_target_rangeZdiscovered_ranger   r   r"   Úalpha_unicode_splitâ   s,    ÿÿrh   )Úresultsr   c                    sh   t ƒ ‰ | D ]8}|D ].}|\}}|ˆ kr2|gˆ |< qˆ |  |¡ qq
‡ fdd„ˆ D ƒ}t|dd„ dd�S )z‹
    This function merge results previously given by the function coherence_ratio.
    The return type is the same as coherence_ratio.
    c                    s.   g | ]&}|t tˆ | ƒtˆ | ƒ d ƒf‘qS )r^   )ÚroundÚsumrV   )r   r5   ©Zper_language_ratiosr   r"   r#     s   úþþz*merge_coherence_ratios.<locals>.<listcomp>c                 S   s   | d S rM   r   rN   r   r   r"   rP   "  rQ   z(merge_coherence_ratios.<locals>.<lambda>TrR   )r   r3   r+   )ri   ÚresultZ
sub_resultr5   rW   Úmerger   rl   r"   Úmerge_coherence_ratios	  s    

øro   i   çš™™™™™¹?)rc   Ú	thresholdÚlg_inclusionr   c                 C   sì   g }d}d}|dk	r|  d¡ng }d|kr8d}| d¡ t| ƒD ]˜}t|ƒ}| ¡ }	tdd„ |	D ƒƒ}
|
tkrpq@d	d
„ |	D ƒ}|pŠt||ƒD ]J}t||ƒ}||k r¦qŒn|dkr¶|d7 }| 	|t
|dƒf¡ |dkrŒ q@qŒq@t|dd„ dd�S )z¨
    Detect ANY language that can be identified in given sequence. The sequence will be analysed by layers.
    A layer = Character extraction by alphabets/ranges.
    Fr   Nú,r9   Tc                 s   s   | ]\}}|V  qd S rI   r   ©r   rK   Úor   r   r"   rJ   <  s     z"coherence_ratio.<locals>.<genexpr>c                 S   s   g | ]\}}|‘qS r   r   rt   r   r   r"   r#   A  s     z#coherence_ratio.<locals>.<listcomp>gš™™™™™é?r
   r^   é   c                 S   s   | d S rM   r   rN   r   r   r"   rP   T  rQ   z!coherence_ratio.<locals>.<lambda>rR   )ÚsplitÚremoverh   r   Úmost_commonrk   r   rX   rb   r3   rj   r+   )rc   rq   rr   ri   rH   Zsufficient_match_countZlg_inclusion_listZlayerZsequence_frequenciesry   r!   Zpopular_character_orderedr5   rW   r   r   r"   Úcoherence_ratio%  s<    	
 ÿ ÿrz   )F)rp   N)+r%   Úcodecsr   Úcollectionsr   r   Ú	functoolsr   Útypingr   r   r   r	   Zassetsr   Zconstantr   r   r   r   Zmdr   Úmodelsr   Úutilsr   r   r   r   r   Ústrr0   r8   r:   rC   ÚboolrG   rX   Úfloatrb   rh   ro   rz   r   r   r   r"   Ú<module>   sL   	% ÿ þ% þ<'   ÿ  þ