Your IP : 3.21.247.61

Current Path : /opt/alt/python37/lib64/python3.7/site-packages/pyrsistent/__pycache__/
Upload File :
Current File : //opt/alt/python37/lib64/python3.7/site-packages/pyrsistent/__pycache__/_pbag.cpython-37.pyc

B

+�[b�@s�ddlmZmZmZmZddlmZddlmZdd�Z	Gdd�de
�Ze�e�e�e�e�e�e�e�d	d
�Z
dd�Zee��Zd
S)�)�	Container�Iterable�Sized�Hashable�)�reduce)�pmapcCs|�||�|d�d�S)Nrr)�set�get)Zcounters�element�r�C/opt/alt/python37/lib64/python3.7/site-packages/pyrsistent/_pbag.py�_add_to_counterssrc@s�eZdZdZdZdd�Zdd�Zdd�Zd	d
�Zdd�Z	d
d�Z
dd�Zdd�Zdd�Z
dd�Zdd�ZeZeZeZdd�Zdd�Zdd�Zdd �Zd!d"�Zd#S)$�PBaga�
    A persistent bag/multiset type.

    Requires elements to be hashable, and allows duplicates, but has no
    ordering. Bags are hashable.

    Do not instantiate directly, instead use the factory functions :py:func:`b`
    or :py:func:`pbag` to create an instance.

    Some examples:

    >>> s = pbag([1, 2, 3, 1])
    >>> s2 = s.add(4)
    >>> s3 = s2.remove(1)
    >>> s
    pbag([1, 1, 2, 3])
    >>> s2
    pbag([1, 1, 2, 3, 4])
    >>> s3
    pbag([1, 2, 3, 4])
    )�_counts�__weakref__cCs
||_dS)N)r)�self�countsrrr
�__init__#sz
PBag.__init__cCstt|j|��S)z�
        Add an element to the bag.

        >>> s = pbag([1])
        >>> s2 = s.add(1)
        >>> s3 = s.add(2)
        >>> s2
        pbag([1, 1])
        >>> s3
        pbag([1, 2])
        )rrr)rrrrr
�add&szPBag.addcCs|rttt||j��S|S)z�
        Update bag with all elements in iterable.

        >>> s = pbag([1])
        >>> s.update([1, 2])
        pbag([1, 1, 2])
        )rrrr)r�iterablerrr
�update4szPBag.updatecCsP||jkrt|��n4|j|dkr0|j�|�}n|j�||j|d�}t|�S)z�
        Remove an element from the bag.

        >>> s = pbag([1, 1, 2])
        >>> s2 = s.remove(1)
        >>> s3 = s.remove(2)
        >>> s2
        pbag([1, 2])
        >>> s3
        pbag([1, 1])
        r)r�KeyError�remover	r)rrZnewcrrr
rAs

zPBag.removecCs|j�|d�S)z�
        Return the number of times an element appears.


        >>> pbag([]).count('non-existent')
        0
        >>> pbag([1, 1, 2]).count(1)
        2
        r)rr
)rrrrr
�countUs
z
PBag.countcCst|j���S)ze
        Return the length including duplicates.

        >>> len(pbag([1, 1, 2]))
        3
        )�sumrZ
itervalues)rrrr
�__len__aszPBag.__len__ccs4x.|j��D] \}}xt|�D]
}|VqWqWdS)z�
        Return an iterator of all elements, including duplicates.

        >>> list(pbag([1, 1, 2]))
        [1, 1, 2]
        >>> list(pbag([1, 2]))
        [1, 2]
        N)r�	iteritems�range)r�eltr�irrr
�__iter__js	z
PBag.__iter__cCs
||jkS)z�
        Check if an element is in the bag.

        >>> 1 in pbag([1, 1, 2])
        True
        >>> 0 in pbag([1, 2])
        False
        )r)rrrrr
�__contains__ws	zPBag.__contains__cCsd�t|��S)Nz	pbag({0}))�format�list)rrrr
�__repr__�sz
PBag.__repr__cCs t|�tk	rtd��|j|jkS)z�
        Check if two bags are equivalent, honoring the number of duplicates,
        and ignoring insertion order.

        >>> pbag([1, 1, 2]) == pbag([1, 2])
        False
        >>> pbag([2, 1, 0]) == pbag([0, 1, 2])
        True
        z Can only compare PBag with PBags)�typer�	TypeErrorr)r�otherrrr
�__eq__�s
zPBag.__eq__cCstd��dS)NzPBags are not orderable)r')rr(rrr
�__lt__�szPBag.__lt__cCsNt|t�stS|j��}x(|j��D]\}}|�|�|||<q$Wt|���S)z� 
        Combine elements from two PBags.

        >>> pbag([1, 2, 2]) + pbag([2, 3, 3])
        pbag([1, 2, 2, 2, 3, 3])
        )�
isinstancer�NotImplementedr�evolverrr�
persistent)rr(�result�elem�other_countrrr
�__add__�s

zPBag.__add__cCsnt|t�stS|j��}xH|j��D]:\}}|�|�|}|dkrL|||<q$||kr$|�|�q$Wt|���S)z� 
        Remove elements from one PBag that are present in another.

        >>> pbag([1, 2, 2, 2, 3]) - pbag([2, 3, 3, 4])
        pbag([1, 2, 2])
        r)	r+rr,rr-rrrr.)rr(r/r0r1�newcountrrr
�__sub__�s


zPBag.__sub__cCsXt|t�stS|j��}x2|j��D]$\}}|�|�}t||�}|||<q$Wt|���S)z� 
        Union: Keep elements that are present in either of two PBags.

        >>> pbag([1, 2, 2, 2]) | pbag([2, 3, 3])
        pbag([1, 2, 2, 2, 3, 3])
        )	r+rr,rr-rr�maxr.)rr(r/r0r1rr3rrr
�__or__�s



zPBag.__or__cCs\t|t�stSt���}x6|j��D](\}}t||�|��}|dkr$|||<q$Wt|�	��S)z�
        Intersection: Only keep elements that are present in both PBags.
        
        >>> pbag([1, 2, 2, 2]) & pbag([2, 3, 3])
        pbag([2])
        r)
r+rr,rr-rr�minrr.)rr(r/r0rr3rrr
�__and__�s

zPBag.__and__cCs
t|j�S)z�
        Hash based on value of elements.

        >>> m = pmap({pbag([1, 2]): "it's here!"})
        >>> m[pbag([2, 1])]
        "it's here!"
        >>> pbag([1, 1, 2]) in m
        False
        )�hashr)rrrr
�__hash__�s
z
PBag.__hash__N)�__name__�
__module__�__qualname__�__doc__�	__slots__rrrrrrr!r"r%r)r*�__le__�__gt__�__ge__r2r4r6r8r:rrrr
r
s*
	
rcGst|�S)z�
    Construct a persistent bag.

    Takes an arbitrary number of arguments to insert into the new persistent
    bag.

    >>> b(1, 2, 3, 2)
    pbag([1, 2, 2, 3])
    )�pbag)�elementsrrr
�b�s
rEcCs|stSttt|t���S)z�
    Convert an iterable to a persistent bag.

    Takes an iterable with elements to insert.

    >>> pbag([1, 2, 3, 2])
    pbag([1, 2, 2, 3])
    )�_EMPTY_PBAGrrrr)rDrrr
rC�s	rCN)�_compatrrrr�	functoolsrZpyrsistent._pmaprr�objectr�registerrErCrFrrrr
�<module>s`





?>