From b349a3142a0a6c7f570176227b64ec49aec9eb52 Mon Sep 17 00:00:00 2001 From: thekiwismarthome <134335563+thekiwismarthome@users.noreply.github.com> Date: Fri, 13 Feb 2026 21:19:57 +1300 Subject: [PATCH] Update storage.py --- custom_components/shopping_list_manager/storage.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/custom_components/shopping_list_manager/storage.py b/custom_components/shopping_list_manager/storage.py index d87f176..d39fa6d 100644 --- a/custom_components/shopping_list_manager/storage.py +++ b/custom_components/shopping_list_manager/storage.py @@ -22,15 +22,17 @@ _LOGGER = logging.getLogger(__name__) class ShoppingListStorage: """Handle storage for shopping lists.""" - def __init__(self, hass: HomeAssistant, component_path: str) -> None: + def __init__(self, hass: HomeAssistant, component_path: str, country: str = "NZ") -> None: """Initialize storage. Args: hass: Home Assistant instance component_path: Path to the component directory + country: Country code (NZ, AU, US, GB, CA, etc.) """ self.hass = hass self._component_path = component_path + self._country = country # Store country self._store_lists = Store(hass, STORAGE_VERSION, STORAGE_KEY_LISTS) self._store_items = Store(hass, STORAGE_VERSION, STORAGE_KEY_ITEMS) self._store_products = Store(hass, STORAGE_VERSION, STORAGE_KEY_PRODUCTS)