mygpo/cache.py

   1 from hashlib import sha1
   2 from django.core.cache import cache
   3 from functools import wraps
   4
   5
   6 NOT_IN_CACHE = object()
   7
   8
   9 def cache_result(**cache_kwargs):
  10     """ Decorator to cache the result of a function call
  11
  12     Usage
  13
  14     @cache_result('myfunc', timeout=60)
  15     def my_function(a, b, c):
  16         pass
  17     """
  18
  19     def _wrapper(f):
  20
  21         @wraps(f)
  22         def _get(*args, **kwargs):
  23
  24             key = sha1(str(f.__module__) + str(f.__name__) + str(args) + str(kwargs)).hexdigest()
  25
  26             # the timeout parameter can't be used when getting from a cache
  27             get_kwargs = dict(cache_kwargs)
  28             get_kwargs.pop('timeout', None)
  29
  30             value = cache.get(key, NOT_IN_CACHE, **get_kwargs)
  31
  32             if value is NOT_IN_CACHE:
  33                 value = f(*args, **kwargs)
  34
  35                 cache.set(key, value, **cache_kwargs)
  36
  37             return value
  38
  39
  40         return _get
  41
  42     return _wrapper