Fix shield to use cloud entities only

Author: Muriel2Horak

custom_components/oig_cloud/service_shield.py

@@ -14,6 +14,7 @@
 )
 from homeassistant.util.dt import now as dt_now
 
+from .const import DOMAIN
 from .shared.logging import setup_simple_telemetry
 
 _LOGGER = logging.getLogger(__name__)
@@ -1477,42 +1478,60 @@ def _extract_param_type(entity_id: str) -> str:
                 return "value"  # Fallback
 
         def find_entity(suffix: str) -> str | None:
-            _LOGGER.info(f"[FIND ENTITY] Hledám entitu se suffixem: {suffix}")
-            matching_entities = []
-            for entity in self.hass.states.async_all():
-                if entity.entity_id.endswith(suffix):
-                    matching_entities.append(entity.entity_id)
+            _LOGGER.info("[FIND ENTITY] Hledám cloud entitu se suffixem: %s", suffix)
 
-            if matching_entities:
-                # Preferuj klasické senzory před binary/ostatními
-                preferred_prefixes = ("sensor.", "number.", "binary_sensor.")
-                for prefix in preferred_prefixes:
-                    for entity_id in matching_entities:
-                        if entity_id.startswith(prefix):
-                            _LOGGER.info(
-                                "[FIND ENTITY] Vybrána preferovaná entita: %s",
-                                entity_id,
-                            )
-                            return entity_id
-                _LOGGER.info(
-                    f"[FIND ENTITY] Nalezeno {len(matching_entities)} entit: {matching_entities}"
-                )
-                return matching_entities[0]
-            else:
-                _LOGGER.warning(
-                    f"[FIND ENTITY] NENALEZENA žádná entita se suffixem: {suffix}"
-                )
-                # Debug: Vypíšeme všechny invertor entity
-                all_invertor = [
-                    e.entity_id
-                    for e in self.hass.states.async_all()
-                    if "invertor" in e.entity_id.lower()
-                ]
+            # Preferuj pouze cloud senzory: sensor.oig_<box_id>_<suffix>
+            box_id = None
+            for key in ("box_id", "inverter_sn"):
+                val = self.entry.options.get(key) or self.entry.data.get(key)
+                if isinstance(val, str) and val.isdigit():
+                    box_id = val
+                    break
+
+            if not box_id:
+                try:
+                    from .oig_cloud_sensor import resolve_box_id
+
+                    coordinator = None
+                    for entry_data in self.hass.data.get(DOMAIN, {}).values():
+                        if not isinstance(entry_data, dict):
+                            continue
+                        if entry_data.get("service_shield") != self:
+                            continue
+                        coordinator = entry_data.get("coordinator")
+                        break
+                    if coordinator:
+                        resolved = resolve_box_id(coordinator)
+                        if isinstance(resolved, str) and resolved.isdigit():
+                            box_id = resolved
+                except Exception:
+                    box_id = None
+
+            if not box_id:
                 _LOGGER.warning(
-                    f"[FIND ENTITY] Všechny invertor entity: {all_invertor}"
+                    "[FIND ENTITY] box_id nelze určit, cloud entitu pro suffix '%s' nelze vybrat",
+                    suffix,
                 )
                 return None
 
+            prefix = f"sensor.oig_{box_id}_"
+            matching_entities = [
+                entity.entity_id
+                for entity in self.hass.states.async_all()
+                if entity.entity_id.startswith(prefix)
+                and entity.entity_id.endswith(suffix)
+            ]
+
+            if matching_entities:
+                entity_id = matching_entities[0]
+                _LOGGER.info("[FIND ENTITY] Vybrána cloud entita: %s", entity_id)
+                return entity_id
+
+            _LOGGER.warning(
+                "[FIND ENTITY] NENALEZENA cloud entita %s*%s", prefix, suffix
+            )
+            return None
+
         # OPRAVA: Speciální handling pro set_formating_mode - nemá žádný senzor k ověření
         if service_name == "oig_cloud.set_formating_mode":
             # Vytvoříme fiktivní entitu pro 2minutové sledování

custom_components/oig_cloud/strings.json

@@ -200,6 +200,9 @@
           "distribution_fee_nt_kwh": "Poplatek za distribuci NT (CZK/kWh)",
           "tariff_vt_start_weekday": "VT začátek pracovní den (hodina)",
           "tariff_nt_start_weekday": "NT začátek pracovní den (hodina1,hodina2)",
+          "tariff_vt_start_weekend": "VT začátek víkend (hodina)",
+          "tariff_nt_start_weekend": "NT začátek víkend (hodina1,hodina2)",
+          "tariff_weekend_same_as_weekday": "Víkend stejné jako pracovní dny",
           "vat_rate": "DPH (%)",
           "go_back": "⬅️ Chci se vrátit zpět (zaškrtnout + kliknout Odeslat)"
         },
@@ -209,6 +212,9 @@
           "distribution_fee_nt_kwh": "Distribuční poplatek NT (nízký tarif). Např. 0.91 CZK/kWh",
           "tariff_vt_start_weekday": "Začátek VT v pracovní den. Formát: hodina (např. '6' = 06:00)",
           "tariff_nt_start_weekday": "Začátek NT v pracovní den. Formát: hodina1,hodina2 (např. '22,2' = 22:00 večer a 02:00 ráno)",
+          "tariff_vt_start_weekend": "Začátek VT o víkendu. Formát: hodina (nechte prázdné pro NT celý den)",
+          "tariff_nt_start_weekend": "Začátek NT o víkendu. Formát: hodina1,hodina2 (např. '0' = NT celý den)",
+          "tariff_weekend_same_as_weekday": "Použít stejné časy VT/NT i o víkendu",
           "vat_rate": "Sazba DPH v procentech. Standardně 21%",
           "go_back": "Zaškrtněte toto políčko a stiskněte tlačítko 'Odeslat' - vrátíte se o krok zpět"
         }
@@ -455,6 +461,9 @@
           "distribution_fee_nt_kwh": "Poplatek za distribuci NT (CZK/kWh)",
           "tariff_vt_start_weekday": "VT začátek pracovní den (hodina)",
           "tariff_nt_start_weekday": "NT začátek pracovní den (hodina1,hodina2)",
+          "tariff_vt_start_weekend": "VT začátek víkend (hodina)",
+          "tariff_nt_start_weekend": "NT začátek víkend (hodina1,hodina2)",
+          "tariff_weekend_same_as_weekday": "Víkend stejné jako pracovní dny",
           "vat_rate": "DPH (%)",
           "go_back": "⬅️ Chci se vrátit zpět (zaškrtnout + kliknout Odeslat)"
         },
@@ -464,6 +473,9 @@
           "distribution_fee_nt_kwh": "Distribuční poplatek NT (nízký tarif). Např. 0.91 CZK/kWh",
           "tariff_vt_start_weekday": "Začátek VT v pracovní den. Formát: hodina (např. '6' = 06:00)",
           "tariff_nt_start_weekday": "Začátek NT v pracovní den. Formát: hodina1,hodina2 (např. '22,2' = 22:00 večer a 02:00 ráno)",
+          "tariff_vt_start_weekend": "Začátek VT o víkendu. Formát: hodina (nechte prázdné pro NT celý den)",
+          "tariff_nt_start_weekend": "Začátek NT o víkendu. Formát: hodina1,hodina2 (např. '0' = NT celý den)",
+          "tariff_weekend_same_as_weekday": "Použít stejné časy VT/NT i o víkendu",
           "vat_rate": "Sazba DPH v procentech. Standardně 21%",
           "go_back": "Zaškrtněte toto políčko a stiskněte tlačítko 'Odeslat' - vrátíte se o krok zpět"
         }