tyba-client 0.4.15__tar.gz → 0.4.16__tar.gz

{tyba_client-0.4.15 → tyba_client-0.4.16}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: tyba-client
-Version: 0.4.15
+Version: 0.4.16
 Summary: A Python API client for the Tyba Public API
 License: MIT
 Author: Tyler Nisonoff
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: dataclasses-json (>=0.6.4,<0.7.0)
 Requires-Dist: generation-models (>=0.10.6,<0.11.0)
 Requires-Dist: marshmallow (>=3.12.1,<4.0.0)

{tyba_client-0.4.15 → tyba_client-0.4.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tyba-client"
-version = "0.4.15"
+version = "0.4.16"
 description = "A Python API client for the Tyba Public API"
 authors = ["Tyler Nisonoff <tyler@tybaenergy.com>"]
 license = "MIT"

{tyba_client-0.4.15 → tyba_client-0.4.16}/tyba_client/operations.py

@@ -1,6 +1,7 @@
 import datetime
+import json
 from datetime import date
-from typing import Optional
+from typing import Optional, Literal
 
 
 class Operations(object):
@@ -12,6 +13,10 @@ class Operations(object):
         response.raise_for_status()
         return response.text
 
+
+    def post(self, route, json):
+        return self.client.post(f"operations/{route}", json=json)
+
     def performance_report(
             self,
             start_date: date,
@@ -82,3 +87,58 @@ class Operations(object):
         if org_id:
             params["org_id"] = org_id
         return self.get("internal_api/assets", params=params)
+
+    def set_asset_overrides(
+            self,
+            asset_names: list[str],
+            field: str,
+            aggregation: Literal["global", "single_day", "single_day_hourly", "12x24"],
+            values,
+            service: Optional[str] = None,
+            date: Optional[datetime.date] = None,
+    ):
+        assumption =  {
+          "field": field,
+         "data": {"aggregation":aggregation }
+        }
+        match aggregation:
+            case "12x24":
+                assumption["data"] = {"aggregation": "12x24"} | values
+            case "single_day_hourly":
+                assumption["data"] = {
+                    "aggregation": "single_day_hourly",
+                    "date": date,
+                    "values": values
+                }
+            case "single_day":
+                assumption["data"] = {
+                    "aggregation": "single_day_hourly",
+                    "values": values,
+                    "date": date
+                }
+            case "global":
+                assumption["data"] = {
+                    "aggregation": "global",
+                    "value": values
+                }
+        if service:
+            assumption["service"] = service
+        request_data = {"asset_names": asset_names, "assumption": assumption}
+
+        res = self.post("internal_api/assets/override/", json=request_data)
+        if res.status_code != 200:
+            return {
+                "status_code": res.status_code,
+                "reason": res.reason,
+                "message": res.text
+            }
+        else:
+            return json.loads(res.text)
+
+
+
+    def overrides_schema(
+            self,
+    ):
+
+        return json.loads(self.get("internal_api/overrides_schema"))