from django.conf import settings
from django.core.cache import cache
from functools import wraps


def cached_method(cache_key, timeout=settings.CACHE_TIMEOUT):
    """Cache a method, using it's first argument to set a cache key.

    Params:

    * cache_key is a format string used to set a cache key, e.g. "{}-foo"

    Usage:

        class SomeThing:
            @cached_method("{}-key")
            def get_stuff(self, obj):
                # ...

    In the amove `get_stuff` method, obj.id will be used to generate the cache
    key. NOTE: this was intended to be used on serializer methods.

    """
    def decorate(func):

        @wraps(func)
        def wrapper(*args, **kwargs):
            if len(args) > 1:  # extract the first arg & use as the cache key
                obj_id = getattr(args[1], "id", str(args[1]))
                key = cache_key.format(obj_id)
                result = cache.get(key)
                if result is None:
                    result = func(*args, **kwargs)
                    cache.set(key, result, timeout=timeout)
                return result
            # Nothing to use as a cache key, just call the method.
            return func(*args, **kwargs)
        return wrapper
    return decorate

from django.conf import settings
from django.core.cache import cache
from functools import wraps


def cached_method(cache_key, timeout=settings.CACHE_TIMEOUT):
    """Cache a method, using the ID attribute of it's first argument to set
    a cache key. NOTE: If this first argument for the cached method doesn't
    have an ID attribute, nothing will happen.

    Params:

    * cache_key is a format string used to set a cache key, e.g. "{}-foo"

    Usage:

        class SomeThing:
            @cached_method("{}-key")
            def get_stuff(self, obj):
                # ...

    In the above `get_stuff` method, `obj.id` will be used to generate the
    cache key. NOTE: this was intended to be used on serializer methods.

    """
    def decorate(func):

        @wraps(func)
        def wrapper(*args, **kwargs):
            if len(args) > 1:
                # extract the first objected passed into the function & use
                # its id attribute as part of the cache key
                cache_object = args[1]
                if not hasattr(cache_object, 'id'):
                    return None  # just bail if there's no ID.

                key = cache_key.format(cache_object.id)
                result = cache.get(key)
                if result is None:
                    result = func(*args, **kwargs)
                    cache.set(key, result, timeout=timeout)
                return result
            # Nothing to use as a cache key, just call the method.
            return func(*args, **kwargs)
        return wrapper
    return decorate

Update for cached_method so it doesn't try to do anythin if the first argument doesn't have an ID attribute
Update for cached_method so it doesn't try to do anythin if the first argument doesn't have an ID attribute
