tyba-client 0.4.2__tar.gz → 0.4.3__tar.gz

{tyba_client-0.4.2 → tyba_client-0.4.3}/PKG-INFO

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

{tyba_client-0.4.2 → tyba_client-0.4.3}/pyproject.toml

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

{tyba_client-0.4.2 → tyba_client-0.4.3}/tyba_client/client.py

@@ -1,4 +1,5 @@
 import pandas as pd
+import typing as t
 from requests import Response
 
 from tyba_client.models import GenerationModel, PVStorageModel, StandaloneStorageModel
@@ -16,11 +17,11 @@ logger = get_logger()
 class Ancillary(object):
     """_"""
 
-    def __init__(self, prices):
-        self.prices = prices
+    def __init__(self, services):
+        self.services = services
 
     def get(self, route, params=None):
-        return self.prices.get(f"ancillary/{route}", params=params)
+        return self.services.get(f"ancillary/{route}", params=params)
 
     def get_pricing_regions(self, *, iso, service, market):
         """_"""
@@ -44,11 +45,15 @@ class Ancillary(object):
 class LMP(object):
     """_"""
 
-    def __init__(self, prices):
-        self.prices = prices
+    def __init__(self, services):
+        self.services = services
+        self._route_base = "lmp"
 
     def get(self, route, params=None):
-        return self.prices.get(f"lmp/{route}", params=params)
+        return self.services.get(f"{self._route_base}/{route}", params=params)
+
+    def post(self, route, json):
+        return self.services.post(f"{self._route_base}/{route}", json=json)
 
     def get_all_nodes(self, *, iso):
         """_"""
@@ -66,6 +71,12 @@ class LMP(object):
             },
         )
 
+    def search_nodes(self, location: t.Optional[str] = None, node_name_filter: t.Optional[str] = None, iso_override: t.Optional[str] = None):
+        return self.get(route="search-nodes",
+                        params={"location": location,
+                                "node_name_filter": node_name_filter,
+                                "iso_override": iso_override})
+
 
 class Services(object):
     """_"""
@@ -74,9 +85,13 @@ class Services(object):
         self.client = client
         self.ancillary = Ancillary(self)
         self.lmp = LMP(self)
+        self._route_base = "services"
 
     def get(self, route, params=None):
-        return self.client.get(f"services/{route}", params=params)
+        return self.client.get(f"{self._route_base}/{route}", params=params)
+
+    def post(self, route, json):
+        return self.client.post(f"{self._route_base}/{route}", json=json)
 
     def get_all_isos(self):
         """_"""