Update storage.py

This commit is contained in:
thekiwismarthome
2026-02-13 21:21:58 +13:00
committed by GitHub
parent b349a3142a
commit e09f9004a6
@@ -90,23 +90,23 @@ class ShoppingListStorage:
_LOGGER.debug("Loaded %d categories", len(self._categories)) _LOGGER.debug("Loaded %d categories", len(self._categories))
else: else:
# Initialize with default categories from JSON file # Initialize with default categories from JSON file
country_code = getattr(self.hass.config, 'country', None) default_categories = load_categories(self._component_path, self._country) # Use self._country
default_categories = load_categories(self._component_path, country_code)
self._categories = [Category(**cat) for cat in default_categories] self._categories = [Category(**cat) for cat in default_categories]
await self._save_categories() await self._save_categories()
_LOGGER.info( _LOGGER.info(
"Initialized %d default categories for country: %s", "Initialized %d default categories for country: %s",
len(self._categories), len(self._categories),
country_code or "default" self._country # Use self._country
) )
# NEW: Load product catalog if products are empty # Load product catalog if products are empty
if not self._products: if not self._products:
country_code = getattr(self.hass.config, 'country', None) _LOGGER.info("Loading product catalog for country: %s", self._country)
catalog_products = load_product_catalog(self._component_path, country_code) catalog_products = load_product_catalog(self._component_path, self._country) # Use self._country
if catalog_products: if catalog_products:
_LOGGER.info("Importing %d products from catalog", len(catalog_products)) _LOGGER.info("Importing %d products from catalog", len(catalog_products))
# ... rest of import code ...
for prod_data in catalog_products: for prod_data in catalog_products:
try: try:
# Create Product from catalog data # Create Product from catalog data