You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
128 lines
4.3 KiB
128 lines
4.3 KiB
11 years ago
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
#
|
||
|
|
# Copyright 2011-2013 Codernity (http://codernity.com)
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
|
||
|
|
import functools
|
||
|
|
from random import choice
|
||
|
|
|
||
|
|
|
||
|
|
def cache1lvl(maxsize=100):
|
||
|
|
def decorating_function(user_function):
|
||
|
|
cache1lvl = {}
|
||
|
|
|
||
|
|
@functools.wraps(user_function)
|
||
|
|
def wrapper(key, *args, **kwargs):
|
||
|
|
if isinstance(key, bytes):
|
||
|
|
key = key.decode()
|
||
|
|
# print("cachedddd", key) ## TODO
|
||
|
|
try:
|
||
|
|
#result = cache1lvl[key]
|
||
|
|
result = cache1lvl[key]
|
||
|
|
except KeyError:
|
||
|
|
if len(cache1lvl) == maxsize:
|
||
|
|
for i in range(maxsize // 10 or 1):
|
||
|
|
del cache1lvl[choice(list(cache1lvl.keys()))]
|
||
|
|
## print("#" * 10, key) # TODO
|
||
|
|
## print(user_function) # TODO
|
||
|
|
## print("cache1lvl", key, user_function) # TODO
|
||
|
|
## print(cache1lvl) # TODO
|
||
|
|
cache1lvl[key] = user_function(key, *args, **kwargs)
|
||
|
|
## print(cache1lvl) # TODO
|
||
|
|
result = cache1lvl[key]
|
||
|
|
## print("result caching", result) # TODO
|
||
|
|
# result = user_function(obj, key, *args, **kwargs)
|
||
|
|
if isinstance(result, bytes):
|
||
|
|
result = key.decode()
|
||
|
|
## print("r" * 20, result) # TODO
|
||
|
|
return result
|
||
|
|
|
||
|
|
def clear():
|
||
|
|
cache1lvl.clear()
|
||
|
|
|
||
|
|
def delete(key):
|
||
|
|
if isinstance(key, bytes):
|
||
|
|
key = key.decode()
|
||
|
|
try:
|
||
|
|
del cache1lvl[key]
|
||
|
|
return True
|
||
|
|
except KeyError:
|
||
|
|
return False
|
||
|
|
|
||
|
|
wrapper.clear = clear
|
||
|
|
wrapper.cache = cache1lvl
|
||
|
|
wrapper.delete = delete
|
||
|
|
return wrapper
|
||
|
|
return decorating_function
|
||
|
|
|
||
|
|
|
||
|
|
def cache2lvl(maxsize=100):
|
||
|
|
def decorating_function(user_function):
|
||
|
|
cache = {}
|
||
|
|
|
||
|
|
@functools.wraps(user_function)
|
||
|
|
def wrapper(*args, **kwargs):
|
||
|
|
# return user_function(*args, **kwargs)
|
||
|
|
try:
|
||
|
|
result = cache[args[0]][args[1]]
|
||
|
|
except KeyError:
|
||
|
|
# print wrapper.cache_size
|
||
|
|
if wrapper.cache_size == maxsize:
|
||
|
|
to_delete = maxsize // 10 or 1
|
||
|
|
for i in range(to_delete):
|
||
|
|
key1 = choice(list(cache.keys()))
|
||
|
|
key2 = choice(list(cache[key1].keys()))
|
||
|
|
del cache[key1][key2]
|
||
|
|
if not cache[key1]:
|
||
|
|
del cache[key1]
|
||
|
|
wrapper.cache_size -= to_delete
|
||
|
|
# print wrapper.cache_size
|
||
|
|
result = user_function(*args, **kwargs)
|
||
|
|
try:
|
||
|
|
cache[args[0]][args[1]] = result
|
||
|
|
except KeyError:
|
||
|
|
cache[args[0]] = {args[1]: result}
|
||
|
|
wrapper.cache_size += 1
|
||
|
|
return result
|
||
|
|
|
||
|
|
def clear():
|
||
|
|
cache.clear()
|
||
|
|
wrapper.cache_size = 0
|
||
|
|
|
||
|
|
def delete(key, inner_key=None):
|
||
|
|
if inner_key:
|
||
|
|
try:
|
||
|
|
del cache[key][inner_key]
|
||
|
|
if not cache[key]:
|
||
|
|
del cache[key]
|
||
|
|
wrapper.cache_size -= 1
|
||
|
|
return True
|
||
|
|
except KeyError:
|
||
|
|
return False
|
||
|
|
else:
|
||
|
|
try:
|
||
|
|
wrapper.cache_size -= len(cache[key])
|
||
|
|
del cache[key]
|
||
|
|
return True
|
||
|
|
except KeyError:
|
||
|
|
return False
|
||
|
|
|
||
|
|
wrapper.clear = clear
|
||
|
|
wrapper.cache = cache
|
||
|
|
wrapper.delete = delete
|
||
|
|
wrapper.cache_size = 0
|
||
|
|
return wrapper
|
||
|
|
return decorating_function
|