MarcoBuster/youtube-dl
1
1
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl synced 2024-11-11 11:39:27 +01:00
Code Issues Releases Wiki Activity

[utils] Implement cache for OnDemandPagedList

Browse Source
This commit is contained in:
Yen Chi Hsuan 2016-02-23 12:17:02 +08:00
parent 86a7dbe66e
commit b95dc034ca
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
youtube_dl/utils.py
View File

@ -1570,9 +1570,12 @@ class PagedList(object):
class OnDemandPagedList(PagedList):
def __init__(self, pagefunc, pagesize):
def __init__(self, pagefunc, pagesize, use_cache=False):
self._pagefunc = pagefunc
self._pagesize = pagesize
self._use_cache = use_cache
if use_cache:
self._cache = {}
def getslice(self, start=0, end=None):
res = []
@ -1582,7 +1585,13 @@ class OnDemandPagedList(PagedList):
if start >= nextfirstid:
continue
page_results = list(self._pagefunc(pagenum))
page_results = None
if self._use_cache:
page_results = self._cache.get(pagenum)
if page_results is None:
page_results = list(self._pagefunc(pagenum))
if self._use_cache:
self._cache[pagenum] = page_results
startv = (
start % self._pagesize