diff options
| author | Jordan Cook <jordan.cook@pioneer.com> | 2022-04-20 16:50:40 -0500 |
|---|---|---|
| committer | Jordan Cook <jordan.cook@pioneer.com> | 2022-04-22 17:28:33 -0500 |
| commit | 0dbd82d4d28875f2c0a592dfc89f50bf1c63cb2b (patch) | |
| tree | f3982671c81005c29c39fbd6da241e79509cc0cf /requests_cache/backends/gridfs.py | |
| parent | 57579af3a5c4e683f2dd96f493471077808d1d39 (diff) | |
| download | requests-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.py | 8 |
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: |
