o
    e                     @   sh   d dl mZmZ ddlmZ ddlmZ ddlmZm	Z	m
Z
 ddlmZmZmZmZ G dd deZd	S )
    )OptionalUnion   )CharSetProber)CodingStateMachine)LanguageFilterMachineStateProbingState)HZ_SM_MODELISO2022CN_SM_MODELISO2022JP_SM_MODELISO2022KR_SM_MODELc                       s   e Zd ZdZejfdeddf fddZd fddZede	e
 fd	d
Zede	e
 fddZdefddZdeeef defddZ  ZS )EscCharSetProberz
    This CharSetProber uses a "code scheme" approach for detecting encodings,
    whereby easily recognizable escape or shift sequences are relied on to
    identify these encodings.
    lang_filterreturnNc                    s   t  j|d g | _| jtj@ r | jtt | jtt	 | jtj
@ r.| jtt | jtj@ r<| jtt d| _d | _d | _tj| _|   d S )N)r   r   )super__init__	coding_smr   r   ZCHINESE_SIMPLIFIEDappendr   r
   r   ZJAPANESEr   ZKOREANr   active_sm_count_detected_charset_detected_languager	   Z	DETECTING_statereset)selfr   	__class__ 1lib/python3.10/site-packages/chardet/escprober.pyr   0   s   zEscCharSetProber.__init__c                    s@   t    | jD ]	}d|_|  qt| j| _d | _d | _d S )NT)r   r   r   activelenr   r   r   )r   r   r   r   r   r   @   s   



zEscCharSetProber.resetc                 C      | j S Nr   r   r   r   r   charset_nameI      zEscCharSetProber.charset_namec                 C   r!   r"   )r   r$   r   r   r   languageM   r&   zEscCharSetProber.languagec                 C   s   | j rdS dS )NgGz?g        r#   r$   r   r   r   get_confidenceQ   s   zEscCharSetProber.get_confidencebyte_strc                 C   s   |D ]J}| j D ]D}|jsq||}|tjkr2d|_|  jd8  _| jdkr1tj| _| j	    S q|tj
krKtj| _| | _|j| _| j	    S qq| j	S )NFr   r   )r   r   Z
next_stater   ZERRORr   r	   ZNOT_MEr   stateZITS_MEZFOUND_ITZget_coding_state_machiner   r'   r   )r   r)   cr   Zcoding_stater   r   r   feedT   s(   





zEscCharSetProber.feed)r   N)__name__
__module____qualname____doc__r   ZNONEr   r   propertyr   strr%   r'   floatr(   r   bytes	bytearrayr	   r,   __classcell__r   r   r   r   r   )   s    	"r   N)typingr   r   Zcharsetproberr   Zcodingstatemachiner   Zenumsr   r   r	   Zescsmr
   r   r   r   r   r   r   r   r   <module>   s   