From 78218e12bdbd7fac32c35d897e846abc9330997f Mon Sep 17 00:00:00 2001
From: Eric de Groot <eric@ericdegroot.com>
Date: Wed, 3 Apr 2024 20:54:25 -0700
Subject: [PATCH] Updated powerwall site grid_status property to using site
 data instead of summary

---
 AUTHORS.md                      | 1 +
 teslajsonpy/energy.py           | 2 +-
 tests/unit_tests/test_energy.py | 4 ++++
 3 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/AUTHORS.md b/AUTHORS.md
index 0262d14b..ec92ce1d 100644
--- a/AUTHORS.md
+++ b/AUTHORS.md
@@ -23,3 +23,4 @@
 -   craigrouse [Github](https://github.com/craigrouse)
 -   thierryVT [Github](https://github.com/thierryvt)
 -   llamafilm [Github](https://github.com/llamafilm)
+-   ericdegroot [Github](https://github.com/ericdegroot)
diff --git a/teslajsonpy/energy.py b/teslajsonpy/energy.py
index 7b4dfc39..46c7cdf7 100644
--- a/teslajsonpy/energy.py
+++ b/teslajsonpy/energy.py
@@ -159,7 +159,7 @@ def grid_power(self) -> float:
     @property
     def grid_status(self) -> str:
         """Return grid status."""
-        return self._site_summary.get("grid_status")
+        return self._site_data.get("grid_status")
 
     @property
     def load_power(self) -> float:
diff --git a/tests/unit_tests/test_energy.py b/tests/unit_tests/test_energy.py
index d2bec54d..3deca167 100644
--- a/tests/unit_tests/test_energy.py
+++ b/tests/unit_tests/test_energy.py
@@ -92,6 +92,10 @@ async def test_powerwall_site(monkeypatch):
         _solar_powerwall_site.grid_power
         == SITE_DATA["grid_power"]
     )
+    assert (
+        _solar_powerwall_site.grid_status
+        == SITE_DATA["grid_status"]
+    )
     assert (
         _solar_powerwall_site.load_power
         == SITE_DATA["load_power"]