From 889682f7714728f7e153740e30975aeab2efae09 Mon Sep 17 00:00:00 2001
From: Yegor Vialov <vyalov.egor@gmail.com>
Date: Fri, 1 Nov 2019 13:54:35 +0000
Subject: [PATCH] Resolves #491 Lovelace badges parse issue

---
 lib/home_assistant.class.dart | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/lib/home_assistant.class.dart b/lib/home_assistant.class.dart
index de97595..7f2a81e 100644
--- a/lib/home_assistant.class.dart
+++ b/lib/home_assistant.class.dart
@@ -180,9 +180,17 @@ class HomeAssistant {
 
         if (rawView['badges'] != null && rawView['badges'] is List) {
           rawView['badges'].forEach((entity) {
-            if (entities.isExist(entity)) {
-              Entity e = entities.get(entity);
-              view.badges.add(e);
+            if (entity is String) {
+              if (entities.isExist(entity)) {
+                Entity e = entities.get(entity);
+                view.badges.add(e);
+              }
+            } else {
+              String eId = '${entity['entity']}';
+              if (entities.isExist(eId)) {
+                Entity e = entities.get(eId);
+                view.badges.add(e);
+              }
             }
           });
         }