tyba-client 0.4.5__tar.gz → 0.4.7__tar.gz

{tyba_client-0.4.5 → tyba_client-0.4.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tyba-client
-Version: 0.4.5
+Version: 0.4.7
 Summary: A Python API client for the Tyba Public API
 License: MIT
 Author: Tyler Nisonoff
@@ -16,7 +16,8 @@ Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: dataclasses-json (>=0.6.4,<0.7.0)
 Requires-Dist: generation-models (>=0,<1)
 Requires-Dist: marshmallow (>=3.12.1,<4.0.0)
-Requires-Dist: pandas (>=1.3.2,<2.0.0)
+Requires-Dist: pandas (>=1.3.2,<2.0.0) ; python_version < "3.9"
+Requires-Dist: pandas (>=2.0.0,<3.0.0) ; python_version >= "3.9"
 Requires-Dist: requests (>=2.25.1,<3.0.0)
 Requires-Dist: structlog (>=24.1.0,<25.0.0)
 Requires-Dist: tyba-financial-model (>=0,<1)

{tyba_client-0.4.5 → tyba_client-0.4.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "tyba-client"
-version = "0.4.5"
+version = "0.4.7"
 description = "A Python API client for the Tyba Public API"
 authors = ["Tyler Nisonoff <tyler@tybaenergy.com>"]
 license = "MIT"
@@ -14,7 +14,10 @@ python = "^3.8"
 requests = "^2.25.1"
 dataclasses-json = "^0.6.4"
 marshmallow = "^3.12.1"
-pandas = "^1.3.2"
+pandas = [
+    { version = "^1.3.2", python = "<3.9" },
+    { version = "^2.0.0", python = ">=3.9" }
+]
 tyba-financial-model = "^0"
 structlog = "^24.1.0"
 generation-models = "^0"

{tyba_client-0.4.5 → tyba_client-0.4.7}/tyba_client/client.py

@@ -11,6 +11,8 @@ import time
 from structlog import get_logger
 from typing import Callable
 
+from tyba_client.operations import Operations
+
 logger = get_logger()
 
 
@@ -114,6 +116,7 @@ class Client(object):
         self.host = host
         self.services = Services(self)
         self.forecast = Forecast(self)
+        self.operations = Operations(self)
         self.request_args = {} if request_args is None else request_args
 
     def _auth_header(self):

tyba_client-0.4.7/tyba_client/operations.py

@@ -0,0 +1,60 @@
+from datetime import date
+
+
+class Operations(object):
+    def __init__(self, client):
+        self.client = client
+
+    def get(self, route, params=None):
+        response = self.client.get(f"operations/{route}", params=params)
+        response.raise_for_status()
+        return response.text
+
+    def performance_report(
+            self,
+            start_date: date,
+            end_date: date,
+            asset_name: str,
+    ):
+        return self.get(
+            "internal_api/performance_report",
+            params={
+                "start_date": start_date,
+                "end_date": end_date,
+                "asset_name": asset_name,
+            },
+        )
+
+    def da_snapshot(
+            self,
+            start_date: date,
+            end_date: date,
+            asset_name: str,
+    ):
+        return self.get(
+            "internal_api/da_snapshot",
+            params={
+                "start_date": start_date,
+                "end_date": end_date,
+                "asset_name": asset_name,
+            },
+        )
+
+    def telemetry(
+            self,
+            start_date: date,
+            end_date: date,
+            asset_name: str,
+            interval_mins: int,
+            metrics: list[str]
+    ):
+        return self.get(
+            "internal_api/telemetry",
+            params={
+                "start_date": start_date,
+                "end_date": end_date,
+                "asset_name": asset_name,
+                "interval_mins": interval_mins,
+                "metrics": metrics,
+            },
+        )