Hirad/pylast
2
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Bugfix for caching between sessions

Browse source
This commit is contained in:
Koen van Zuijlen 2020-12-30 17:11:38 +01:00
parent eca1db8622
commit e9bef6db68
1 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/pylast/__init__.py
View file

@ -422,8 +422,8 @@ class _Network:
""" """
if not file_path: if not file_path:
file_descriptor, file_path = tempfile.mkstemp(prefix="pylast_tmp_") self.cache_backend = _ShelfCacheBackend.create_shelf()
os.close(file_descriptor) return
self.cache_backend = _ShelfCacheBackend(file_path) self.cache_backend = _ShelfCacheBackend(file_path)
@ -784,8 +784,11 @@ class LibreFMNetwork(_Network):
class _ShelfCacheBackend: class _ShelfCacheBackend:
"""Used as a backend for caching cacheable requests.""" """Used as a backend for caching cacheable requests."""
def __init__(self, file_path=None): def __init__(self, file_path=None, flag=None):
self.shelf = shelve.open(file_path, flag="n") if flag:
self.shelf = shelve.open(file_path, flag=flag)
else:
self.shelf = shelve.open(file_path)
self.cache_keys = set(self.shelf.keys()) self.cache_keys = set(self.shelf.keys())
def __contains__(self, key): def __contains__(self, key):
@ -801,6 +804,12 @@ class _ShelfCacheBackend:
self.cache_keys.add(key) self.cache_keys.add(key)
self.shelf[key] = xml_string self.shelf[key] = xml_string
@classmethod
def create_shelf(cls):
file_descriptor, file_path = tempfile.mkstemp(prefix="pylast_tmp_")
os.close(file_descriptor)
return cls(file_path=file_path, flag="n")
class _Request: class _Request:
"""Representing an abstract web service operation.""" """Representing an abstract web service operation."""