Add demand rate lookups (#4)

openeihttp/__init__.py

@@ -109,6 +109,33 @@ def current_rate(self) -> float:
 
         return rate
 
+    @property
+    def current_demand_rate(self) -> float:
+        """Return the current rate."""
+        assert self._data is not None
+        weekend = False
+        now = datetime.datetime.today()
+        month = now.month - 1
+        hour = now.hour
+        if now.weekday() > 4:
+            weekend = True
+        table = "demandweekdayschedule"
+        if weekend:
+            table = "demandweekendschedule"
+
+        lookup_table = self._data[table]
+        rate_structure = lookup_table[month][hour]
+
+        rate = self._data["demandratestructure"][rate_structure][0]["rate"]
+
+        return rate
+
+    @property
+    def demand_unit(self) -> str:
+        """Return the demand rate unit."""
+        assert self._data is not None
+        return self._data["demandrateunit"]
+
     @property
     def rate_name(self) -> str:
         """Return the rate name."""

setup.py

@@ -5,7 +5,7 @@
 
 PROJECT_DIR = Path(__file__).parent.resolve()
 README_FILE = PROJECT_DIR / "README.md"
-VERSION = "0.1.3"
+VERSION = "0.1.4"
 
 
 setup(

tests/conftest.py

@@ -37,6 +37,15 @@ def mock_plandata(requests_mock):
     )
 
 
+@pytest.fixture(name="demand_plandata_mock")
+def mock_demand_plandata(requests_mock):
+    """Mock the status reply."""
+    requests_mock.get(
+        "https://api.openei.org/utility_rates?version=latest&format=json&api_key=fakeAPIKey&lat=1&lon=1&sector=Residential&detail=full&getpage=574613aa5457a3557e906f5b",
+        text=load_fixture("plan_demand_data.json"),
+    )
+
+
 @pytest.fixture(name="lookup_mock_404")
 def mock_lookup_404(requests_mock):
     """Mock the status reply."""