Add new cache_parameters() method for functools.lru_cache() to better support pickling

Author: Zheaoli

Doc/library/functools.rst

@@ -111,7 +111,9 @@ The :mod:`functools` module defines the following functions:
    To help measure the effectiveness of the cache and tune the *maxsize*
    parameter, the wrapped function is instrumented with a :func:`cache_info`
    function that returns a :term:`named tuple` showing *hits*, *misses*,
-   *maxsize* and *currsize*.  In a multi-threaded environment, the hits
+   *maxsize* and *currsize*. the wrapped function is also instrumented woth
+   a :func:`cache_parameters` function that returns a :class:`dict` which has
+   showing *maxsize* and *typed*. In a multi-threaded environment, the hits
    and misses are approximate.
 
    The decorator also provides a :func:`cache_clear` function for clearing or
@@ -178,6 +180,9 @@ The :mod:`functools` module defines the following functions:
    .. versionchanged:: 3.8
       Added the *user_function* option.
 
+   .. versionadded:: 3.9
+      Added the function :func:`cache_parameters`
+
 .. decorator:: total_ordering
 
    Given a class defining one or more rich comparison ordering methods, this

Lib/functools.py

@@ -499,13 +499,15 @@ def lru_cache(maxsize=128, typed=False):
         # The user_function was passed in directly via the maxsize argument
         user_function, maxsize = maxsize, 128
         wrapper = _lru_cache_wrapper(user_function, maxsize, typed, _CacheInfo)
+        wrapper.cache_parameters =  lambda: {'maxsize': maxsize, 'typed': typed}
         return update_wrapper(wrapper, user_function)
     elif maxsize is not None:
         raise TypeError(
             'Expected first argument to be an integer, a callable, or None')
 
     def decorating_function(user_function):
         wrapper = _lru_cache_wrapper(user_function, maxsize, typed, _CacheInfo)
+        wrapper.cache_parameters =  lambda: {'maxsize': maxsize, 'typed': typed}
         return update_wrapper(wrapper, user_function)
 
     return decorating_function
@@ -621,7 +623,6 @@ def cache_clear():
             root[:] = [root, root, None, None]
             hits = misses = 0
             full = False
-
     wrapper.cache_info = cache_info
     wrapper.cache_clear = cache_clear
     return wrapper

Lib/test/test_functools.py

@@ -1655,6 +1655,19 @@ def orig(x, y):
                 f_copy = copy.deepcopy(f)
                 self.assertIs(f_copy, f)
 
+    def test_lru_cache_parameters(self):
+        def f():
+            return 1
+        f = self.module.lru_cache(2)(f)
+        self.assertEqual(f.cache_parameters().get("maxsize"), 2)
+        self.assertEqual(f.cache_parameters().get("typed"), False)
+
+        @self.module.lru_cache(maxsize=1000, typed=True)
+        def f():
+            return 1
+        self.assertEqual(f.cache_parameters().get("maxsize"), 1000)
+        self.assertEqual(f.cache_parameters().get("typed"), True)
+
 
 @py_functools.lru_cache()
 def py_cached_func(x, y):

Misc/NEWS.d/next/Library/2019-10-24-08-10-30.bpo-38565.SWSUst.rst

@@ -0,0 +1 @@
+Add new cache_parameters() method for functools.lru_cache() to better support pickling.