Switch rateacuity to state api

Author: MeadBarrel

Justfile

@@ -83,7 +83,7 @@ clean:
 
 
 rateacuity:
-    uv run python -m tariff_fetch.rateacuity.electric
+    uv run python -m tariff_fetch.rateacuity.state
 
 openei:
     uv run python -m tariff_fetch.openei.utility_rates "Consolidated Edison Co-NY Inc"

tariff_fetch/rateacuity/__init__.py

@@ -1,4 +1,2 @@
-from .base import create_context, login
-from .electric import get_electric_tariffs
-
-__all__ = ["create_context", "get_electric_tariffs", "login"]
+from .base import create_context as create_context
+from .state import LoginState as LoginState

tariff_fetch/rateacuity/base.py

@@ -23,7 +23,13 @@ class ScrapingContext(NamedTuple):
 @contextmanager
 def create_context() -> Generator[ScrapingContext]:
     with TemporaryDirectory() as temp_dir:
-        yield ScrapingContext(create_driver_(temp_dir), temp_dir)
+        driver = create_driver_(temp_dir)
+        driver.set_window_size(1920, 1080)
+        try:
+            yield ScrapingContext(driver, temp_dir)
+        except Exception as e:
+            driver.save_screenshot("selenium_error.png")
+            raise e from None
 
 
 def create_driver_(download_path: str) -> webdriver.Chrome: