Change state filter on select to match sensors with route_id instead.

jonkristian · Mar 25, 2023 · aeba8c7 · aeba8c7
1 parent 07c65ca
commit aeba8c7
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/dist/entur-card.js b/dist/entur-card.js
diff --git a/src/entur-card-editor.ts b/src/entur-card-editor.ts
@@ -89,10 +89,15 @@ export class EnturCardEditor
       `;
     }
 
-    // Find all starting with 'sensor.transport'.
-    const entities = Object.keys(this.hass!.states).filter((k) =>
-      k.startsWith("sensor.transport")
-    );
+    // Filter states to only include sensors, and only those with an attribute containing a "route_id".
+    const sensorsWithRouteId = Object.values(this.hass!.states)
+      .filter((entity) => entity.entity_id.startsWith("sensor."))
+      .filter((sensor) =>
+        Object.keys(sensor.attributes).some((key) =>
+          key.toLowerCase().includes("route_id")
+        )
+      )
+      .map((sensor) => sensor.entity_id);
 
     return html`
       <div class="card-config">
@@ -170,7 +175,7 @@ export class EnturCardEditor
           fixedMenuPosition
           naturalMenuWidth
         >
-          ${entities.map(
+          ${sensorsWithRouteId.map(
             (entity) => html`
               <mwc-list-item .value=${entity}> ${entity} </mwc-list-item>
             `