tyba-client 0.4.15__py3-none-any.whl → 0.4.17__py3-none-any.whl

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",
+                    "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"))

{tyba_client-0.4.15.dist-info → tyba_client-0.4.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: tyba-client
-Version: 0.4.15
+Version: 0.4.17
 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.dist-info → tyba_client-0.4.17.dist-info}/RECORD

@@ -1,13 +1,12 @@
-LICENSE,sha256=LbMfEdjEK-IRzvCfdEBhn9UCANze0Rc7hWrQTEj_xvU,1079
 tyba_client/__init__.py,sha256=42STGor_9nKYXumfeV5tiyD_M8VdcddX7CEexmibPBk,22
 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=WkAr_e-15C0wIzP9l_a0uRJGqhObUIEG_iDWo_17TqU,2316
+tyba_client/operations.py,sha256=ZVJ90ow0JFFdS6eXBjonyNYrQx83eQfEXt02Eo657ik,4168
 tyba_client/solar_resource.py,sha256=0Jte2cZpE-USHeW-qac4AN1mI2tmrC-VSahnbs4v3Bs,3683
 tyba_client/utils.py,sha256=n4tUBGlQIwxbLqJcQRiAIbIJA0DaLSjbAxakhukqaeE,876
-tyba_client-0.4.15.dist-info/LICENSE,sha256=LbMfEdjEK-IRzvCfdEBhn9UCANze0Rc7hWrQTEj_xvU,1079
-tyba_client-0.4.15.dist-info/METADATA,sha256=QM0n_-WHWXcLSkjxzVq9CKoKOCbI21RTMqhG36DD_kk,1273
-tyba_client-0.4.15.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-tyba_client-0.4.15.dist-info/RECORD,,
+tyba_client-0.4.17.dist-info/LICENSE,sha256=LbMfEdjEK-IRzvCfdEBhn9UCANze0Rc7hWrQTEj_xvU,1079
+tyba_client-0.4.17.dist-info/METADATA,sha256=3RtGOZcTcFG4AOCBz7514b1o8AS1tX3zV_-qHExXPJU,1324
+tyba_client-0.4.17.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+tyba_client-0.4.17.dist-info/RECORD,,

{tyba_client-0.4.15.dist-info → tyba_client-0.4.17.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 2.1.3
 Root-Is-Purelib: true
 Tag: py3-none-any

tyba_client-0.4.15.dist-info/LICENSE

@@ -1,21 +0,0 @@
-The MIT License (MIT)
-
-Copyright (c) 2021 Tyba Energy.
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.