artem/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2026-04-30 20:46:06 +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 datetime as dt
import typing import typing
from collections import OrderedDict
from threading import Lock from threading import Lock
from yt_dlp.extractor.youtube.pot._provider import BuiltinIEContentProvider 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): def initialize_global_cache(max_size: int):
if _pot_memory_cache.value.get('cache') is None: 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['lock'] = Lock()
_pot_memory_cache.value['max_size'] = max_size _pot_memory_cache.value['max_size'] = max_size
@ -38,7 +37,7 @@ class MemoryLRUPCP(PoTokenCacheProvider, BuiltinIEContentProvider):
def __init__( def __init__(
self, self,
*args, *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, **kwargs,
): ):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
@ -65,7 +64,8 @@ class MemoryLRUPCP(PoTokenCacheProvider, BuiltinIEContentProvider):
self.cache.pop(key) self.cache.pop(key)
self.cache[key] = (value, expires_at) self.cache[key] = (value, expires_at)
if len(self.cache) > self.max_size: 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): def delete(self, key: str):
with self.lock: with self.lock: