o
    ì6Wj2  ã                   @  s¢   d dl mZ d dlmZ d dlmZ d dlmZmZm	Z	m
Z
 ddlmZmZ ddlmZmZmZ G dd	„ d	ƒZG d
d„ dƒZe
eef Ze	e ZG dd„ dƒZdS )é    )Úannotations)Úaliases)Úsub)ÚAnyÚIteratorÚListÚTupleé   )ÚRE_POSSIBLE_ENCODING_INDICATIONÚTOO_BIG_SEQUENCE)Ú	iana_nameÚis_multi_byte_encodingÚunicode_rangec                   @  sH  e Zd Z		dIdJdd„ZdKdd„ZdKdd„ZedLdd„ƒZdMdd„ZdMdd„Z	dNdd „Z
edMd!d"„ƒZedOd$d%„ƒZedPd&d'„ƒZedPd(d)„ƒZedOd*d+„ƒZedMd,d-„ƒZedLd.d/„ƒZedLd0d1„ƒZedLd2d3„ƒZedLd4d5„ƒZedQd6d7„ƒZedRd9d:„ƒZedPd;d<„ƒZedOd=d>„ƒZedOd?d@„ƒZdSdTdDdE„ZedUdGdH„ƒZdS )VÚCharsetMatchNÚpayloadúbytes | bytearrayÚguessed_encodingÚstrÚmean_mess_ratioÚfloatÚhas_sig_or_bomÚboolÚ	languagesÚCoherenceMatchesÚdecoded_payloadú
str | NoneÚpreemptive_declarationc                 C  sL   || _ || _|| _|| _|| _d | _g | _d| _d | _d | _	|| _
|| _d S )Nç        )Ú_payloadÚ	_encodingÚ_mean_mess_ratioÚ
_languagesÚ_has_sig_or_bomÚ_unicode_rangesÚ_leavesÚ_mean_coherence_ratioÚ_output_payloadÚ_output_encodingÚ_stringÚ_preemptive_declaration)Úselfr   r   r   r   r   r   r   © r+   úh/home/esfera/Documents/content_generation/venv/lib/python3.10/site-packages/charset_normalizer/models.pyÚ__init__   s   

zCharsetMatch.__init__ÚotherÚobjectÚreturnc                 C  s>   t |tƒst |tƒrt|ƒ| jkS dS | j|jko| j|jkS )NF)Ú
isinstancer   r   r   ÚencodingÚfingerprint©r*   r.   r+   r+   r,   Ú__eq__(   s
   

zCharsetMatch.__eq__c                 C  sŒ   t |tƒst‚t| j|j ƒ}t| j|j ƒ}|dk r%|dkr%| j|jkS |dk r@|dkr@t| jƒtkr:| j|jk S | j	|j	kS | j|jk S )zQ
        Implemented to make sorted available upon CharsetMatches items.
        g{®Gázt?g{®Gáz”?)
r1   r   Ú
ValueErrorÚabsÚchaosÚ	coherenceÚlenr   r   Úmulti_byte_usage)r*   r.   Úchaos_differenceÚcoherence_differencer+   r+   r,   Ú__lt__/   s   
zCharsetMatch.__lt__c                 C  s   dt t| ƒƒt | jƒ  S )Ng      ð?)r:   r   Úraw©r*   r+   r+   r,   r;   E   s   zCharsetMatch.multi_byte_usagec                 C  sV   | j d u r(t| j| jdƒ| _ | jr(| jdkr(| j r(| j d dkr(| j dd … | _ | j S )NÚstrictÚutf_7r   u   ï»¿r	   )r(   r   r   r   r"   r@   r+   r+   r,   Ú__str__I   s   
ÿ
ÿzCharsetMatch.__str__c                 C  s   d| j › d| j› d�S )Nz<CharsetMatch 'z' fp(z)>)r2   r3   r@   r+   r+   r,   Ú__repr__Y   s   zCharsetMatch.__repr__ÚNonec                 C  s8   t |tƒr	|| krtd |j¡ƒ‚d |_| j |¡ d S )Nz;Unable to add instance <{}> as a submatch of a CharsetMatch)r1   r   r6   ÚformatÚ	__class__r(   r$   Úappendr4   r+   r+   r,   Úadd_submatch\   s   ÿÿzCharsetMatch.add_submatchc                 C  ó   | j S ©N)r   r@   r+   r+   r,   r2   g   ó   zCharsetMatch.encodingú	list[str]c                 C  sD   g }t  ¡ D ]\}}| j|kr| |¡ q| j|kr| |¡ q|S )z‚
        Encoding name are known by many name, using this could help when searching for IBM855 when it's listed as CP855.
        )r   Úitemsr2   rH   )r*   Úalso_known_asÚuÚpr+   r+   r,   Úencoding_aliasesk   s   


€zCharsetMatch.encoding_aliasesc                 C  rJ   rK   ©r"   r@   r+   r+   r,   Úbomx   rL   zCharsetMatch.bomc                 C  rJ   rK   rS   r@   r+   r+   r,   Úbyte_order_mark|   rL   zCharsetMatch.byte_order_markc                 C  s   dd„ | j D ƒS )zÔ
        Return the complete list of possible languages found in decoded sequence.
        Usually not really useful. Returned list may be empty even if 'language' property return something != 'Unknown'.
        c                 S  s   g | ]}|d  ‘qS )r   r+   )Ú.0Úer+   r+   r,   Ú
<listcomp>†   ó    z*CharsetMatch.languages.<locals>.<listcomp>©r!   r@   r+   r+   r,   r   €   s   zCharsetMatch.languagesc                 C  sp   | j s1d| jv r
dS ddlm}m} t| jƒr|| jƒn|| jƒ}t|ƒdks+d|v r-dS |d S | j d d S )z’
        Most probable language found in decoded sequence. If none were detected or inferred, the property will return
        "Unknown".
        ÚasciiÚEnglishr   )Úencoding_languagesÚmb_encoding_languageszLatin BasedÚUnknown)r!   Úcould_be_from_charsetÚcharset_normalizer.cdr]   r^   r   r2   r:   )r*   r]   r^   r   r+   r+   r,   Úlanguageˆ   s   
ÿýzCharsetMatch.languagec                 C  rJ   rK   )r    r@   r+   r+   r,   r8   ¤   rL   zCharsetMatch.chaosc                 C  s   | j sdS | j d d S )Nr   r   r	   rZ   r@   r+   r+   r,   r9   ¨   s   zCharsetMatch.coherencec                 C  ó   t | jd dd�S ©Néd   é   )Úndigits)Úroundr8   r@   r+   r+   r,   Úpercent_chaos®   ó   zCharsetMatch.percent_chaosc                 C  rc   rd   )rh   r9   r@   r+   r+   r,   Úpercent_coherence²   rj   zCharsetMatch.percent_coherencec                 C  rJ   )z+
        Original untouched bytes.
        )r   r@   r+   r+   r,   r?   ¶   s   zCharsetMatch.rawúlist[CharsetMatch]c                 C  rJ   rK   )r$   r@   r+   r+   r,   Úsubmatch½   rL   zCharsetMatch.submatchc                 C  ó   t | jƒdkS ©Nr   )r:   r$   r@   r+   r+   r,   Úhas_submatchÁ   s   zCharsetMatch.has_submatchc                 C  s@   | j d ur| j S dd„ t| ƒD ƒ}ttdd„ |D ƒƒƒ| _ | j S )Nc                 S  s   g | ]}t |ƒ‘qS r+   )r   )rV   Úcharr+   r+   r,   rX   Ê   rY   z*CharsetMatch.alphabets.<locals>.<listcomp>c                 S  s   h | ]}|r|’qS r+   r+   )rV   Úrr+   r+   r,   Ú	<setcomp>Ì   rY   z)CharsetMatch.alphabets.<locals>.<setcomp>)r#   r   ÚsortedÚlist)r*   Údetected_rangesr+   r+   r,   Ú	alphabetsÅ   s
   
zCharsetMatch.alphabetsc                 C  s   | j gdd„ | jD ƒ S )zÜ
        The complete list of encoding that output the exact SAME str result and therefore could be the originating
        encoding.
        This list does include the encoding available in property 'encoding'.
        c                 S  s   g | ]}|j ‘qS r+   )r2   )rV   Úmr+   r+   r,   rX   Ö   s    z6CharsetMatch.could_be_from_charset.<locals>.<listcomp>)r   r$   r@   r+   r+   r,   r`   Ï   s   z"CharsetMatch.could_be_from_charsetÚutf_8r2   Úbytesc                   s~   ˆ j du s
ˆ j |kr<|ˆ _ tˆ ƒ}ˆ jdur5ˆ j ¡ dvr5tt‡ fdd„|dd… dd�}||dd…  }| |d¡ˆ _ˆ jS )	z®
        Method to get re-encoded bytes payload using given target encoding. Default to UTF-8.
        Any errors will be simply ignored by the encoder NOT replaced.
        N)zutf-8Úutf8ry   c                   s<   | j |  ¡ d |  ¡ d …  |  ¡ d tˆ jƒ dd¡¡S )Nr   r	   Ú_ú-)ÚstringÚspanÚreplaceÚgroupsr   r'   )rx   r@   r+   r,   Ú<lambda>ç   s    
þz%CharsetMatch.output.<locals>.<lambda>i    r	   )Úcountr€   )r'   r   r)   Úlowerr   r
   Úencoder&   )r*   r2   Údecoded_stringÚpatched_headerr+   r@   r,   ÚoutputØ   s    
ÿ

ù
zCharsetMatch.outputÚintc                 C  s   t t| ƒƒS )z]
        Retrieve a hash fingerprint of the decoded payload, used for deduplication.
        )Úhashr   r@   r+   r+   r,   r3   õ   s   zCharsetMatch.fingerprint)NN)r   r   r   r   r   r   r   r   r   r   r   r   r   r   )r.   r/   r0   r   )r0   r   ©r0   r   )r.   r   r0   rE   )r0   rM   ©r0   r   )r0   r   )r0   rl   )ry   )r2   r   r0   rz   ©r0   r‰   )Ú__name__Ú
__module__Ú__qualname__r-   r5   r>   Úpropertyr;   rC   rD   rI   r2   rR   rT   rU   r   rb   r8   r9   ri   rk   r?   rm   rp   rw   r`   rˆ   r3   r+   r+   r+   r,   r      sX    ø




	r   c                   @  sb   e Zd ZdZdddd„Zd d	d
„Zd!dd„Zd"dd„Zd#dd„Zd$dd„Z	d%dd„Z
d%dd„ZdS )&ÚCharsetMatchesz³
    Container with every CharsetMatch items ordered by default from most probable to the less one.
    Act like a list(iterable) but does not implements all related methods.
    NÚresultsúlist[CharsetMatch] | Nonec                 C  s   |r	t |ƒ| _d S g | _d S rK   )rt   Ú_results)r*   r“   r+   r+   r,   r-     s   zCharsetMatches.__init__r0   úIterator[CharsetMatch]c                 c  s   � | j E d H  d S rK   ©r•   r@   r+   r+   r,   Ú__iter__  s   €zCharsetMatches.__iter__Úitemú	int | strr   c                 C  sJ   t |tƒr
| j| S t |tƒr#t|dƒ}| jD ]}||jv r"|  S qt‚)z¸
        Retrieve a single item either by its position or encoding name (alias may be used here).
        Raise KeyError upon invalid index or encoding not present in results.
        F)r1   r‰   r•   r   r   r`   ÚKeyError)r*   r™   Úresultr+   r+   r,   Ú__getitem__	  s   





ÿzCharsetMatches.__getitem__r‰   c                 C  s
   t | jƒS rK   ©r:   r•   r@   r+   r+   r,   Ú__len__  s   
zCharsetMatches.__len__r   c                 C  rn   ro   rž   r@   r+   r+   r,   Ú__bool__  s   zCharsetMatches.__bool__rE   c                 C  s|   t |tƒstd t|jƒ¡ƒ‚t|jƒtk r0| j	D ]}|j
|j
kr/|j|jkr/| |¡  dS q| j	 |¡ t| j	ƒ| _	dS )z~
        Insert a single match. Will be inserted accordingly to preserve sort.
        Can be inserted as a submatch.
        z-Cannot append instance '{}' to CharsetMatchesN)r1   r   r6   rF   r   rG   r:   r?   r   r•   r3   r8   rI   rH   rt   )r*   r™   Úmatchr+   r+   r,   rH     s   
ÿÿ

€zCharsetMatches.appendúCharsetMatch | Nonec                 C  s   | j sdS | j d S )zQ
        Simply return the first match. Strict equivalent to matches[0].
        Nr   r—   r@   r+   r+   r,   Úbest1  s   
zCharsetMatches.bestc                 C  s   |   ¡ S )zP
        Redundant method, call the method best(). Kept for BC reasons.
        )r£   r@   r+   r+   r,   Úfirst9  s   zCharsetMatches.firstrK   )r“   r”   )r0   r–   )r™   rš   r0   r   r�   rŒ   )r™   r   r0   rE   )r0   r¢   )rŽ   r�   r�   Ú__doc__r-   r˜   r�   rŸ   r    rH   r£   r¤   r+   r+   r+   r,   r’   ý   s    





r’   c                   @  s.   e Zd Zddd„Zeddd„ƒZddd„ZdS )ÚCliDetectionResultÚpathr   r2   r   rR   rM   Úalternative_encodingsrb   rw   r   r   r8   r   r9   Úunicode_pathÚis_preferredc                 C  sF   || _ |
| _|| _|| _|| _|| _|| _|| _|| _|	| _	|| _
d S rK   )r§   r©   r2   rR   r¨   rb   rw   r   r8   r9   rª   )r*   r§   r2   rR   r¨   rb   rw   r   r8   r9   r©   rª   r+   r+   r,   r-   E  s   
zCliDetectionResult.__init__r0   údict[str, Any]c                 C  s2   | j | j| j| j| j| j| j| j| j| j	| j
dœS )N©r§   r2   rR   r¨   rb   rw   r   r8   r9   r©   rª   r¬   r@   r+   r+   r,   Ú__dict___  s   õzCliDetectionResult.__dict__c                 C  s   ddl m} || jddd�S )Nr   )ÚdumpsTé   )Úensure_asciiÚindent)Újsonr®   r­   )r*   r®   r+   r+   r,   Úto_jsono  s   zCliDetectionResult.to_jsonN)r§   r   r2   r   rR   rM   r¨   rM   rb   r   rw   rM   r   r   r8   r   r9   r   r©   r   rª   r   )r0   r«   r‹   )rŽ   r�   r�   r-   r‘   r­   r³   r+   r+   r+   r,   r¦   D  s
    
r¦   N)Ú
__future__r   Úencodings.aliasesr   Úrer   Útypingr   r   r   r   Úconstantr
   r   Úutilsr   r   r   r   r’   r   r   ÚCoherenceMatchr   r¦   r+   r+   r+   r,   Ú<module>   s     sC