mirror of
https://github.com/clearml/clearml
synced 2025-02-01 01:26:49 +00:00
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import json
|
|
from . import get_cache_dir
|
|
from .defs import SESSION_CACHE_FILE
|
|
|
|
|
|
class SessionCache(object):
|
|
"""
|
|
Handle SDK session cache.
|
|
TODO: Improve error handling to something like "except (FileNotFoundError, PermissionError, JSONDecodeError)"
|
|
TODO: that's both six-compatible and tested
|
|
"""
|
|
@classmethod
|
|
def _load_cache(cls):
|
|
try:
|
|
with (get_cache_dir() / SESSION_CACHE_FILE).open("rt") as fp:
|
|
return json.load(fp)
|
|
except Exception:
|
|
return {}
|
|
|
|
@classmethod
|
|
def _store_cache(cls, cache):
|
|
try:
|
|
get_cache_dir().mkdir(parents=True, exist_ok=True)
|
|
with (get_cache_dir() / SESSION_CACHE_FILE).open("wt") as fp:
|
|
json.dump(cache, fp)
|
|
except Exception:
|
|
pass
|
|
|
|
@classmethod
|
|
def store_dict(cls, unique_cache_name, dict_object):
|
|
# type: (str, dict) -> None
|
|
cache = cls._load_cache()
|
|
cache[unique_cache_name] = dict_object
|
|
cls._store_cache(cache)
|
|
|
|
@classmethod
|
|
def load_dict(cls, unique_cache_name):
|
|
# type: (str) -> dict
|
|
cache = cls._load_cache()
|
|
return cache.get(unique_cache_name, {}) if cache else {}
|