summaryrefslogtreecommitdiff
path: root/requests_cache/backends/gridfs.py
diff options
context:
space:
mode:
authorJordan Cook <jordan.cook@pioneer.com>2022-04-20 16:50:40 -0500
committerJordan Cook <jordan.cook@pioneer.com>2022-04-22 17:28:33 -0500
commit0dbd82d4d28875f2c0a592dfc89f50bf1c63cb2b (patch)
treef3982671c81005c29c39fbd6da241e79509cc0cf /requests_cache/backends/gridfs.py
parent57579af3a5c4e683f2dd96f493471077808d1d39 (diff)
downloadrequests-cache-0dbd82d4d28875f2c0a592dfc89f50bf1c63cb2b.tar.gz
Merge *PickleDict storage classes into parent classes
Diffstat (limited to 'requests_cache/backends/gridfs.py')
-rw-r--r--requests_cache/backends/gridfs.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/requests_cache/backends/gridfs.py b/requests_cache/backends/gridfs.py
index eb139d5..dc66323 100644
--- a/requests_cache/backends/gridfs.py
+++ b/requests_cache/backends/gridfs.py
@@ -29,7 +29,7 @@ class GridFSCache(BaseCache):
def __init__(self, db_name: str, **kwargs):
super().__init__(cache_name=db_name, **kwargs)
- self.responses = GridFSPickleDict(db_name, **kwargs)
+ self.responses = GridFSDict(db_name, **kwargs)
self.redirects = MongoDict(
db_name, collection_name='redirects', connection=self.responses.connection, **kwargs
)
@@ -39,7 +39,7 @@ class GridFSCache(BaseCache):
return super().remove_expired_responses(*args, **kwargs)
-class GridFSPickleDict(BaseStorage):
+class GridFSDict(BaseStorage):
"""A dictionary-like interface for a GridFS database
Args:
@@ -63,13 +63,13 @@ class GridFSPickleDict(BaseStorage):
result = self.fs.find_one({'_id': key})
if result is None:
raise KeyError
- return self.serializer.loads(result.read())
+ return self.deserialize(result.read())
except CorruptGridFile as e:
logger.warning(e, exc_info=True)
raise KeyError
def __setitem__(self, key, item):
- value = self.serializer.dumps(item)
+ value = self.serialize(item)
encoding = None if isinstance(value, bytes) else 'utf-8'
with self._lock: