3
ft`T @ sZ d dl mZ d dl mZ d dl mZ d dlmZ ddddd Zd
d ZG d d
d
eZ dS ) )absolute_import)print_function)unicode_literals)reduce )nameZsummary descriptionZurlc C s" t | }dg|| t| }|S )z, Ordered sset with empty strings prepended. )lensorted)ZssetZlengthZcurrentl r #/usr/lib/python3.6/match_counter.py_canonize_string_set" s r c @ sf e Zd ZdZedd Zdd Zdd Zdd Zd
d Z d d
Z
dd Z dd Z dddZ
dd ZdS ) MatchCounterzMap packages to which of their attributes matched in a search against
what values.
The mapping is: ``package -> [(key, needle), ... ]``.
c s fdd}t t||S )Nc s> | d }| d }t |}|dkr6||kr6dt| S t| S )Nr r r
r )getattrWEIGHTS)matchkeyneedleZhaystack)pkgr r weight4 s
z*MatchCounter._eval_weights..weight)summap)r Zmatchesr r )r r
_eval_weights1 s zMatchCounter._eval_weightsc s fdd}|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 | | | jfS )N)r r
)r )selfr r get_keyK s z'MatchCounter._key_func..get_keyr )r r r )r r _key_func? s zMatchCounter._key_funcc s rt fdd D S dS )z0Return the max count of needles of all packages.c 3 s | ]}t j|V qd S )N)r
matched_needles).0r )r r r W s z,MatchCounter._max_needles..r )max)r r )r r _max_needlesT s zMatchCounter._max_needlesc C s | j |g j||f d S )N)
setdefaultappend)r r r r r r r addZ s zMatchCounter.addc C s& x | D ]}t d|| | f qW d S )Nz%s %s)print)r r r r r dump] s
zMatchCounter.dumpc s t fdd| D S )Nc 3 s | ]}t |d V qdS )r N)r )r" m)r r r r# b s z1MatchCounter.matched_haystacks..)set)r r r )r r matched_haystacksa s z MatchCounter.matched_haystacksc C s6 g }x,| | D ] }|d |kr q|j |d qW |S )Nr )r' )r r resultir r r matched_keysd s zMatchCounter.matched_keysc C s t dd | | D S )Nc s s | ]}|d V qdS )r Nr )r" r+ r r r r# n s z/MatchCounter.matched_needles..)r, )r r r r r r! m s z MatchCounter.matched_needlesFNc C s |r|n| j }t|| j dS )N)r )keysr r )r reverseZlimit_tor1 r r r r p s zMatchCounter.sortedc s t fdd dS )Nc s | t | S )N)r
)totalr )r r r u s z$MatchCounter.total..r )r )r r )r r r3 t s zMatchCounter.total)FN)__name__
__module__ __qualname____doc__ staticmethodr r r% r( r* r- r0 r! r r3 r r r r r ) s
r N)
Z
__future__r r r functoolsr r r dictr r r r r s