[Cache] Add clear function

This commit is contained in:
dirkf 2025-05-02 13:17:18 +01:00
parent c052a16f72
commit ec75141bf0
2 changed files with 43 additions and 11 deletions

View File

@ -66,6 +66,18 @@ class TestCache(unittest.TestCase):
new_version = '.'.join(('%0.2d' % ((v + 1) if i == 0 else v, )) for i, v in enumerate(version_tuple(__version__))) new_version = '.'.join(('%0.2d' % ((v + 1) if i == 0 else v, )) for i, v in enumerate(version_tuple(__version__)))
self.assertIs(c.load('test_cache', 'k.', min_ver=new_version), None) self.assertIs(c.load('test_cache', 'k.', min_ver=new_version), None)
def test_cache_clear(self):
ydl = FakeYDL({
'cachedir': self.test_dir,
})
c = Cache(ydl)
c.store('test_cache', 'k.', 'kay')
c.store('test_cache', 'l.', 'ell')
self.assertEqual(c.load('test_cache', 'k.'), 'kay')
c.clear('test_cache', 'k.')
self.assertEqual(c.load('test_cache', 'k.'), None)
self.assertEqual(c.load('test_cache', 'l.'), 'ell')
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@ -1,6 +1,7 @@
# coding: utf-8 # coding: utf-8
from __future__ import unicode_literals from __future__ import unicode_literals
import errno
import json import json
import os import os
import re import re
@ -8,7 +9,6 @@ import shutil
import traceback import traceback
from .compat import ( from .compat import (
compat_contextlib_suppress,
compat_getenv, compat_getenv,
compat_open as open, compat_open as open,
compat_os_makedirs, compat_os_makedirs,
@ -78,6 +78,22 @@ class Cache(object):
tb = traceback.format_exc() tb = traceback.format_exc()
self._report_warning('Writing cache to {fn!r} failed: {tb}'.format(fn=fn, tb=tb)) self._report_warning('Writing cache to {fn!r} failed: {tb}'.format(fn=fn, tb=tb))
def clear(self, section, key, dtype='json'):
if not self.enabled:
return
fn = self._get_cache_fn(section, key, dtype)
self._write_debug('Clearing {section}.{key} from cache'.format(section=section, key=key))
try:
os.remove(fn)
except Exception as e:
if getattr(e, 'errno') == errno.ENOENT:
# file not found
return
tb = traceback.format_exc()
self._report_warning('Clearing cache from {fn!r} failed: {tb}'.format(fn=fn, tb=tb))
def _validate(self, data, min_ver): def _validate(self, data, min_ver):
version = traverse_obj(data, self._VERSION_KEY) version = traverse_obj(data, self._VERSION_KEY)
if not version: # Backward compatibility if not version: # Backward compatibility
@ -94,17 +110,21 @@ class Cache(object):
return default return default
cache_fn = self._get_cache_fn(section, key, dtype) cache_fn = self._get_cache_fn(section, key, dtype)
with compat_contextlib_suppress(IOError): # If no cache available try:
with open(cache_fn, encoding='utf-8') as cachef:
self._write_debug('Loading {section}.{key} from cache'.format(section=section, key=key), only_once=True)
return self._validate(json.load(cachef), min_ver)
except (ValueError, KeyError):
try: try:
with open(cache_fn, encoding='utf-8') as cachef: file_size = 'size: %d' % os.path.getsize(cache_fn)
self._write_debug('Loading {section}.{key} from cache'.format(section=section, key=key), only_once=True) except (OSError, IOError) as oe:
return self._validate(json.load(cachef), min_ver) file_size = error_to_compat_str(oe)
except (ValueError, KeyError): self._report_warning('Cache retrieval from %s failed (%s)' % (cache_fn, file_size))
try: except Exception as e:
file_size = os.path.getsize(cache_fn) if getattr(e, 'errno') == errno.ENOENT:
except (OSError, IOError) as oe: # no cache available
file_size = error_to_compat_str(oe) return
self._report_warning('Cache retrieval from %s failed (%s)' % (cache_fn, file_size)) self._report_warning('Cache retrieval from %s failed' % (cache_fn,))
return default return default