3

�ft`T�@sZdd
lm
Z

ddlmZ
ddlmZ
ddlmZ
ddddd	�Zd
d�ZGdd
�d
e�Z	dS)�)
�absolute_import)
�print_function)
�unicode_literals)
�reduce����
)�nameZsummary�descriptionZurlcCs"t|�
}d
g
|
|t
|�
}|S)z, Ordered sset with empty strings prepended. �)�len�sorted)ZssetZlengthZcurrent�
l�r�#/usr/lib/python3.6/match_counter.py�_canonize_string_set"s

rc@sfeZ
dZd
Zedd��
Zdd�Zdd�Zdd	�Zd
d�Z	dd
�Z
dd�Zdd�Zddd�
Z
dd�ZdS)�MatchCounterz�Map packages to which of their attributes matched in a search against
    what values.

    The mapping is: ``package -> [(key, needle), ... ]``.

    cs�f
d
d�}tt
||
��
S)Nc
s>|d
}
|d}t�|
�}|
dkr6||kr6dt
|
St
|
S)Nr
r	r
r)�getattr�WEIGHTS)�match�key�needleZhaystack)
�pkgrr�weight4s





z*MatchCounter._eval_weights.<locals>.weight)�sum�map)rZmatchesrr)
rr�
_eval_weights1s	zMatchCounter._eval_weightsc
s�f
d
d�}
|
S)a�
Get the key function used for sorting matches.

        It is not enough to only look at the matches and order them by the sum
        of their weighted hits. In case this number is the same we have to
        ensure that the same matched needles are next to each other in the
        result.

        Returned function is:
        pkg -> (weights_sum, canonized_needles_set, -distance)

        c

s�j|�|�|j
fS)
N)rr
)
r)
�selfrr�get_keyKsz'MatchCounter._key_func.<locals>.get_keyr)rrr)
rr�	_key_func?szMatchCounter._key_funcc

s�rt�f
d
d��D�
�
SdS)z0Return the max count of needles of all packages.c
3s|]}
t�j
|
�
�
V
qdS)
N)r
�matched_needles)�.0r)
rrr�	<genexpr>Wsz,MatchCounter._max_needles.<locals>.<genexpr>r
)
�max)
rr)
rr�_max_needlesTs

zMatchCounter._max_needlescCs|j|
g�j
||f�

dS)
N)�
setdefault�append)rrrrrrr�addZs
zMatchCounter.addc
Cs&x |D]}
td
|
||
f�

qWdS)Nz%s	%s)
�print)rrrrr�dump]s


zMatchCounter.dumpcst�f
d
d�|�D�
�
S)Nc
3s|]}
t�|
d�V
qd
S)r
N)
r)r"�
m)
rrrr#bsz1MatchCounter.matched_haystacks.<locals>.<genexpr>)
�set)rrr)
rr�matched_haystacksas
zMatchCounter.matched_haystackscCs6g}x,||
D] }|d
|kr q|j|d
�

qW|S)Nr
)
r')rr�result�
irrr�matched_keysds




zMatchCounter.matched_keyscCstd
d�||
D�
�
S)Nc
ss|]}
|
dV
qd
S)r	Nr)r"r+rrrr#nsz/MatchCounter.matched_needles.<locals>.<genexpr>)
r,)rrrrrr!ms
zMatchCounter.matched_needlesFNcCs |r|n|j�}t
||j�d
�S)N)
r)�keysrr )r�reverseZlimit_tor1rrrrps

zMatchCounter.sortedc

st�f
d
d��d�S)Ncs|t�|
�
S)
N)
r
)�totalr)
rrr�<lambda>usz$MatchCounter.total.<locals>.<lambda>r
)
r)
rr)
rrr3ts
zMatchCounter.total)FN)�__name__�
__module__�__qualname__�__doc__�staticmethodrr r%r(r*r-r0r!rr3rrrrr)s	
rN)
Z
__future__rrr�	functoolsrrr�dictrrrrr�<module>s