o խj @sUddlmZddlZddlmZddlmZddlmZm Z m Z m Z ddl m Z mZmZmZmZddlmZdd lmZmZdd lmZmZmZmZmZmZed Ze Z!e!"e#d gZ$d e%d<gZ&d e%d<e D]!Z'zee'r{e$(e'ne&(e'Wqne)ye&(e'Yqnwe$e&Z*d e%d<         d6d7d(d)Z+         d6d8d,d-Z,         d6d9d0d1Z-         d:d;d4d5Z.dS)<) annotationsN)PathLike)BinaryIO)coherence_ratioencoding_languagesmb_encoding_languagesmerge_coherence_ratios)IANA_SUPPORTEDIANA_SUPPORTED_SIMILARTOO_BIG_SEQUENCETOO_SMALL_SEQUENCETRACE) mess_ratio) CharsetMatchCharsetMatches)any_specified_encodingcut_sequence_chunks iana_nameidentify_sig_or_bomis_multi_byte_encodingshould_strip_sig_or_bomcharset_normalizerz)%(asctime)s | %(levelname)s | %(message)sz list[str] _mb_supported _sb_supportedIANA_SUPPORTED_MB_FIRST皙?TF皙? sequencesbytes | bytearraystepsint chunk_size thresholdfloat cp_isolationlist[str] | None cp_exclusionpreemptive_behaviourboolexplainlanguage_thresholdenable_fallbackreturnrc @ Cs t|ttfstdt||rtj} tt t t t |} | dkrDt d|r8tt t | tt|dddgdgS|durZtt d d |d d |D}ng}|durrtt d d |dd |D}ng}| ||krtt d||| d}| }|dkr| ||krt| |}t |tk} t |tk} | rtt d| n | rtt d| g}|rt|nd}|dur||tt d|t}g}g}t}t}i}d}t}d}d}d}d}d}d}t}t}t|\} }!| dur|| tt dt |!| |dd|vr#|d|tD]F}"|r4|"|vr4q'|r>|"|vr>q'|"|vrEq'||"d}#| |"k}$|$oVt|"}%|"dvrh|$shtt d|"q'|"dvry|$sytt d|"q'|"|vrtt d|"q'|"|vrtt d|"q'zt|"}&Wnttfytt d|"Yq'w|r|&stt |"}'ntt!|"}'|'"|stt d|"|'|q'|r|&s||krtt d|"||q'|r|&stt d |"q'zX| r|&durt#|%dur|dtd!n |t |!td!|"d"n2|"d#kr=|$r=t#||"d"}#|#r<|#dd$kr<|#dd}#nt#|%durE|n|t |!d|"d"}#Wn+t$t%fy|}(zt|(t%sltt d%|"t#|(||"WYd}(~(q'd}(~(wwt&|$sdnt |!| t| |})|&o|#duot |#| k}*|*rtt d&|"|#dur|&st'|#}+|(|+},|,dur|,\}-}.}/|/rbt||"|-|$|.| dus|"|ddfvr|#nd|d'}0||0||"tt d(|"t)|-d)d*d+|"|ddfvr(|-d,kr(|-dkr#t d-|0j*|rtt t | t|0gS||0t |r`|dus7||vr`d|vr`d|vr`|+}1t d-|1j*|rYtt t | t|1gSq'||"tt d.|"| r|"dd|d/d0fvrt||"||$g|#|d'}2|"|kr|2}n |"dkr|2}n|2}q'tt |)d1}3t,|3d2}3d}4d}5g}6g}7zLt-||"|)||$|%|!|&|# D]=}8|6|8|7t.|8||d3uodt |kod2kn|7d4|kr|4d7}4|4|3ks|$r|%durnqWn!t$y}(ztt d5|"t#|(|3}4d3}5WYd}(~(nd}(~(ww|5sY| rY|&sYz|td6dj/|"d7d8Wn#t$yX}(ztt d9|"t#|(||"WYd}(~(q'd}(~(ww|7rdt0|7t |7nd}9|9|ksp|4|3kr||"|"t1vr|2t1|"|#dur|&s|3t'|#|9gdftt d:|"|4t)|9d)d*d+| r|"dd|d/d0fvr|5st||"||$g|#|d'}2|"|kr|2}n |"dkr|2}n|2}q'tt d;|"t)|9d)d*d+|&st |"}:nt!|"}:|:rtt d<|"t#|:g};|"dkr |6D]}8t4|8||:rd=|:nd}<|;|<qt5|;}=nt5|;}=|=r1tt d>|=|"t||"|9|$|=| dusD|"|ddfvrF|#nd|d'}>||>|#durc|&sc|3t'|#|9|=d3f|rr|&sr|9d?krr|d7}|"|ddfvr|9d,kr|9dkrt d-|>j*|rtt t | t|>gS||>t |r|dus||vrd|vrd|vr|+}1t d-|1j*|rtt t | t|1gS|s|&s|=rt,d@dA|=DddBnd}?|?dCkrd|vrd|vrd3}|2|:tt dD|"|9|?|sL|&rL|*rL|#durLt |#| dEkrL|"dFvrLd|vrLd|vrLd3}tt dG|"|9t |#| t |#| d)|"| krmt dH|"|rdtt t | t||"gSq't |dkr|s|s|rtt dI|rt dJ|j*||n2|r|dus|r|r|j6|j6ks|durt dK||n |rt dL|||rt dM|+j*t |dnt dN|rtt t | |S)Oaf Given a raw bytes sequence, return the best possibles charset usable to render str objects. If there is no results, it is a strong indicator that the source is binary/not text. By default, the process will extract 5 blocks of 512o each to assess the mess and coherence of a given sequence. And will give up a particular code page after 20% of measured mess. Those criteria are customizable at will. The preemptive behavior DOES NOT replace the traditional detection workflow, it prioritize a particular code page but never take it for granted. Can improve the performance. You may want to focus your attention to some code page or/and not others, use cp_isolation and cp_exclusion for that purpose. This function will strip the SIG in the payload/sequence every time except on UTF-16, UTF-32. By default the library does not setup any handler other than the NullHandler, if you choose to set the 'explain' toggle to True it will alter the logger configuration to add a StreamHandler that is suitable for debugging. Custom logging format and handler can be set manually. z3Expected object of type bytes or bytearray, got: {}rzszfrom_bytes..zacp_exclusion is set. use this flag for debugging purpose. limited list of encoding excluded : %s.cSr2r3r4r5r8r8r9r:~r;z^override steps (%i) and chunk_size (%i) as content does not fit (%i byte(s) given) parameters.rz>Trying to detect encoding from a tiny portion of ({}) byte(s).zIUsing lazy str decoding because the payload is quite large, ({}) byte(s).z@Detected declarative mark in sequence. Priority +1 given for %s.zIDetected a SIG or BOM mark on first %i byte(s). Priority +1 given for %s.ascii>utf_16utf_32z\Encoding %s won't be tested as-is because it require a BOM. Will try some sub-encoder LE/BE.>utf_7zREncoding %s won't be tested as-is because detection is unreliable without BOM/SIG.zY%s is deemed too similar to a code page that was already considered unsuited. Continuing!zESkipping %s: already fast-tracked from a similar successful encoding.z2Encoding %s does not provide an IncrementalDecoderzbSkipping %s: definitive match already found, this encoding targets different languages (%s vs %s).zXSkipping %s: already accumulated %d same-family results after definitive match (cap=%d).zCSkipping single-byte %s: multi-byte definitive match already found.gA)encodingr@uz9Code page %s does not fit given bytes sequence at ALL. %szpCode page %s is a multi byte encoding table and it appear that at least one character was encoded using n-bytes.)preemptive_declarationzM%s fast-tracked (identical decoded payload to a prior encoding, chaos=%f %%).d)ndigitsrz.Encoding detection: %s is most likely the one.zZ%s fast-skipped (identical decoded payload to a prior encoding that failed chaos probing).r>r?TzaLazyStr Loading: After MD chunk decode, code page %s does not fit given bytes sequence at ALL. %sgj@strict)errorsz^LazyStr Loading: After final lookup, code page %s does not fit given bytes sequence at ALL. %szc%s was excluded because of initial chaos probing. Gave up %i time(s). Computed mean chaos is %f %%.z=%s passed initial chaos probing. Mean measured chaos is %f %%z&{} should target any language(s) of {},z We detected language {} using {}g{Gz?css|]\}}|VqdS)Nr8)r6_vr8r8r9 szfrom_bytes..)defaultg?zyDefinitive match found: %s (chaos=%.3f, coherence=%.2f). Encodings targeting different language families will be skipped.g\(\?> r@r0r>r? utf_16_be utf_16_le utf_32_be utf_32_le utf_8_sigzjMulti-byte definitive match: %s (chaos=%.3f, decoded=%d/%d=%.1f%%). Single-byte encodings will be skipped.zoEncoding detection: %s is most likely the one as we detected a BOM or SIG within the beginning of the sequence.zONothing got out of the detection process. Using ASCII/UTF-8/Specified fallback.z7Encoding detection: %s will be used as a fallback matchz:Encoding detection: utf_8 will be used as a fallback matchz:Encoding detection: ascii will be used as a fallback matchz]Encoding detection: Found %s as plausible (best-candidate) for content. With %i alternatives.z=Encoding detection: Unable to determine any suitable charset.)7 isinstance bytearraybytes TypeErrorformattypeloggerlevel addHandlerexplain_handlersetLevelrlendebug removeHandlerrrlogjoinr#r r rappendsetrraddrrModuleNotFoundError ImportErrorrr intersectionstrUnicodeDecodeError LookupErrorrangehashgetroundrAbestmaxrrdecodesumr update setdefaultrr fingerprint)@r r"r$r%r'r)r*r,r-r.previous_logger_levellengthis_too_small_sequenceis_too_large_sequenceprioritized_encodingsspecified_encodingtestedtested_but_hard_failuretested_but_soft_failuresoft_failure_skipsuccess_fast_trackedpayload_result_cachedefinitive_match_founddefinitive_target_languages post_definitive_sb_success_countPOST_DEFINITIVE_SB_CAPmb_definitive_match_foundfallback_ascii fallback_u8fallback_specifiedresultsearly_stop_results sig_encoding sig_payload encoding_ianadecoded_payloadbom_or_sig_availablestrip_sig_or_bomis_multi_byte_decoder enc_languageser_multi_byte_bonus payload_hashcached cached_mess cached_cd cached_passed fast_matchprobable_resultfallback_entrymax_chunk_gave_upearly_stop_countlazy_str_hard_failure md_chunks md_ratioschunkmean_mess_ratiotarget_languages cd_ratioschunk_languagescd_ratios_merged current_matchbest_coherencer8r8r9 from_bytes9s                                                         &                                               rfprc Cst|||||||||| S)z Same thing than the function from_bytes but using a file pointer that is already ready. Will not close the file pointer. )rread) rr"r$r%r'r)r*r,r-r.r8r8r9from_fp`srpathstr | bytes | PathLikec CsHt|d} t| ||||||||| WdS1swYdS)z Same thing than the function from_bytes but with one extra step. Opening and reading given file path in binary mode. Can raise IOError. rbN)openr) rr"r$r%r'r)r*r,r-r.rr8r8r9 from_path~s $rfp_or_path_or_payload!PathLike | str | BinaryIO | bytesc Cst|ttfrt|||||||||| d } | St|ttfr0t|||||||||| d } | St|||||||||| d } | S)a) Detect if the given input (file, bytes, or path) points to a binary file. aka. not a string. Based on the same main heuristic algorithms and default kwargs at the sole exception that fallbacks match are disabled to be stricter around ASCII-compatible but unlikely to be a string. ) r"r$r%r'r)r*r,r-r.)rUrkrrrWrVrr) rr"r$r%r'r)r*r,r-r.guessesr8r8r9 is_binarys\- r) rrrNNTFrT)r r!r"r#r$r#r%r&r'r(r)r(r*r+r,r+r-r&r.r+r/r)rrr"r#r$r#r%r&r'r(r)r(r*r+r,r+r-r&r.r+r/r)rrr"r#r$r#r%r&r'r(r)r(r*r+r,r+r-r&r.r+r/r) rrrNNTFrF)rrr"r#r$r#r%r&r'r(r)r(r*r+r,r+r-r&r.r+r/r+)/ __future__rloggingosrtypingrcdrrrr constantr r r r rmdrmodelsrrutilsrrrrrr getLoggerr[ StreamHandlerr^ setFormatter Formatterr__annotations__r_supported_encrerirrrrrr8r8r8r9s          /   !