switch2 0.1.1__tar.gz → 0.1.3__tar.gz

{switch2-0.1.1 → switch2-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: switch2
-Version: 0.1.1
+Version: 0.1.3
 Summary: Async client library for the Switch2 energy portal
 Author-email: Jelmer Vernooij <jelmer@jelmer.uk>
 License-Expression: Apache-2.0
@@ -38,16 +38,17 @@ import asyncio
 from switch2 import Switch2ApiClient
 
 async def main():
-    client = Switch2ApiClient("you@example.com", "your-password")
-    try:
+    async with Switch2ApiClient("you@example.com", "your-password") as client:
         data = await client.fetch_data()
         print(f"Customer: {data.customer.name}")
+        if data.account_balance is not None:
+            print(f"Balance: £{data.account_balance.balance}")
         for reading in data.readings:
             print(f"  {reading.date}: {reading.amount} {reading.unit}")
         for bill in data.bills:
             print(f"  {bill.date}: £{bill.amount}")
-    finally:
-        await client.close()
+            detail = await client.fetch_bill_detail(bill)
+            print(f"    invoice {detail.invoice_number}, total £{detail.total}")
 
 asyncio.run(main())
 ```

{switch2-0.1.1 → switch2-0.1.3}/README.md

@@ -15,16 +15,17 @@ import asyncio
 from switch2 import Switch2ApiClient
 
 async def main():
-    client = Switch2ApiClient("you@example.com", "your-password")
-    try:
+    async with Switch2ApiClient("you@example.com", "your-password") as client:
         data = await client.fetch_data()
         print(f"Customer: {data.customer.name}")
+        if data.account_balance is not None:
+            print(f"Balance: £{data.account_balance.balance}")
         for reading in data.readings:
             print(f"  {reading.date}: {reading.amount} {reading.unit}")
         for bill in data.bills:
             print(f"  {bill.date}: £{bill.amount}")
-    finally:
-        await client.close()
+            detail = await client.fetch_bill_detail(bill)
+            print(f"    invoice {detail.invoice_number}, total £{detail.total}")
 
 asyncio.run(main())
 ```

{switch2-0.1.1 → switch2-0.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "switch2"
-version = "0.1.1"
+version = "0.1.3"
 description = "Async client library for the Switch2 energy portal"
 readme = "README.md"
 license = "Apache-2.0"
@@ -33,6 +33,9 @@ Homepage = "https://github.com/jelmer/python-switch2"
 Repository = "https://github.com/jelmer/python-switch2"
 Issues = "https://github.com/jelmer/python-switch2/issues"
 
+[tool.setuptools.package-data]
+switch2 = ["py.typed"]
+
 [tool.mypy]
 packages = ["switch2"]
 strict = true

{switch2-0.1.1 → switch2-0.1.3}/switch2/__init__.py

@@ -1,6 +1,7 @@
 """Standalone client library for the Switch2 energy portal."""
 
 from .api import (
+    AccountBalance,
     Bill,
     BillCharge,
     BillDetail,
@@ -13,6 +14,7 @@ from .api import (
 )
 
 __all__ = [
+    "AccountBalance",
     "Bill",
     "BillCharge",
     "BillDetail",

{switch2-0.1.1 → switch2-0.1.3}/switch2/api.py

@@ -3,8 +3,10 @@
 from __future__ import annotations
 
 import logging
+from collections.abc import Callable
 from dataclasses import dataclass
 from datetime import datetime
+from types import TracebackType
 
 import aiohttp
 from bs4 import BeautifulSoup, Tag
@@ -16,6 +18,8 @@ LOGIN_URL = f"{BASE_URL}/Login"
 METER_HISTORY_URL = f"{BASE_URL}/MeterReadings/History"
 BILL_HISTORY_URL = f"{BASE_URL}/Credit/BillHistory"
 
+DEFAULT_TIMEOUT = aiohttp.ClientTimeout(total=60)
+
 
 class Switch2AuthError(Exception):
     """Raised when authentication fails."""
@@ -81,6 +85,14 @@ class BillDetail:
     download_url: str
 
 
+@dataclass
+class AccountBalance:
+    """Current account balance from the Switch2 portal."""
+
+    balance: float
+    last_updated: datetime
+
+
 @dataclass
 class Switch2Data:
     """All data fetched from the Switch2 portal."""
@@ -89,6 +101,7 @@ class Switch2Data:
     readings: list[MeterReading]
     registers: dict[str, str]  # register_id -> register_name
     bills: list[Bill]
+    account_balance: AccountBalance | None
 
 
 def _get_attr(tag: Tag, attr: str, default: str = "") -> str:
@@ -109,7 +122,7 @@ class Switch2ApiClient:
 
     async def _ensure_session(self) -> aiohttp.ClientSession:
         if self._session is None or self._session.closed:
-            self._session = aiohttp.ClientSession()
+            self._session = aiohttp.ClientSession(timeout=DEFAULT_TIMEOUT)
         return self._session
 
     async def close(self) -> None:
@@ -117,8 +130,19 @@ class Switch2ApiClient:
         if self._session and not self._session.closed:
             await self._session.close()
 
-    async def authenticate(self) -> CustomerInfo:
-        """Log in to the Switch2 portal and return customer info."""
+    async def __aenter__(self) -> "Switch2ApiClient":
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc: BaseException | None,
+        tb: TracebackType | None,
+    ) -> None:
+        await self.close()
+
+    async def authenticate(self) -> tuple[CustomerInfo, BeautifulSoup]:
+        """Log in to the Switch2 portal and return customer info and dashboard soup."""
         session = await self._ensure_session()
 
         try:
@@ -156,7 +180,7 @@ class Switch2ApiClient:
             if not customer.name:
                 raise Switch2AuthError("Login failed: no customer info returned")
 
-            return customer
+            return customer, soup
 
         except aiohttp.ClientError as err:
             raise Switch2ConnectionError(
@@ -186,7 +210,8 @@ class Switch2ApiClient:
 
     async def fetch_data(self) -> Switch2Data:
         """Authenticate and fetch all meter data."""
-        customer = await self.authenticate()
+        customer, dashboard_soup = await self.authenticate()
+        account_balance = _parse_account_balance(dashboard_soup)
         session = await self._ensure_session()
 
         try:
@@ -246,6 +271,7 @@ class Switch2ApiClient:
                 readings=readings,
                 registers=registers,
                 bills=bills,
+                account_balance=account_balance,
             )
 
         except aiohttp.ClientError as err:
@@ -347,6 +373,47 @@ def _parse_currency(text: str) -> float:
     return -value if negative else value
 
 
+def _parse_account_balance(soup: BeautifulSoup) -> AccountBalance | None:
+    """Extract the current account balance from the dashboard page."""
+    amount_el = soup.select_one(".dashboard-credit-amount-desktop")
+    if not amount_el:
+        return None
+    balance = _parse_currency(amount_el.text)
+    updated_el = soup.select_one(".dashboard-credit-lastUpdated")
+    if not updated_el:
+        return None
+    updated_text = updated_el.text.strip()
+    # Text is like "Last updated 27/02/2026 10:13"
+    if updated_text.lower().startswith("last updated"):
+        updated_text = updated_text[len("last updated") :].strip()
+    last_updated = datetime.strptime(updated_text, "%d/%m/%Y %H:%M")
+    return AccountBalance(balance=balance, last_updated=last_updated)
+
+
+def _has_class(tag: Tag, class_name: str) -> bool:
+    """Return True if the tag has the given CSS class."""
+    classes = tag.get("class")
+    if classes is None:
+        return False
+    if isinstance(classes, str):
+        return class_name in classes.split()
+    return class_name in classes
+
+
+def _parse_charge_row(row: Tag) -> BillCharge | None:
+    """Parse a single .bill-table-row into a BillCharge, or None if incomplete."""
+    desc_el = row.select_one(".bill-table-row-item-left")
+    units_el = row.select_one(".bill-table-row-item")
+    charge_el = row.select_one(".bill-table-row-item-right")
+    if not desc_el or not charge_el:
+        return None
+    return BillCharge(
+        description=desc_el.text.strip(),
+        units=units_el.text.strip() if units_el else "",
+        charge=_parse_currency(charge_el.text),
+    )
+
+
 def _parse_bill_charges(
     soup: BeautifulSoup, container_selector: str
 ) -> list[BillCharge]:
@@ -354,20 +421,51 @@ def _parse_bill_charges(
     charges: list[BillCharge] = []
     rows = soup.select(f"{container_selector} .bill-table-row-desktop .bill-table-row")
     for row in rows:
-        desc_el = row.select_one(".bill-table-row-item-left")
-        units_el = row.select_one(".bill-table-row-item")
-        charge_el = row.select_one(".bill-table-row-item-right")
-        if desc_el and charge_el:
-            charges.append(
-                BillCharge(
-                    description=desc_el.text.strip(),
-                    units=units_el.text.strip() if units_el else "",
-                    charge=_parse_currency(charge_el.text),
-                )
-            )
+        charge = _parse_charge_row(row)
+        if charge is not None:
+            charges.append(charge)
     return charges
 
 
+def _parse_keyed_currency_rows(
+    container: Tag | None,
+    fields: dict[str, Callable[[str], bool]],
+) -> dict[str, float]:
+    """Parse .bill-total-table-row rows, matching labels to keys via predicates.
+
+    Returns a mapping for each predicate that matched a row; missing keys are
+    logged at warning level so silent zeros don't hide upstream HTML changes.
+    """
+    found: dict[str, float] = {}
+    if container is None:
+        for key in fields:
+            _LOGGER.warning("Missing expected bill-total field: %s", key)
+        return found
+    for row in container.select(".bill-total-table-row"):
+        label_el = row.select_one(".bill-total-table-row-item-left")
+        value_el = row.select_one(".bill-total-table-row-item-right")
+        if not label_el or not value_el:
+            continue
+        label = label_el.text.strip()
+        for key, matches in fields.items():
+            if key not in found and matches(label):
+                found[key] = _parse_currency(value_el.text)
+                break
+    for key in fields:
+        if key not in found:
+            _LOGGER.warning("Missing expected bill-total field: %s", key)
+    return found
+
+
+def _parse_currency_header(soup: BeautifulSoup, selector: str, name: str) -> float:
+    """Parse a single currency value from a header selector, defaulting to 0.0."""
+    el = soup.select_one(selector)
+    if el is None:
+        _LOGGER.warning("Missing expected %s header", name)
+        return 0.0
+    return _parse_currency(el.text)
+
+
 def _parse_bill_detail(soup: BeautifulSoup) -> BillDetail:
     """Extract bill detail from the bill detail page."""
     # Header: invoice number and date of issue
@@ -409,110 +507,81 @@ def _parse_bill_detail(soup: BeautifulSoup) -> BillDetail:
         soup, "#BillItemsContainer > .bill-table-row:first-of-type"
     )
     # If the CSS child selector doesn't match, fall back to looking at
-    # desktop rows that come before .other-charges-table-row
+    # desktop rows that come before .other-charges-table-row.
     if not consumption_charges:
         items_container = soup.select_one("#BillItemsContainer")
         if items_container:
-            consumption_rows: list[Tag] = []
             for child in items_container.children:
                 if not isinstance(child, Tag):
                     continue
-                if "other-charges-table-row" in child.get("class", []):
+                if _has_class(child, "other-charges-table-row"):
                     break
-                desktop = child.select_one(".bill-table-row-desktop .bill-table-row")
-                if desktop:
-                    consumption_rows.append(desktop)
-                elif "bill-table-row-desktop" in child.get("class", []):
-                    inner = child.select_one(".bill-table-row")
-                    if inner:
-                        consumption_rows.append(inner)
-            for row in consumption_rows:
-                desc_el = row.select_one(".bill-table-row-item-left")
-                units_el = row.select_one(".bill-table-row-item")
-                charge_el = row.select_one(".bill-table-row-item-right")
-                if desc_el and charge_el:
-                    consumption_charges.append(
-                        BillCharge(
-                            description=desc_el.text.strip(),
-                            units=(units_el.text.strip() if units_el else ""),
-                            charge=_parse_currency(charge_el.text),
-                        )
-                    )
-
-    # Other charges
+                desktop_row = child.select_one(
+                    ".bill-table-row-desktop .bill-table-row"
+                )
+                if desktop_row is None and _has_class(child, "bill-table-row-desktop"):
+                    desktop_row = child.select_one(".bill-table-row")
+                if desktop_row is not None:
+                    charge = _parse_charge_row(desktop_row)
+                    if charge is not None:
+                        consumption_charges.append(charge)
+
+    # Other charges: desktop rows that follow the .other-charges-table-row header,
+    # stopping at the first sibling that isn't a desktop or narrow row.
     other_charges: list[BillCharge] = []
     other_header = soup.select_one(".other-charges-table-row")
     if other_header:
         sibling = other_header.next_sibling
         while sibling:
             if isinstance(sibling, Tag):
-                if "bill-table-row-desktop" in sibling.get("class", []):
+                if _has_class(sibling, "bill-table-row-desktop"):
                     charge_row = sibling.select_one(".bill-table-row")
-                    if charge_row:
-                        desc_el = charge_row.select_one(".bill-table-row-item-left")
-                        units_el = charge_row.select_one(".bill-table-row-item")
-                        charge_el = charge_row.select_one(".bill-table-row-item-right")
-                        if desc_el and charge_el:
-                            other_charges.append(
-                                BillCharge(
-                                    description=desc_el.text.strip(),
-                                    units=(units_el.text.strip() if units_el else ""),
-                                    charge=_parse_currency(charge_el.text),
-                                )
-                            )
-                elif "bill-table-row-narrow" not in sibling.get("class", []):
+                    if charge_row is not None:
+                        charge = _parse_charge_row(charge_row)
+                        if charge is not None:
+                            other_charges.append(charge)
+                elif not _has_class(sibling, "bill-table-row-narrow"):
                     break
             sibling = sibling.next_sibling
 
-    # Totals
-    totals_container = soup.select_one("#BillTotalsCollapsibleContent")
-    total_excl_vat = 0.0
-    vat = 0.0
-    total_rows = (
-        totals_container.select(".bill-total-table-row") if totals_container else []
+    # Totals. Defaults to 0.0 stay for backwards compatibility with callers, but
+    # log when an expected field is missing so problems aren't completely silent.
+    totals = _parse_keyed_currency_rows(
+        soup.select_one("#BillTotalsCollapsibleContent"),
+        {
+            "total_excl_vat": lambda label: label.startswith(
+                "Total charges excluding VAT"
+            ),
+            "vat": lambda label: label.startswith("VAT"),
+        },
     )
-    for row in total_rows:
-        label_el = row.select_one(".bill-total-table-row-item-left")
-        value_el = row.select_one(".bill-total-table-row-item-right")
-        if label_el and value_el:
-            label = label_el.text.strip()
-            if label.startswith("Total charges excluding VAT"):
-                total_excl_vat = _parse_currency(value_el.text)
-            elif label.startswith("VAT"):
-                vat = _parse_currency(value_el.text)
+    total_excl_vat = totals.get("total_excl_vat", 0.0)
+    vat = totals.get("vat", 0.0)
 
     # Bill total from the collapsible header
-    total = 0.0
-    total_header = soup.select_one(
-        "#BillTotalsContainer > .collapsible-header .bill-total-table-row-item-right"
+    total = _parse_currency_header(
+        soup,
+        "#BillTotalsContainer > .collapsible-header .bill-total-table-row-item-right",
+        "bill total",
     )
-    if total_header:
-        total = _parse_currency(total_header.text)
 
     # Account balance
-    balance_container = soup.select_one("#AccountBalanceCollapsibleContent")
-    previous_balance = 0.0
-    payments_received = 0.0
-    balance_rows = (
-        balance_container.select(".bill-total-table-row") if balance_container else []
+    balances = _parse_keyed_currency_rows(
+        soup.select_one("#AccountBalanceCollapsibleContent"),
+        {
+            "previous_balance": lambda label: label == "Previous account balance",
+            "payments_received": lambda label: label == "Payments received",
+        },
     )
-    for row in balance_rows:
-        label_el = row.select_one(".bill-total-table-row-item-left")
-        value_el = row.select_one(".bill-total-table-row-item-right")
-        if label_el and value_el:
-            label = label_el.text.strip()
-            if label == "Previous account balance":
-                previous_balance = _parse_currency(value_el.text)
-            elif label == "Payments received":
-                payments_received = _parse_currency(value_el.text)
-
-    balance = 0.0
-    balance_header = soup.select_one(
+    previous_balance = balances.get("previous_balance", 0.0)
+    payments_received = balances.get("payments_received", 0.0)
+
+    balance = _parse_currency_header(
+        soup,
         "#AccountBalanceContainer > .collapsible-header"
-        " .bill-total-table-row-item-right"
+        " .bill-total-table-row-item-right",
+        "account balance",
     )
-    if balance_header:
-        balance = _parse_currency(balance_header.text)
 
     # Download URL
     download_url = ""

{switch2-0.1.1 → switch2-0.1.3}/switch2.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: switch2
-Version: 0.1.1
+Version: 0.1.3
 Summary: Async client library for the Switch2 energy portal
 Author-email: Jelmer Vernooij <jelmer@jelmer.uk>
 License-Expression: Apache-2.0
@@ -38,16 +38,17 @@ import asyncio
 from switch2 import Switch2ApiClient
 
 async def main():
-    client = Switch2ApiClient("you@example.com", "your-password")
-    try:
+    async with Switch2ApiClient("you@example.com", "your-password") as client:
         data = await client.fetch_data()
         print(f"Customer: {data.customer.name}")
+        if data.account_balance is not None:
+            print(f"Balance: £{data.account_balance.balance}")
         for reading in data.readings:
             print(f"  {reading.date}: {reading.amount} {reading.unit}")
         for bill in data.bills:
             print(f"  {bill.date}: £{bill.amount}")
-    finally:
-        await client.close()
+            detail = await client.fetch_bill_detail(bill)
+            print(f"    invoice {detail.invoice_number}, total £{detail.total}")
 
 asyncio.run(main())
 ```

{switch2-0.1.1 → switch2-0.1.3}/switch2.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ README.md
 pyproject.toml
 switch2/__init__.py
 switch2/api.py
+switch2/py.typed
 switch2.egg-info/PKG-INFO
 switch2.egg-info/SOURCES.txt
 switch2.egg-info/dependency_links.txt

{switch2-0.1.1 → switch2-0.1.3}/tests/test_api.py

@@ -1,11 +1,14 @@
 """Tests for the Switch2 API client."""
 
+import asyncio
 import unittest
 from datetime import datetime
 
 from bs4 import BeautifulSoup
 
 from switch2.api import (
+    Switch2ApiClient,
+    _parse_account_balance,
     _parse_bill_detail,
     _parse_bills,
     _parse_currency,
@@ -350,3 +353,52 @@ class ParseBillDetailTests(unittest.TestCase):
             detail.download_url,
             "https://my.switch2.co.uk/Credit/Bill/Download/2289896",
         )
+
+
+class ClientContextManagerTests(unittest.TestCase):
+    def test_aclose_without_session(self) -> None:
+        async def run() -> None:
+            async with Switch2ApiClient("e", "p") as client:
+                self.assertIsNone(client._session)
+
+        asyncio.run(run())
+
+
+class ParseAccountBalanceTests(unittest.TestCase):
+    def test_balance_from_dashboard(self) -> None:
+        html = """
+        <div id="CreditPaymentBalanceDashTile">
+            <div class="dashboard-credit-amount-desktop font-largest">
+                &#163;0.00
+            </div>
+            <div class="dashboard-credit-lastUpdated font-small-light">
+                Last updated 27/02/2026 10:13
+            </div>
+        </div>
+        """
+        soup = BeautifulSoup(html, "html.parser")
+        balance = _parse_account_balance(soup)
+        assert balance is not None
+        self.assertEqual(balance.balance, 0.00)
+        self.assertEqual(balance.last_updated, datetime(2026, 2, 27, 10, 13))
+
+    def test_balance_with_amount_owed(self) -> None:
+        html = """
+        <div id="CreditPaymentBalanceDashTile">
+            <div class="dashboard-credit-amount-desktop font-largest">
+                &#163;172.26
+            </div>
+            <div class="dashboard-credit-lastUpdated font-small-light">
+                Last updated 15/03/2026 14:30
+            </div>
+        </div>
+        """
+        soup = BeautifulSoup(html, "html.parser")
+        balance = _parse_account_balance(soup)
+        assert balance is not None
+        self.assertEqual(balance.balance, 172.26)
+        self.assertEqual(balance.last_updated, datetime(2026, 3, 15, 14, 30))
+
+    def test_no_balance_element(self) -> None:
+        soup = BeautifulSoup("<div></div>", "html.parser")
+        self.assertIsNone(_parse_account_balance(soup))