sqlmap/lib/core/decorators.py

#!/usr/bin/env python

"""
Copyright (c) 2006-2017 sqlmap developers (http://sqlmap.org/)
See the file 'LICENSE' for copying permission
"""

def cachedmethod(f, cache={}):
    """
    Method with a cached content

    Reference: http://code.activestate.com/recipes/325205-cache-decorator-in-python-24/
    """

    def _(*args, **kwargs):
        try:
            key = hash((f, tuple(args), frozenset(kwargs.items())))
            if key not in cache:
                cache[key] = f(*args, **kwargs)
        except:
            key = hash("".join(str(_) for _ in (f, args, kwargs)))
            if key not in cache:
                cache[key] = f(*args, **kwargs)

        return cache[key]

    return _