3
�ft`T � @ sZ d dl mZ d dl mZ d dl mZ d dlmZ ddddd �Zd
d � ZG d d
� d
e�Z dS )� )�absolute_import)�print_function)�unicode_literals)�reduce� � � � )�nameZsummary� descriptionZurlc C s" t | �}dg|| 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 r c @ sf e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� 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), ... ]``.
c s � fdd�}t t||��S )Nc s> | d }| d }t � |�}|dkr6||kr6dt| S t| S )Nr r r
r )�getattr�WEIGHTS)�match�key�needleZhaystack)�pkgr r �weight4 s
z*MatchCounter._eval_weights..weight)�sum�map)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)�
setdefault�append)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 �result�ir 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� d�S )N)r )�keysr r )r �reverseZlimit_tor1 r r r r p s zMatchCounter.sortedc s t � fdd�� d�S )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