artem/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2026-04-06 17:03:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

use python dict for memory cache

Browse Source
This commit is contained in:
coletdjnz 2025-05-18 12:49:34 +12:00
parent 68bc679666
commit 3cfde45e46
No known key found for this signature in database
GPG Key ID: 91984263BB39894A
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
yt_dlp/extractor/youtube/pot/_builtin/memory_cache.py
View File

@ -2,7 +2,6 @@ from __future__ import annotations
import datetime as dt
import typing
from collections import OrderedDict
from threading import Lock
from yt_dlp.extractor.youtube.pot._provider import BuiltinIEContentProvider
@ -16,7 +15,7 @@ from yt_dlp.extractor.youtube.pot.cache import (
def initialize_global_cache(max_size: int):
if _pot_memory_cache.value.get('cache') is None:
_pot_memory_cache.value['cache'] = OrderedDict()
_pot_memory_cache.value['cache'] = {}
_pot_memory_cache.value['lock'] = Lock()
_pot_memory_cache.value['max_size'] = max_size
@ -38,7 +37,7 @@ class MemoryLRUPCP(PoTokenCacheProvider, BuiltinIEContentProvider):
def __init__(
self,
*args,
initialize_cache: typing.Callable[[int], tuple[OrderedDict[str, tuple[str, int]], Lock, int]] = initialize_global_cache,
initialize_cache: typing.Callable[[int], tuple[dict[str, tuple[str, int]], Lock, int]] = initialize_global_cache,
**kwargs,
):
super().__init__(*args, **kwargs)
@ -65,7 +64,8 @@ class MemoryLRUPCP(PoTokenCacheProvider, BuiltinIEContentProvider):
self.cache.pop(key)
self.cache[key] = (value, expires_at)
if len(self.cache) > self.max_size:
self.cache.popitem(last=False)
oldest_key = next(iter(self.cache))
self.cache.pop(oldest_key)
def delete(self, key: str):
with self.lock: