Ashish/yt-dlp
1
1
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-12-22 18:17:17 -05:00
Code Issues 7 Projects Releases Wiki Activity

[utils] classproperty: Add cache support

Browse Source
This commit is contained in:
pukkandan 2022-11-13 08:29:49 +05:30
parent 0a4b2f4180
commit 83cc7b8aae
No known key found for this signature in database
GPG Key ID: 7EEE9E1E817D0A39
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
yt_dlp/utils.py
View File

@ -5847,14 +5847,23 @@ def cached_method(f):
class classproperty:
"""property access for class methods"""
"""property access for class methods with optional caching"""
def __new__(cls, func=None, *args, **kwargs):
if not func:
return functools.partial(cls, *args, **kwargs)
return super().__new__(cls)
def __init__(self, func):
def __init__(self, func, *, cache=False):
functools.update_wrapper(self, func)
self.func = func
self._cache = {} if cache else None
def __get__(self, _, cls):
return self.func(cls)
if self._cache is None:
return self.func(cls)
elif cls not in self._cache:
self._cache[cls] = self.func(cls)
return self._cache[cls]
class Namespace(types.SimpleNamespace):