gh-102618: Introduce a functools.cached_method decorator#150002
gh-102618: Introduce a functools.cached_method decorator#150002sirosen wants to merge 15 commits into
functools.cached_method decorator#150002Conversation
Documentation build overview
245 files changed ·
|
resolves python#102618 This definition of `cached_method` is based on the discussion on DPO: https://discuss.python.org/t/107164 Some tradeoffs need to be made in any version of such a decorator. In this particular implementation, the choices made are as follows: - lru_cache will be used under the hood, and `cache_clear()` and `cache_info()` will be exposed - caches will be stored in a separate dict, indexed by `id(self)` -- meaning instances do not need to be hashable and will not share caches - weakrefs will be used to delete entries from the cache, so the instances must be weak-referencable - lru_cache itself is not threadsafe, but initialization of the caches is threadsafe -- this avoids confusing scenarios in which cache entries "disappear" New documentation is included, marked for 3.16, and a small number of new tests are added.
Thanks to feedback from @jaraco, this adjustment makes the public interface for `cached_method` a function which is responsible for the argument management, rather than trying to use `__call__` on the descriptor itself. This frees up `__call__` to lookup the relevant LRU-cached function and invoke it, which makes the `_cached_method` descriptor suitable for use as a `property.fget` callable. Tests are updated to indicate that a decorator can be prepared and then used repeatedly (which was previously an explicit error), and can be used under a property decorator.
- Fix doc references - Use imperative mood in a comment Co-authored-by: Jason R. Coombs <308610+jaraco@users.noreply.github.com>
|
Jelle pointed out to me that although I have tried to be careful about the locking I included, to minimize contention, the lock I defined is per- I'll push a change to remove the locking and clarify the thread-safety documentation. |
|
@willingc , this is the PR that I mentioned to you; you said you wanted a ping to take a look. 🙂 |
Although this logic tried to avoid lock contention, there were still scenarios under which it was possible and could negatively impact users. Removal and adjustment of the documentation puts responsibility for locking/safety onto users, similar to other parts of the stdlib.
|
FWIW. The |
Co-authored-by: Edgar Ramírez Mondragón <16805946+edgarrmondragon@users.noreply.github.com>
jaraco
left a comment
There was a problem hiding this comment.
In my opinion, this implementation is fairly clean and simple. The documentation shows pretty clearly what the user's expectations should be. The tradeoffs chosen have been well considered and a reasonable path elected. I propose to get this merged and give it some exposure through the beta releases. From there iterate or pivot.
StanFromIreland
left a comment
There was a problem hiding this comment.
This needs a What's New in 3.16 entry.
Rather than doing an old-style deferred import, with explanatory comments for why the deferred import style is used. Convert these to a top-level `lazy import weakref`. No explanatory comment is used, on the grounds that the new syntax is sufficiently low in cognitive cost that experienced maintainers will quickly recognize it and assume that the pattern is used to either break a cycle or reduce import costs.
|
I've applied the lazy import change. My commit message explains why I did not preserve any comment about it, but I'm open to feedback on that. I also added the What's New entry. |
hugovk
left a comment
There was a problem hiding this comment.
We have docs tooling that will update next to whatever the next pre-release is, for example:
"Added in version 3.16.0a0 (unreleased)"
|
|
||
| * Added :func:`functools.cached_method` decorator for applying a cache to | ||
| instance methods. | ||
| (Contributed by Stephen Rosen in :gh:`150002`.) |
There was a problem hiding this comment.
| (Contributed by Stephen Rosen in :gh:`150002`.) | |
| (Contributed by Stephen Rosen in :gh:`102618`.) |
Issue is preferred here IIRC, right Hugo?
There was a problem hiding this comment.
Is that right? That seems odd. "Contributed by X in Y" suggests to me that "Y" is the implementation (PR), not specification (issue).
| def cached_method(func=None, /, maxsize=None, typed=False): | ||
| if func is None: | ||
| def decorator(func): | ||
| return _cached_method(func, maxsize=maxsize, typed=typed) |
There was a problem hiding this comment.
lru_cache does a little dance to ensure maxsize is not junk:
Lines 586 to 600 in 1604043
I think we should do it here too to avoid confusing errors.
There was a problem hiding this comment.
It came out slightly differently in 8086aa5 because the signatures are a bit different, but the idea is the same.
cached_method is able to have a slightly better signature because it uses a positional-only param, not having the burdens of history.
|
|
||
| def _cached_method_weakref_callback(cache_dict, id_key): | ||
| def callback(ref): | ||
| cache_dict.pop(id_key) |
There was a problem hiding this comment.
No default, what about races?
There was a problem hiding this comment.
I didn't realize, at first, that a race was possible, but I'm pretty sure it is if two concurrent writers create separate weakrefs. I'll add pop(id_key, None).
Do we want to further double-check for this kind of safety?
Co-authored-by: Hugo van Kemenade <1324225+hugovk@users.noreply.github.com> Co-authored-by: Stan Ulbrych <stan@python.org>
Co-authored-by: Stan Ulbrych <stan@python.org>
And add it to functools' `__all__` list.
Multiple weakrefs can be created when there are concurrent writes to a "new cache". When they finalize, their attempts to clear the dict should be kept safe by using a pop() default.
resolves #102618
This definition of
cached_methodis based on the discussion on DPO:https://discuss.python.org/t/107164
Some tradeoffs need to be made in any version of such a decorator. In this particular implementation, the choices made are as follows:
lru_cache will be used under the hood, and
cache_clear()andcache_info()will be exposedcaches will be stored in a separate dict, indexed by
id(self)-- meaning instances do not need to be hashable and will not share cachesweakrefs will be used to delete entries from the cache, so the instances must be weak-referencable
lru_cache itself is not threadsafe, but initialization of the caches is threadsafe -- this avoids confusing scenarios in which cache entries "disappear"EDIT: in response to review feedback, this locking was removed; users can lock if they want thread safety
New documentation is included, marked for 3.16, and a small number of new
tests are added.
In addition to @rhettinger's review, as the listed owner for functools, @sobolevn expressed interest in reviewing changes.
@kenahoo and @stevendaprano may also want a look based on their prior interest.
Everyone I've spoken with about this change1 has been supportive of the core idea. In particular @jaraco noted that he would support this addition, and maintains a very different implementation. There are many other ways of writing this, with different tradeoffs. Perhaps the most notable class of those are versions which store caches on the instance itself, like this one suggested here in the DPO discussion -- this writes the cached method into
__dict__, similarly tocached_property.Of course, I'm happy to alter or re-approach any part of this per review! 😄
Footnotes
at PyCon US 2026 ↩