a y h@sddlmZddlmZddlmZmZddZddZdd Z d d Z d d d ddZ d d d ddZ d d d ddZ d d d ddZd S)) annotations)conv_sequences)is_none setupPandascCs<|d}d}|||7}|||7}||||7}|dS)Ng@) pattern_lentext_len common_charsZtranspositionssimrrV/var/www/html/assistant/venv/lib/python3.9/site-packages/rapidfuzz/distance/Jaro_py.py_jaro_calculate_similarity s   rcCs(|r|s dSt||t||d}||kS)zC filter matches below score_cutoff based on string lengths Fr)rmin)r r score_cutoffr rrr _jaro_length_filtersrcCs|sdSt|||d}||kS)zY filter matches below score_cutoff based on string lengths and common characters Fr)r)r r r rr rrr _jaro_common_char_filtersrcCsxt|}t|}d}||krF|dd}|||krn|d||}n(|dd}|||krn|d||}|||fS)zB find bounds and skip out of bound parts of the sequences rrN)len)s1s2r r boundrrr _jaro_bounds(s    rN processorrcCstt|st|rdS|dur2||}||}|s>|s>dS|durJd}t||\}}t|}t|}t|||sxdS|dkr|dkrt|d|dkSt||\}}}dg|}dg|}d} t|D]r\} } td| |} t | ||d} t | | dD]:}||s||| krd|| <||<| d7} q̐qqt ||| |sTdSd}}t|D]T\} }|rdt ||D]}||r||d}qq||| ||krd|d7}qdt ||| |S)a Calculates the jaro similarity Parameters ---------- s1 : Sequence[Hashable] First string to compare. s2 : Sequence[Hashable] Second string to compare. processor: callable, optional Optional callable that is used to preprocess the strings before comparing them. Default is None, which deactivates this behaviour. score_cutoff : float, optional Optional argument for a score threshold as a float between 0 and 1.0. For ratio < score_cutoff 0 is returned instead. Default is None, which deactivates this behaviour. Returns ------- similarity : float similarity between s1 and s2 as a float between 0 and 1.0 rN?rrFT) rrrrrfloatr enumeratemaxrrangerr)rrrrr r rZs1_flagsZs2_flagsr iZs1_chlowhijkZ trans_countZs1_frrr similarity=sR       r%cCst||||dS)a Calculates the normalized jaro similarity Parameters ---------- s1 : Sequence[Hashable] First string to compare. s2 : Sequence[Hashable] Second string to compare. processor: callable, optional Optional callable that is used to preprocess the strings before comparing them. Default is None, which deactivates this behaviour. score_cutoff : float, optional Optional argument for a score threshold as a float between 0 and 1.0. For ratio < score_cutoff 0 is returned instead. Default is None, which deactivates this behaviour. Returns ------- normalized similarity : float normalized similarity between s1 and s2 as a float between 0 and 1.0 r)r%rrrrrrr normalized_similaritysr'cCs|tt|st|rdS|dur2||}||}|dusB|dkrFdnd|}t|||d}d|}|dust||krx|SdS)a Calculates the jaro distance Parameters ---------- s1 : Sequence[Hashable] First string to compare. s2 : Sequence[Hashable] Second string to compare. processor: callable, optional Optional callable that is used to preprocess the strings before comparing them. Default is None, which deactivates this behaviour. score_cutoff : float, optional Optional argument for a score threshold as a float between 0 and 1.0. For ratio < score_cutoff 0 is returned instead. Default is None, which deactivates this behaviour. Returns ------- distance : float distance between s1 and s2 as a float between 1.0 and 0.0 rN)r)rrr%)rrrrZcutoff_distancer distrrr distancesr)cCst||||dS)a Calculates the normalized jaro distance Parameters ---------- s1 : Sequence[Hashable] First string to compare. s2 : Sequence[Hashable] Second string to compare. processor: callable, optional Optional callable that is used to preprocess the strings before comparing them. Default is None, which deactivates this behaviour. score_cutoff : float, optional Optional argument for a score threshold as a float between 0 and 1.0. For ratio < score_cutoff 0 is returned instead. Default is None, which deactivates this behaviour. Returns ------- normalized distance : float normalized distance between s1 and s2 as a float between 1.0 and 0.0 r)r)r&rrr normalized_distancesr*) __future__rZrapidfuzz._common_pyrZrapidfuzz._utilsrrrrrrr%r'r)r*rrrr s$      ^ $ /