Add a context manager to TahomaClient (#16)

Author: tetienne

.github/workflows/main.yaml

@@ -9,29 +9,9 @@ on:
       - master
 
 jobs:
-  tests:
-    name: "Python ${{ matrix.python-version }}"
-    runs-on: "ubuntu-latest"
-
-    strategy:
-      matrix:
-        python-version: ["3.6", "3.7", "3.8"]
-
+  pre-commit:
+    runs-on: ubuntu-latest
     steps:
-      - uses: "actions/checkout@v2"
-      - uses: "actions/setup-python@v1"
-        with:
-          python-version: "${{ matrix.python-version }}"
-
-      - name: Install and set up Poetry
-        run: |
-          curl -o get-poetry.py https://raw.githubusercontent.com/python-poetry/poetry/master/get-poetry.py
-          python get-poetry.py -y
-
-      - name: "Install tox"
-        run: |
-          python -m pip install --upgrade tox tox-gh-actions
-      - name: "Run tox targets for ${{ matrix.python-version }}"
-        run: |
-          source $HOME/.poetry/env
-          python -m tox
+    - uses: actions/checkout@v1
+    - uses: actions/setup-python@v1
+    - uses: pre-commit/action@v2.0.0

tahoma_api/client.py

@@ -1,6 +1,9 @@
 """ Python wrapper for the Tahoma API """
+from __future__ import annotations
+
 import urllib.parse
-from typing import Any, Dict, List, Optional, Union
+from types import TracebackType
+from typing import Any, Dict, List, Optional, Type, Union
 
 import aiohttp
 import humps
@@ -34,6 +37,17 @@ def __init__(self, username: str, password: str, api_url: str = API_URL) -> None
 
         self.session = aiohttp.ClientSession()
 
+    async def __aenter__(self) -> TahomaClient:
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type: Optional[Type[BaseException]],
+        exc_value: Optional[BaseException],
+        traceback: Optional[TracebackType],
+    ) -> None:
+        await self.close()
+
     async def close(self) -> None:
         """Close the session."""
         await self.session.close()

test.py

@@ -3,39 +3,34 @@
 
 from tahoma_api.client import TahomaClient
 
-# TODO use .env file
 USERNAME = ""
 PASSWORD = ""
 
 
 async def main() -> None:
-    client = TahomaClient(USERNAME, PASSWORD)
+    async with TahomaClient(USERNAME, PASSWORD) as client:
+        try:
+            await client.login()
+        except Exception as exception:  # pylint: disable=broad-except
+            print(exception)
+            return await client.close()
 
-    try:
-        await client.login()
-    except Exception as exception:  # pylint: disable=broad-except
-        print(exception)
-        return await client.close()
+        devices = await client.get_devices()
 
-    devices = await client.get_devices()
+        for device in devices:
+            print(f"{device.label} ({device.id}) - {device.controllable_name}")
+            print(f"{device.widget} - {device.ui_class}")
+            # print(device.states)
+            # print(device.definition)
 
-    for device in devices:
-        print(f"{device.label} ({device.id}) - {device.controllable_name}")
-        print(f"{device.widget} - {device.ui_class}")
-        # print(device.states)
-        # print(device.definition)
+        # Create an event listener and poll it
+        listener_id = await client.register_event_listener()
 
-    # Create an event listener and poll it
-    listener_id = await client.register_event_listener()
+        while True:
+            events = await client.fetch_event_listener(listener_id)
+            print(events)
 
-    while True:
-        events = await client.fetch_event_listener(listener_id)
-        print(events)
-
-        time.sleep(2)
-
-    # Make sure you call client.close() when you are done
-    await client.close()
+            time.sleep(2)
 
 
 asyncio.run(main())