65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
|
from homeassistant.helpers.json import JSONEncoder
|
||
|
|
from homeassistant.helpers.storage import Store
|
||
|
|
from homeassistant.util import json as json_util
|
||
|
|
|
||
|
|
from ..const import VERSION_STORAGE
|
||
|
|
from .logger import LOGGER
|
||
|
|
|
||
|
|
_LOGGER = LOGGER
|
||
|
|
|
||
|
|
class StateGridStore(Store):
|
||
|
|
"""A subclass of Store that allows multiple loads in the executor."""
|
||
|
|
|
||
|
|
def load(self):
|
||
|
|
"""Load the data from disk if version matches."""
|
||
|
|
try:
|
||
|
|
data = json_util.load_json(self.path)
|
||
|
|
except (
|
||
|
|
BaseException # lgtm [py/catch-base-exception] pylint: disable=broad-except
|
||
|
|
) as exception:
|
||
|
|
_LOGGER.critical(
|
||
|
|
"Could not load '%s', restore it from a backup or delete the file: %s",
|
||
|
|
self.path,
|
||
|
|
exception,
|
||
|
|
)
|
||
|
|
if data == {} or data["version"] != self.version:
|
||
|
|
return None
|
||
|
|
return data["data"]
|
||
|
|
|
||
|
|
|
||
|
|
def _get_store_for_key(hass, key, encoder):
|
||
|
|
"""Create a Store object for the key."""
|
||
|
|
return StateGridStore(hass, VERSION_STORAGE, key, encoder=encoder, atomic_writes=True)
|
||
|
|
|
||
|
|
|
||
|
|
def get_store_for_key(hass, key):
|
||
|
|
"""Create a Store object for the key."""
|
||
|
|
return _get_store_for_key(hass, key, JSONEncoder)
|
||
|
|
|
||
|
|
|
||
|
|
async def async_load_from_store(hass, key):
|
||
|
|
"""Load the retained data from store and return de-serialized data."""
|
||
|
|
return await get_store_for_key(hass, key).async_load() or {}
|
||
|
|
|
||
|
|
|
||
|
|
async def async_save_to_store(hass, key, data):
|
||
|
|
"""Generate dynamic data to store and save it to the filesystem.
|
||
|
|
|
||
|
|
The data is only written if the content on the disk has changed
|
||
|
|
by reading the existing content and comparing it.
|
||
|
|
|
||
|
|
If the data has changed this will generate two executor jobs
|
||
|
|
|
||
|
|
If the data has not changed this will generate one executor job
|
||
|
|
"""
|
||
|
|
current = await async_load_from_store(hass, key)
|
||
|
|
if current is None or current != data:
|
||
|
|
await get_store_for_key(hass, key).async_save(data)
|
||
|
|
|
||
|
|
|
||
|
|
async def async_remove_store(hass, key):
|
||
|
|
"""Remove a store element that should no longer be used."""
|
||
|
|
if "/" not in key:
|
||
|
|
return
|
||
|
|
await get_store_for_key(hass, key).async_remove()
|