name : _pbag.cpython-39.pyc
a

`�]_J�@s�ddlmZmZmZmZddlmZddlmZdd�Z	Gdd�de
�Ze�e�e�e�e�e�e�e�dd	�Z
d
d�Zee��ZdS)
�)�	Container�Iterable�Sized�Hashable)�reduce)�pmapcCs|�||�|d�d�S)Nr�)�set�get)Zcounters�element�r�6/usr/lib64/python3.9/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)�selfZcountsr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||jvrt|��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
rr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__ccs,|j��D]\}}t|�D]
}|Vqq
dS)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
||jvS)z�
        Check if an element is in the bag.

        >>> 1 in pbag([1, 1, 2])
        True
        >>> 0 in pbag([1, 2])
        False
        r)rr!rrr
�__contains__ws	zPBag.__contains__cCsd�t|��S)Nz	pbag({0}))�format�listrrrr
�__repr__�sz
PBag.__repr__cCs t|�turtd��|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)r*rrr
�__lt__�szPBag.__lt__cCsJt|t�stS|j��}|j��D]\}}|�|�|||<q"t|���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__cCsjt|t�stS|j��}|j��D]:\}}|�|�|}|dkrJ|||<q"||vr"|�|�q"t|���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/rr0rrrr1)rr+r2r3r4�newcountrrr
�__sub__�s


zPBag.__sub__cCsTt|t�stS|j��}|j��D]$\}}|�|�}t||�}|||<q"t|���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/rr0rr�maxr1)rr+r2r3r4rr6rrr
�__or__�s




zPBag.__or__cCsXt|t�stSt���}|j��D](\}}t||�|��}|dkr"|||<q"t|�	��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/rr0rr�minrr1)rr+r2r3rr6r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
        )�hashrrrrr
�__hash__�s
z
PBag.__hash__N)�__name__�
__module__�__qualname__�__doc__�	__slots__rrrrrrr#r$r'r,r-�__le__�__gt__�__ge__r5r7r9r;r=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
rIcCs|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_PBAGrrrrrGrrr
rF�s	rFN)�collections.abcrrrr�	functoolsrZpyrsistent._pmaprr�objectr�registerrIrFrJrrrr
�<module>s`






© 2025 Cubjrnet7