Source code for shuup.utils.decorators

# This file is part of Shuup.
#
# Copyright (c) 2012-2021, Shuup Commerce Inc. All rights reserved.
#
# This source code is licensed under the OSL-3.0 license found in the
# LICENSE file in the root directory of this source tree.
from __future__ import unicode_literals

import functools
import traceback


def non_reentrant(func):
    @functools.wraps(func)
    def wrapped(self, *args, **kwargs):
        name = func.__name__
        if not hasattr(self, "_non_reentrant_check"):
            self._non_reentrant_check = {}
        invocation_stack = self._non_reentrant_check.get(name)
        if invocation_stack:
            msg = "Error! Trying to re-entrantly call %s. Last invocation was" % name
            stack_lines = traceback.format_list(invocation_stack)
            raise RuntimeError(msg, stack_lines)
        self._non_reentrant_check[name] = traceback.extract_stack()
        try:
            return func(self, *args, **kwargs)
        finally:
            del self._non_reentrant_check[name]

    return wrapped