From e45e39a3da67a2793733b35e129af5adb5b30b2e Mon Sep 17 00:00:00 2001
From: Adrian Rumpold <a.rumpold@gmail.com>
Date: Fri, 18 Jun 2021 18:20:04 +0200
Subject: [PATCH] fix: Return entire response as entity state

---
 .../home_assistant_covid19_augsburg/sensor.py | 22 +++++++++----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/custom_components/home_assistant_covid19_augsburg/sensor.py b/custom_components/home_assistant_covid19_augsburg/sensor.py
index 7b536d0..36f105c 100644
--- a/custom_components/home_assistant_covid19_augsburg/sensor.py
+++ b/custom_components/home_assistant_covid19_augsburg/sensor.py
@@ -41,18 +41,18 @@ class CoronaAugsburgSensor(Entity):
 
     @property
     def state(self):
-        return self.coordinator.data.total_cases
+        return self.coordinator.data
 
-    @property
-    def device_state_attributes(self):
-        return {
-            "date": self.coordinator.data.date,
-            "total_cases": self.coordinator.data.total_cases,
-            "num_dead": self.coordinator.data.num_dead,
-            "num_recovered": self.coordinator.data.num_recovered,
-            "num_infected": self.coordinator.data.num_infected,
-            "incidence": self.coordinator.data.incidence,
-        }
+    # @property
+    # def device_state_attributes(self):
+    # return {
+    #     "date": self.coordinator.data.date,
+    #     "total_cases": self.coordinator.data.total_cases,
+    #     "num_dead": self.coordinator.data.num_dead,
+    #     "num_recovered": self.coordinator.data.num_recovered,
+    #     "num_infected": self.coordinator.data.num_infected,
+    #     "incidence": self.coordinator.data.incidence,
+    # }
 
     async def async_added_to_hass(self):
         """When entity is added to hass."""