remove logger

Author: marcelveldt

blueconnect/__init__.py

@@ -1,7 +1,6 @@
 """Unofficial python library for the Blue Riiot Blue Connect API."""
 
 import asyncio
-import logging
 from typing import List
 
 from .api import BlueConnectApi
@@ -13,8 +12,6 @@
     TemperatureUnit,
 )
 
-LOGGER = logging.getLogger()
-
 
 class BlueConnectSimpleAPI:
     """Class that provides a common structure around the Blue Connect API for a single/main swimming pool."""
@@ -39,12 +36,17 @@ async def fetch_data(self) -> None:
         main_pool_id = user_info.user_preferences.main_swimming_pool_id
         self._temperature_unit = user_info.user_preferences.display_temperature_unit
         self._pool_info = await self._api.get_swimming_pool(main_pool_id)
-        self._pool_feed_message = (await self._api.get_swimming_pool_feed(main_pool_id, self._language)).current_message
+        self._pool_feed_message = (
+            await self._api.get_swimming_pool_feed(main_pool_id, self._language)
+        ).current_message
         blue_devices = await self._api.get_swimming_pool_blue_devices(main_pool_id)
         self._pool_blue_device = blue_devices[0] if blue_devices else None
         if self._pool_blue_device:
-            self._pool_measurements = (await self._api.get_last_measurements(
-                main_pool_id, self._pool_blue_device.serial)).measurements
+            self._pool_measurements = (
+                await self._api.get_last_measurements(
+                    main_pool_id, self._pool_blue_device.serial
+                )
+            ).measurements
 
     @property
     def pool(self) -> SwimmingPool:

blueconnect/api.py

@@ -1,7 +1,6 @@
 """Unofficial python library for the Blue Riiot Blue Connect API."""
 
 import asyncio
-import logging
 import time
 from typing import List, Optional
 
@@ -25,7 +24,6 @@
     "Accept": "*/*",
 }
 BASE_URL = "https://api.riiotlabs.com/prod/"
-LOGGER = logging.getLogger()
 
 
 class BlueConnectApi:
@@ -35,9 +33,7 @@ def __init__(self, username: str, password: str) -> None:
         """Inititialize the api connection, a valid username and password must be provided."""
         self._username = username
         self._password = password
-        self._http_session = aiohttp.ClientSession(
-            connector=aiohttp.TCPConnector()
-        )
+        self._http_session = aiohttp.ClientSession(connector=aiohttp.TCPConnector())
         self._token_info = {}
 
     def close(self) -> None:
@@ -71,12 +67,16 @@ async def get_swimming_pool(self, swimming_pool_id: str) -> SwimmingPool:
         data = await self.__get_data(f"swimming_pool/{swimming_pool_id}")
         return SwimmingPool.from_json(data)
 
-    async def get_swimming_pool_status(self, swimming_pool_id: str) -> SwimmingPoolStatus:
+    async def get_swimming_pool_status(
+        self, swimming_pool_id: str
+    ) -> SwimmingPoolStatus:
         """Retrieve status for a specific swimming pool."""
         data = await self.__get_data(f"swimming_pool/{swimming_pool_id}/status")
         return SwimmingPoolStatus.from_json(data)
 
-    async def get_swimming_pool_blue_devices(self, swimming_pool_id: str) -> List[BlueDevice]:
+    async def get_swimming_pool_blue_devices(
+        self, swimming_pool_id: str
+    ) -> List[BlueDevice]:
         """Retrieve Blue devices for a specific swimming pool."""
         data = await self.__get_data(f"swimming_pool/{swimming_pool_id}/blue")
         result = []
