Effacer tout lru_cache en Python

Question

J'ai des fonctions en python qui ont des caches avec lru_cache e.g.

@lru_cache(maxsize=None) 
def my_function(): 
    ... 

Bien que je puisse effacer individuellement les caches avec par ex. my_function.cache_clear() est-il un moyen d'effacer les caches de chaque fonction à la fois? [Je pensais qu'il y avait peut-être un moyen de retourner tous les noms de fonctions chargés en mémoire, puis de les faire défiler en effaçant le cache de chacun].

Je cherche spécifiquement à mettre en œuvre dans le cadre d'un repli, pour les cas où, par exemple, 90% de la mémoire sur ma machine est utilisée.

Answer 1

Vous pouvez créer un décorateur modifié qui prend également note des fonctions mises en cache:

cached_functions = [] 

def clearable_lru_cache(*args, **kwargs): 
    def decorator(func): 
     func = lru_cache(*args, **kwargs)(func) 
     cached_functions.append(func) 
     return func 

    return decorator 

def clear_all_cached_functions(): 
    for func in cached_functions: 
     func.cache_clear() 

Si vous voulez vraiment, vous pouvez également utiliser rapiéçage de singe pour remplacer le décorateur d'origine.

Test:

@clearable_lru_cache() 
def foo(x): 
    print('foo', x) 

@clearable_lru_cache() 
def bar(x): 
    print('bar', x) 

for i in [1, 2]: 
    for j in [1, 2]: 
     print('Calling functions') 
     for k in [1, 2, 3]: 
      for f in [foo, bar]: 
       f(k) 
     print('Functions called - if you saw nothing they were cached') 
    print('Clearing cache') 
    clear_all_cached_functions() 

Sortie:

Calling functions 
foo 1 
bar 1 
foo 2 
bar 2 
foo 3 
bar 3 
Functions called - if you saw nothing they were cached 
Calling functions 
Functions called - if you saw nothing they were cached 
Clearing cache 
Calling functions 
foo 1 
bar 1 
foo 2 
bar 2 
foo 3 
bar 3 
Functions called - if you saw nothing they were cached 
Calling functions 
Functions called - if you saw nothing they were cached 
Clearing cache