35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
|
"""Storage handers."""
|
||
|
|
# pylint: disable=import-outside-toplevel
|
||
|
|
from homeassistant.helpers.json import JSONEncoder
|
||
|
|
|
||
|
|
from custom_components.hacs.const import VERSION_STORAGE
|
||
|
|
|
||
|
|
|
||
|
|
def get_store_for_key(hass, key):
|
||
|
|
"""Create a Store object for the key."""
|
||
|
|
key = key if "/" in key else f"hacs.{key}"
|
||
|
|
from homeassistant.helpers.storage import Store
|
||
|
|
|
||
|
|
return Store(hass, VERSION_STORAGE, key, encoder=JSONEncoder)
|
||
|
|
|
||
|
|
|
||
|
|
async def async_load_from_store(hass, key):
|
||
|
|
"""Load the retained data from store and return de-serialized data."""
|
||
|
|
store = get_store_for_key(hass, key)
|
||
|
|
restored = await store.async_load()
|
||
|
|
if restored is None:
|
||
|
|
return {}
|
||
|
|
return restored
|
||
|
|
|
||
|
|
|
||
|
|
async def async_save_to_store(hass, key, data):
|
||
|
|
"""Generate dynamic data to store and save it to the filesystem."""
|
||
|
|
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()
|