jacebrowning / yorm

_STORE_AFTER_METHODS = set()

def load_before(method):

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

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

        return method

    _LOAD_BEFORE_METHODS.add(method.__name__)

    @functools.wraps(method)

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

        """Decorated method."""

        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

    _STORE_AFTER_METHODS.add(method.__name__)

    @functools.wraps(method)

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

        """Decorated method."""

        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='_'):