zware/awx
1
0
Fork 0
mirror of https://github.com/ZwareBear/awx.git synced 2026-04-11 02:21:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

get rid of decorator dependency

Browse Source
This commit is contained in:
AlanCoding
2018-10-30 11:54:36 -04:00
parent 92f0893764
commit d8d710a83d
4 changed files with 29 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
awx/main/utils/common.py
View File

@@ -18,14 +18,11 @@ import contextlib
import tempfile
import six
import psutil
from functools import reduce
from functools import reduce, wraps
from StringIO import StringIO
from decimal import Decimal
# Decorator
from decorator import decorator
# Django
from django.core.exceptions import ObjectDoesNotExist
from django.db import DatabaseError
@@ -136,31 +133,35 @@ def memoize(ttl=60, cache_key=None, track_function=False):
'''
Decorator to wrap a function and cache its result.
'''
if cache_key and track_function:
raise IllegalArgumentError("Can not specify cache_key when track_function is True")
cache = get_memoize_cache()
def _memoizer(f, *args, **kwargs):
if cache_key and track_function:
raise IllegalArgumentError("Can not specify cache_key when track_function is True")
if track_function:
cache_dict_key = slugify('%r %r' % (args, kwargs))
key = slugify("%s" % f.__name__)
cache_dict = cache.get(key) or dict()
if cache_dict_key not in cache_dict:
value = f(*args, **kwargs)
cache_dict[cache_dict_key] = value
cache.set(key, cache_dict, ttl)
def memoize_decorator(f):
@wraps(f)
def _memoizer(*args, **kwargs):
if track_function:
cache_dict_key = slugify('%r %r' % (args, kwargs))
key = slugify("%s" % f.__name__)
cache_dict = cache.get(key) or dict()
if cache_dict_key not in cache_dict:
value = f(*args, **kwargs)
cache_dict[cache_dict_key] = value
cache.set(key, cache_dict, ttl)
else:
value = cache_dict[cache_dict_key]
else:
value = cache_dict[cache_dict_key]
else:
key = cache_key or slugify('%s %r %r' % (f.__name__, args, kwargs))
value = cache.get(key)
if value is None:
value = f(*args, **kwargs)
cache.set(key, value, ttl)
key = cache_key or slugify('%s %r %r' % (f.__name__, args, kwargs))
value = cache.get(key)
if value is None:
value = f(*args, **kwargs)
cache.set(key, value, ttl)
return value
return decorator(_memoizer)
return value
return _memoizer
return memoize_decorator
def memoize_delete(function_name):