Enode API Changes

pv_site_api/_db_helpers.py

@@ -14,7 +14,7 @@
 import sqlalchemy as sa
 import structlog
 from pvsite_datamodel.read.generation import get_pv_generation_by_sites
-from pvsite_datamodel.sqlmodels import ForecastSQL, ForecastValueSQL, SiteSQL
+from pvsite_datamodel.sqlmodels import ForecastSQL, ForecastValueSQL, InverterSQL, SiteSQL
 from sqlalchemy.orm import Session, aliased
 
 from .pydantic_models import (
@@ -60,6 +60,12 @@ def _get_forecasts_for_horizon(
     return list(session.execute(stmt))
 
 
+def _get_inverters_by_site(session: Session, site_uuid: str) -> list[Row]:
+    query = session.query(InverterSQL).filter(InverterSQL.site_uuid == site_uuid)
+
+    return query.all()
+
+
 def _get_latest_forecast_by_sites(
     session: Session, site_uuids: list[str], start_utc: Optional[dt.datetime] = None
 ) -> list[Row]:

pv_site_api/enode_auth.py

@@ -0,0 +1,39 @@
+from typing import Optional
+
+import httpx
+
+
+class EnodeAuth(httpx.Auth):
+    def __init__(
+        self, client_id: str, client_secret: str, token_url: str, access_token: Optional[str] = None
+    ):
+        self._client_id = client_id
+        self._client_secret = client_secret
+        self._token_url = token_url
+        self._access_token = access_token
+
+    def auth_flow(self, request: httpx.Request):
+        # Add the Authorization header to the request using the current access token
+        request.headers["Authorization"] = f"Bearer {self._access_token}"
+        response = yield request
+
+        if response.status_code == 401:
+            # The access token is no longer valid, refresh it
+            token_response = yield self._build_refresh_request()
+            self._update_access_token(token_response)
+            # Update the request's Authorization header with the new access token
+            request.headers["Authorization"] = f"Bearer {self._access_token}"
+            # Resend the request with the new access token
+            response = yield request
+        return response
+
+    def _build_refresh_request(self):
+        basic_auth = httpx.BasicAuth(self._client_id, self._client_secret)
+
+        data = {"grant_type": "client_credentials"}
+        request = next(basic_auth.auth_flow(httpx.Request("POST", self._token_url, data=data)))
+        return request
+
+    def _update_access_token(self, response):
+        response.read()
+        self._access_token = response.json()["access_token"]

pv_site_api/fake.py

@@ -6,6 +6,11 @@
 
 from .pydantic_models import (
     Forecast,
+    InverterInformation,
+    InverterLocation,
+    InverterProductionState,
+    Inverters,
+    InverterValues,
     MultiplePVActual,
     PVActualValue,
     PVSiteAPIStatus,
@@ -19,6 +24,35 @@
 fake_client_uuid = "c97f68cd-50e0-49bb-a850-108d4a9f7b7e"
 
 
+def make_fake_inverters() -> Inverters:
+    """Make fake inverters"""
+    inverter = InverterValues(
+        id="string",
+        vendor="EMA",
+        chargingLocationId="8d90101b-3f2f-462a-bbb4-1ed320d33bbe",
+        lastSeen="2020-04-07T17:04:26Z",
+        isReachable=True,
+        productionState=InverterProductionState(
+            productionRate=0,
+            isProducing=True,
+            totalLifetimeProduction=100152.56,
+            lastUpdated="2020-04-07T17:04:26Z",
+        ),
+        information=InverterInformation(
+            id="string",
+            brand="EMA",
+            model="Sunny Boy",
+            siteName="Sunny Plant",
+            installationDate="2020-04-07T17:04:26Z",
+        ),
+        location=InverterLocation(latitude=10.7197486, longitude=59.9173985),
+    )
+    inverters_list = Inverters(
+        inverters=[inverter],
+    )
+    return inverters_list
+
+
 def make_fake_site() -> PVSites:
     """Make a fake site"""
     pv_site = PVSiteMetadata(
@@ -86,3 +120,8 @@ def make_fake_status() -> PVSiteAPIStatus:
         message="The API is up and running",
     )
     return pv_api_status
+
+
+def make_fake_enode_link_url() -> str:
+    """Make fake Enode link URL"""
+    return "https://enode.com"