comparison mercurial/util.py @ 43907:68af0228fedc

util: implement sortdict.insert()

As flagged by pytype (reported via Matt Harbison, thanks). This was
broken by bd0fd3ff9916 (util: rewrite sortdict using Python 2.7's
OrderedDict, 2017-05-16). We actually call insert() on
namespaces.py:100, but we clearly don't have test coverage of that an
no users have reported it AFAIK.

Differential Revision: https://phab.mercurial-scm.org/D7680

43906:727cf6acadfe

    >>> d2
    sortdict([('a', 0), ('b', 1)])
    >>> d2.update([(b'a', 2)])
    >>> list(d2.keys()) # should still be in last-set order
    ['b', 'a']
    '''

    def __setitem__(self, key, value):
        if key in self:
            del self[key]

        # __setitem__() isn't called as of PyPy 5.8.0
        def update(self, src):
            if isinstance(src, dict):
                src = pycompat.iteritems(src)
            for k, v in src:
                self[k] = v


class cowdict(cow, dict):
    """copy-on-write dict

43907:68af0228fedc

    >>> d2
    sortdict([('a', 0), ('b', 1)])
    >>> d2.update([(b'a', 2)])
    >>> list(d2.keys()) # should still be in last-set order
    ['b', 'a']
    >>> d1.insert(1, b'a.5', 0.5)
    >>> d1
    sortdict([('a', 0), ('a.5', 0.5), ('b', 1)])
    '''

    def __setitem__(self, key, value):
        if key in self:
            del self[key]

        # __setitem__() isn't called as of PyPy 5.8.0
        def update(self, src):
            if isinstance(src, dict):
                src = pycompat.iteritems(src)
            for k, v in src:
                self[k] = v

    def insert(self, position, key, value):
        for (i, (k, v)) in enumerate(list(self.items())):
            if i == position:
                self[key] = value
            if i >= position:
                del self[k]
                self[k] = v


class cowdict(cow, dict):
    """copy-on-write dict