comparison mercurial/dirstate.py @ 49964:f979d8af8648

dirstate: rename `@requires_no_parents_change` too

This match the rename of the context manager.

49963:23b70ce09e55

        return func(self, *args, **kwargs)

    return wrap


def requires_no_parents_change(func):
    def wrap(self, *args, **kwargs):
        if self.pendingparentchange():
            msg = 'calling `%s` inside of a changing_parents context'
            msg %= func.__name__
            raise error.ProgrammingError(msg)

        return self._map.copymap.get(file, None)

    def copies(self):
        return self._map.copymap

    @requires_no_parents_change
    def set_tracked(self, filename, reset_copy=False):
        """a "public" method for generic code to mark a file as tracked

        This function is to be called outside of "update/merge" case. For
        example by a command like `hg add X`.

            self._map.copymap.pop(filename, None)
        if pre_tracked:
            self._dirty_tracked_set = True
        return pre_tracked

    @requires_no_parents_change
    def set_untracked(self, filename):
        """a "public" method for generic code to mark a file as untracked

        This function is to be called outside of "update/merge" case. For
        example by a command like `hg remove X`.

        if ret:
            self._dirty = True
            self._dirty_tracked_set = True
        return ret

    @requires_no_parents_change
    def set_clean(self, filename, parentfiledata):
        """record that the current state of the file on disk is known to be clean"""
        self._dirty = True
        if not self._map[filename].tracked:
            self._check_new_tracked_filename(filename)
        (mode, size, mtime) = parentfiledata
        self._map.set_clean(filename, mode, size, mtime)

    @requires_no_parents_change
    def set_possibly_dirty(self, filename):
        """record that the current state of the file on disk is unknown"""
        self._dirty = True
        self._map.set_possibly_dirty(filename)

49964:f979d8af8648

        return func(self, *args, **kwargs)

    return wrap


def requires_not_changing_parents(func):
    def wrap(self, *args, **kwargs):
        if self.pendingparentchange():
            msg = 'calling `%s` inside of a changing_parents context'
            msg %= func.__name__
            raise error.ProgrammingError(msg)

        return self._map.copymap.get(file, None)

    def copies(self):
        return self._map.copymap

    @requires_not_changing_parents
    def set_tracked(self, filename, reset_copy=False):
        """a "public" method for generic code to mark a file as tracked

        This function is to be called outside of "update/merge" case. For
        example by a command like `hg add X`.

            self._map.copymap.pop(filename, None)
        if pre_tracked:
            self._dirty_tracked_set = True
        return pre_tracked

    @requires_not_changing_parents
    def set_untracked(self, filename):
        """a "public" method for generic code to mark a file as untracked

        This function is to be called outside of "update/merge" case. For
        example by a command like `hg remove X`.

        if ret:
            self._dirty = True
            self._dirty_tracked_set = True
        return ret

    @requires_not_changing_parents
    def set_clean(self, filename, parentfiledata):
        """record that the current state of the file on disk is known to be clean"""
        self._dirty = True
        if not self._map[filename].tracked:
            self._check_new_tracked_filename(filename)
        (mode, size, mtime) = parentfiledata
        self._map.set_clean(filename, mode, size, mtime)

    @requires_not_changing_parents
    def set_possibly_dirty(self, filename):
        """record that the current state of the file on disk is unknown"""
        self._dirty = True
        self._map.set_possibly_dirty(filename)