@@ -85,14 +85,18 @@ async def get_swimming_pool_blue_devices(self, swimming_pool_id: str) -> List[Bl
             result.append(blue_device)
         return result
 
-    async def get_swimming_pool_feed(self, swimming_pool_id: str, language: str = "en") -> SwimmingPoolFeed:
+    async def get_swimming_pool_feed(
+        self, swimming_pool_id: str, language: str = "en"
+    ) -> SwimmingPoolFeed:
         """Retrieve feed for a specific swimming pool."""
         data = await self.__get_data(
             f"swimming_pool/{swimming_pool_id}/feed?lang={language}"
         )
         return SwimmingPoolFeed.from_json(data)
 
-    async def get_last_measurements(self, swimming_pool_id: str, blue_device_serial: str) -> SwimmingPoolLastMeasurements:
+    async def get_last_measurements(
+        self, swimming_pool_id: str, blue_device_serial: str
+    ) -> SwimmingPoolLastMeasurements:
         """Retrieve last measurements for a specific swimming pool."""
         data = await self.__get_data(
             f"swimming_pool/{swimming_pool_id}/blue/{blue_device_serial}/lastMeasurements?mode=blue_and_strip"
@@ -135,5 +139,8 @@ async def __get_data(self, endpoint: str, params: dict = None) -> Optional[dict]
         ) as response:
             if response.status != 200:
                 error_msg = await response.text()
-                raise Exception("Error while retrieving data for endpoint %s: %s" % (endpoint, error_msg))
+                raise Exception(
+                    "Error while retrieving data for endpoint %s: %s"
+                    % (endpoint, error_msg)
+                )
             return await response.json()

blueconnect/models.py

@@ -19,7 +19,7 @@ def transform_value(value, value_type):
     if isinstance(value, list):
         item_type = value_type.__args__[0]
         return [transform_value(item, item_type) for item in value]
-    if hasattr(value_type, 'from_json'):
+    if hasattr(value_type, "from_json"):
         return value_type.from_json(value)
     return value_type(value)
 
@@ -48,6 +48,7 @@ def from_json(cls, json_obj):
 @dataclass
 class BlueDevice(ModelBase):
     """Model for a Blue Connect device."""
+
     serial: str
     hw_generation: int
     hw_product_type: str
@@ -59,6 +60,7 @@ class BlueDevice(ModelBase):
 
 class MeasurementTrend(Enum):
     """Enum for a measurement trend."""
+
     STABLE = "stable"
     INCREASE = "increase"
     DECREASE = "decrease"
@@ -67,6 +69,7 @@ class MeasurementTrend(Enum):
 @dataclass
 class SwimmingPoolMeasurement(ModelBase):
     """Model for a swimming pool measurement."""
+
     name: str
     priority: int
     timestamp: datetime
@@ -85,10 +88,11 @@ class SwimmingPoolMeasurement(ModelBase):
 @dataclass
 class SwimmingPoolLastMeasurements(ModelBase):
     """Model for swimming pool last measurements data."""
+
     status: str
     swimming_pool_id: str
     data: List[SwimmingPoolMeasurement]
-    blue_device_serial: str = ''
+    blue_device_serial: str = ""
     last_blue_measure_timestamp: datetime = None
     last_strip_timestamp: datetime = None
 
@@ -101,6 +105,7 @@ def measurements(self):
 @dataclass
 class UserInfo(ModelBase):
     """Model for UserInfo."""
+
     user_id: str
     first_name: str
     last_name: str
@@ -110,13 +115,15 @@ class UserInfo(ModelBase):
 
 class TemperatureUnit(Enum):
     """Enum for TemperatureUnit."""
+
     CELSIUS = "celsius"
     FAHRENHEIT = "fahrenheit"
 
 
 @dataclass
 class UserPreferences(ModelBase):
     """Model for UserPreferences."""
+
     display_temperature_unit: TemperatureUnit
     display_unit_system: str
     main_swimming_pool_id: str
@@ -125,13 +132,15 @@ class UserPreferences(ModelBase):
 @dataclass
 class User(ModelBase):
     """Model for a User."""
+
     user_info: UserInfo
     user_preferences: UserPreferences
 
 
 @dataclass
 class SwimmingPool(ModelBase):
     """Model for a SwimmingPool."""
+
     updated: datetime
     swimming_pool_id: str
     created: datetime
@@ -142,6 +151,7 @@ class SwimmingPool(ModelBase):
 @dataclass
 class SwimmingPoolStatusTask(ModelBase):
     """Model for a SwimmingPoolStatusTask."""
+
     status_id: str
     since: datetime
     data: dict
@@ -155,6 +165,7 @@ class SwimmingPoolStatusTask(ModelBase):
 @dataclass
 class SwimmingPoolStatus(ModelBase):
     """Model for SwimmingPoolStatus."""
+
     since: datetime
     status_id: str
     global_status_code: str
@@ -170,6 +181,7 @@ class SwimmingPoolStatus(ModelBase):
 @dataclass
 class SwimmingPoolFeedMessage(ModelBase):
     """Model for a SwimmingPoolFeedMessage."""
+
     id: str
     title: str
     message: str
@@ -178,6 +190,7 @@ class SwimmingPoolFeedMessage(ModelBase):
 @dataclass
 class SwimmingPoolFeed(ModelBase):
     """Model for a SwimmingPoolFeedMessage."""
+
     swimming_pool_id: str
     timestamp: datetime
     data: List[SwimmingPoolFeedMessage]

test/test_base.py

@@ -1,12 +1,14 @@
 """Tests for Blue Connect api library."""
 
-from ..blueconnect import BlueConnectSimpleAPI
 import pytest
 
-api = BlueConnectSimpleAPI("test", "test", "nl")
+from blueconnect import BlueConnectSimpleAPI
 
 # TODO TODO TODO
 
 @pytest.mark.asyncio
 async def test_base():
-    print(api.measurements)
+    # api = BlueConnectSimpleAPI("test", "test", "nl")
+    # print(api.measurements)
+
+    pass