comparison mercurial/dirstate.py @ 50108:f7981f202b7a

dirstate: add a `require_changing_any` decorator

We will need it for a couple of usecase (e.g `dirstate.copy`).

50107:cad3a68c0e0c

def requires_changing_files(func):
    def wrap(self, *args, **kwargs):
        if not self.is_changing_files:
            msg = 'calling `%s` outside of a `changing_files`'
            msg %= func.__name__
            raise error.ProgrammingError(msg)
        return func(self, *args, **kwargs)

    return wrap

50108:f7981f202b7a

def requires_changing_files(func):
    def wrap(self, *args, **kwargs):
        if not self.is_changing_files:
            msg = 'calling `%s` outside of a `changing_files`'
            msg %= func.__name__
            raise error.ProgrammingError(msg)
        return func(self, *args, **kwargs)

    return wrap


def requires_changing_any(func):
    def wrap(self, *args, **kwargs):
        if not self.is_changing_any:
            msg = 'calling `%s` outside of a changing context'
            msg %= func.__name__
            raise error.ProgrammingError(msg)
        if self._invalidated_context:
            msg = 'calling `%s` after the dirstate was invalidated'
            raise error.ProgrammingError(msg)
        return func(self, *args, **kwargs)

    return wrap