tyba-client 0.4.16__py3-none-any.whl → 0.4.18__py3-none-any.whl

tyba_client/operations.py

@@ -3,6 +3,8 @@ import json
 from datetime import date
 from typing import Optional, Literal
 
+from tyba_client.models import ValidationError
+
 
 class Operations(object):
     def __init__(self, client):
@@ -21,59 +23,91 @@ class Operations(object):
             self,
             start_date: date,
             end_date: date,
-            asset_name: str,
+            asset_name: str | None,
+            display_name: str | None = None
     ):
-        return self.get(
-            "internal_api/performance_report",
-            params={
-                "start_date": start_date,
-                "end_date": end_date,
-                "asset_name": asset_name,
-            },
-        )
+        if not (asset_name or display_name):
+            raise ValidationError("Must provide either 'asset_name' or 'display_name'.")
+
+        params = {
+            "start_date": start_date,
+            "end_date": end_date
+        }
+
+        if asset_name:
+            params["asset_name"] = asset_name
+        else:
+            params["asset_display_name"] = display_name
+
+        return self.get("internal_api/performance_report", params=params)
 
     def da_snapshot(
             self,
             start_date: date,
             end_date: date,
-            asset_name: str,
+            asset_name: str | None,
+            display_name: str | None = None
     ):
+        if not (asset_name or display_name):
+            raise ValidationError("Must provide either 'asset_name' or 'display_name'.")
+        params= {
+            "start_date": start_date,
+            "end_date": end_date,
+        }
+        if asset_name:
+            params["asset_name"] = asset_name
+        else:
+            params["asset_display_name"] = display_name
+
+
         return self.get(
             "internal_api/da_snapshot",
-            params={
-                "start_date": start_date,
-                "end_date": end_date,
-                "asset_name": asset_name,
-            },
+            params=params,
         )
 
     def telemetry(
             self,
             start_date: date,
             end_date: date,
-            asset_name: str,
+            asset_name: str | None,
             interval_mins: int,
             metrics: list[str],
             solar_asset_telemetry: bool = False,
+            display_name: str | None = None
     ):
-        return self.get(
-            "internal_api/telemetry",
-            params={
+        if not (asset_name or display_name):
+            raise ValidationError("Must provide either 'asset_name' or 'display_name'.")
+        params= {
                 "start_date": start_date,
                 "end_date": end_date,
-                "asset_name": asset_name,
                 "interval_mins": interval_mins,
                 "metrics": metrics,
                 "solar_asset_telemetry": solar_asset_telemetry,
-            },
-        )
+            }
+        if asset_name:
+            params["asset_name"] = asset_name
+        else:
+            params["asset_display_name"] = display_name
+
+        return self.get(
+                "internal_api/telemetry",
+                params=params,
+            )
 
     def asset_details(
             self,
-            asset_name: str,
+            asset_name: str | None,
             date: Optional[datetime.date] = None,
+            display_name: str | None = None
     ):
-        params = {"asset_name": asset_name}
+        if not (asset_name or display_name):
+            raise ValidationError("Must provide either 'asset_name' or 'display_name'.")
+        params= {}
+        if asset_name:
+            params["asset_name"] = asset_name
+        else:
+            params["asset_display_name"] = display_name
+
         if date is not None:
             params["date"] = date
         return self.get("internal_api/asset_details", params=params)
@@ -112,7 +146,7 @@ class Operations(object):
                 }
             case "single_day":
                 assumption["data"] = {
-                    "aggregation": "single_day_hourly",
+                    "aggregation": "single_day",
                     "values": values,
                     "date": date
                 }

{tyba_client-0.4.16.dist-info → tyba_client-0.4.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: tyba-client
-Version: 0.4.16
+Version: 0.4.18
 Summary: A Python API client for the Tyba Public API
 License: MIT
 Author: Tyler Nisonoff

{tyba_client-0.4.16.dist-info → tyba_client-0.4.18.dist-info}/RECORD

@@ -3,10 +3,10 @@ tyba_client/client.py,sha256=RAvjZkjWDDeY4eJXD2oELF4E1DOa02AqAcvWftsxISY,23295
 tyba_client/forecast.py,sha256=P9GuKPrTCQpdxatSPCck5dezfMIe7otCjOiylyPVh-s,8383
 tyba_client/io.py,sha256=0FVRtUjSDzyWYBvIMSJH7bCclCRqD2Y-pbcsJ-Ufyo0,6226
 tyba_client/models.py,sha256=NOo39qStMkWBhfLPkUu37MaKMe3E2yFc5KAzUnFy96M,17569
-tyba_client/operations.py,sha256=cOAyeV2vz_6FEY4jlOPsMNLmJ8TvIaZgfyYJjMZEejg,4175
+tyba_client/operations.py,sha256=j03o3hMffnUttRzxqiGx3qUonCplbI6UENK2W-dz1vY,5390
 tyba_client/solar_resource.py,sha256=0Jte2cZpE-USHeW-qac4AN1mI2tmrC-VSahnbs4v3Bs,3683
 tyba_client/utils.py,sha256=n4tUBGlQIwxbLqJcQRiAIbIJA0DaLSjbAxakhukqaeE,876
-tyba_client-0.4.16.dist-info/LICENSE,sha256=LbMfEdjEK-IRzvCfdEBhn9UCANze0Rc7hWrQTEj_xvU,1079
-tyba_client-0.4.16.dist-info/METADATA,sha256=v4flCDkfvS6JTnCAEEsNmVkywi0FFNJ5MImlCPtDgyE,1324
-tyba_client-0.4.16.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-tyba_client-0.4.16.dist-info/RECORD,,
+tyba_client-0.4.18.dist-info/LICENSE,sha256=LbMfEdjEK-IRzvCfdEBhn9UCANze0Rc7hWrQTEj_xvU,1079
+tyba_client-0.4.18.dist-info/METADATA,sha256=GW_akUmZaFqFqM6bajA_0GXOFf2LFcPzho7m1It-A60,1324
+tyba_client-0.4.18.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+tyba_client-0.4.18.dist-info/RECORD,,