jacebrowning / yorm

log = logging.getLogger(__name__)

def load_before(method):

    """Decorator for methods that should load before call."""

    if getattr(method, '_load_before', False):

        return method

    @functools.wraps(method)

    def wrapped(self, *args, **kwargs):

        """Decorated method."""

        __tracebackhide__ = True  # pylint: disable=unused-variable

        if not _private_call(method, args):

            mapper = common.get_mapper(self)

            if mapper and mapper.modified:

                log.debug("Loading before call: %s", method.__name__)

                mapper.load()

                if mapper.auto_save_after_load:

                    mapper.save()

                    mapper.modified = False

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

    setattr(wrapped, '_load_before', True)

    return wrapped

def save_after(method):

    return wrapped

def save_after(method):

    """Decorator for methods that should save after call."""

    if getattr(method, '_save_after', False):

        return method

    @functools.wraps(method)

    def wrapped(self, *args, **kwargs):

        """Decorated method."""

        __tracebackhide__ = True  # pylint: disable=unused-variable

        result = method(self, *args, **kwargs)

        if not _private_call(method, args):

            mapper = common.get_mapper(self)

            if mapper and mapper.auto_save:

                log.debug("Saving after call: %s", method.__name__)

                mapper.save()

        return result

    setattr(wrapped, '_save_after', True)

    return wrapped

def _private_call(method, args, prefix='_'):