Moved attributes updates

changed to better versioning
jampez77 · Sep 21, 2023 · 4ce0073 · 4ce0073
1 parent 4773e79
commit 4ce0073
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/custom_components/dvla/manifest.json b/custom_components/dvla/manifest.json
@@ -10,6 +10,6 @@
   "issue_tracker": "https://github.com/jampez77/DVLA-Vehicle-Enquiry-Service/issues",
   "requirements": [],
   "ssdp": [],
-  "version": "1.0.0",
+  "version": "2023.9.0",
   "zeroconf": []
 }
diff --git a/custom_components/dvla/sensor.py b/custom_components/dvla/sensor.py
@@ -122,6 +122,8 @@ def icon(self) -> str:
 
     @property
     def extra_state_attributes(self) -> dict[str, Any]:
+        for key in self.coordinator.data:
+            self.attrs[key] = self.coordinator.data[key]
         return self.attrs
 
     async def async_update(self) -> None:
@@ -130,9 +132,6 @@ async def async_update(self) -> None:
         Only used by the generic entity update service.
         """
         try:
-            for key in self.coordinator.data:
-                self.attrs[key] = self.coordinator.data[key]
-
             self._state = (
                 str(self.coordinator.data["yearOfManufacture"])
                 + " "