anas/spotipy
1
0
Fork 0
mirror of https://github.com/spotipy-dev/spotipy.git synced 2026-06-19 01:03:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add encoder_cls argument to CacheFileHandler (#941)

Browse Source 
Fixes
This commit is contained in:
Shawn Cruz 2023-02-12 17:58:55 -05:00 committed by GitHub
parent c53511bbbe
commit b3f308d289
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CHANGELOG.md
View File

@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
## Unreleased ## Unreleased
- Added optional `encoder_cls` argument to `CacheFileHandler`, which overwrite default encoder for token before writing to disk
### Added ### Added
### Fixed ### Fixed

9
spotipy/cache_handler.py
View File

@ -50,15 +50,18 @@ class CacheFileHandler(CacheHandler):
def __init__(self, def __init__(self,
cache_path=None, cache_path=None,
username=None): username=None,
encoder_cls=None):
""" """
Parameters: Parameters:
* cache_path: May be supplied, will otherwise be generated * cache_path: May be supplied, will otherwise be generated
(takes precedence over `username`) (takes precedence over `username`)
* username: May be supplied or set as environment variable * username: May be supplied or set as environment variable
(will set `cache_path` to `.cache-{username}`) (will set `cache_path` to `.cache-{username}`)
* encoder_cls: May be supplied as a means of overwriting the
default serializer used for writing tokens to disk
""" """
self.encoder_cls = encoder_cls
if cache_path: if cache_path:
self.cache_path = cache_path self.cache_path = cache_path
else: else:
@ -88,7 +91,7 @@ class CacheFileHandler(CacheHandler):
def save_token_to_cache(self, token_info): def save_token_to_cache(self, token_info):
try: try:
f = open(self.cache_path, "w") f = open(self.cache_path, "w")
f.write(json.dumps(token_info)) f.write(json.dumps(token_info, cls=self.encoder_cls))
f.close() f.close()
except IOError: except IOError:
logger.warning('Couldn\'t write token to cache at: %s', logger.warning('Couldn\'t write token to cache at: %s',