s2-python 0.2.0__tar.gz → 0.2.2__tar.gz

{s2_python-0.2.0 → s2_python-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: s2-python
-Version: 0.2.0
+Version: 0.2.2
 Summary: S2 Protocol Python Wrapper
 Home-page: https://github.com/flexiblepower/s2-ws-json-python
 Author: Flexiblepower
@@ -17,7 +17,7 @@ Description-Content-Type: text/x-rst; charset=UTF-8
 Requires-Dist: pydantic~=2.8.2
 Requires-Dist: pytz
 Requires-Dist: click
-Requires-Dist: websockets~=13.0.1
+Requires-Dist: websockets~=13.1
 Provides-Extra: testing
 Requires-Dist: pytest; extra == "testing"
 Requires-Dist: pytest-coverage; extra == "testing"
@@ -62,6 +62,18 @@ You can install this package using pip or any Python dependency manager that col
 
 The packages on Pypi may be found `here <https://pypi.org/project/s2-python/>`_
 
+Mypy support
+------------
+s2-python uses pydantic at its core to define the various S2 messages. As such, the pydantic mypy plugin is required
+for type checking to succeed.
+
+Add to your pyproject.toml:
+
+.. code-block:: toml
+
+    [tool.mypy]
+    plugins = ['pydantic.mypy']
+
 Example
 ---------
 

{s2_python-0.2.0 → s2_python-0.2.2}/README.rst

@@ -22,6 +22,18 @@ You can install this package using pip or any Python dependency manager that col
 
 The packages on Pypi may be found `here <https://pypi.org/project/s2-python/>`_
 
+Mypy support
+------------
+s2-python uses pydantic at its core to define the various S2 messages. As such, the pydantic mypy plugin is required
+for type checking to succeed.
+
+Add to your pyproject.toml:
+
+.. code-block:: toml
+
+    [tool.mypy]
+    plugins = ['pydantic.mypy']
+
 Example
 ---------
 

{s2_python-0.2.0 → s2_python-0.2.2}/setup.cfg

@@ -8,7 +8,7 @@ license_files = LICENSE.txt
 long_description = file: README.rst
 long_description_content_type = text/x-rst; charset=UTF-8
 url = https://github.com/flexiblepower/s2-ws-json-python
-version = 0.2.0
+version = 0.2.2
 platforms = Linux
 classifiers = 
 	Development Status :: 4 - Beta
@@ -29,7 +29,7 @@ install_requires =
 	pydantic~=2.8.2
 	pytz
 	click
-	websockets~=13.0.1
+	websockets~=13.1
 
 [options.packages.find]
 where = src

{s2_python-0.2.0 → s2_python-0.2.2}/src/s2_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: s2-python
-Version: 0.2.0
+Version: 0.2.2
 Summary: S2 Protocol Python Wrapper
 Home-page: https://github.com/flexiblepower/s2-ws-json-python
 Author: Flexiblepower
@@ -17,7 +17,7 @@ Description-Content-Type: text/x-rst; charset=UTF-8
 Requires-Dist: pydantic~=2.8.2
 Requires-Dist: pytz
 Requires-Dist: click
-Requires-Dist: websockets~=13.0.1
+Requires-Dist: websockets~=13.1
 Provides-Extra: testing
 Requires-Dist: pytest; extra == "testing"
 Requires-Dist: pytest-coverage; extra == "testing"
@@ -62,6 +62,18 @@ You can install this package using pip or any Python dependency manager that col
 
 The packages on Pypi may be found `here <https://pypi.org/project/s2-python/>`_
 
+Mypy support
+------------
+s2-python uses pydantic at its core to define the various S2 messages. As such, the pydantic mypy plugin is required
+for type checking to succeed.
+
+Add to your pyproject.toml:
+
+.. code-block:: toml
+
+    [tool.mypy]
+    plugins = ['pydantic.mypy']
+
 Example
 ---------
 

{s2_python-0.2.0 → s2_python-0.2.2}/src/s2_python.egg-info/SOURCES.txt

@@ -10,6 +10,7 @@ src/s2_python.egg-info/not-zip-safe
 src/s2_python.egg-info/requires.txt
 src/s2_python.egg-info/top_level.txt
 src/s2python/__init__.py
+src/s2python/py.typed
 src/s2python/reception_status_awaiter.py
 src/s2python/s2_connection.py
 src/s2python/s2_control_type.py

{s2_python-0.2.0 → s2_python-0.2.2}/src/s2_python.egg-info/requires.txt

@@ -1,7 +1,7 @@
 pydantic~=2.8.2
 pytz
 click
-websockets~=13.0.1
+websockets~=13.1
 
 [development]
 pip-tools

s2_python-0.2.2/src/s2python/common/number_range.py

@@ -0,0 +1,22 @@
+from typing import Any
+
+from s2python.validate_values_mixin import S2Message, catch_and_convert_exceptions
+from s2python.generated.gen_s2 import NumberRange as GenNumberRange
+
+
+@catch_and_convert_exceptions
+class NumberRange(GenNumberRange, S2Message["NumberRange"]):
+    model_config = GenNumberRange.model_config
+    model_config["validate_assignment"] = True
+
+    def __hash__(self) -> int:
+        return hash(f"{self.start_of_range}|{self.end_of_range}")
+
+    def __eq__(self, other: Any) -> bool:
+        if isinstance(other, NumberRange):
+            return (
+                self.start_of_range == other.start_of_range
+                and self.end_of_range == other.end_of_range
+            )
+
+        return False

s2_python-0.2.2/src/s2python/frbc/frbc_fill_level_target_profile_element.py

@@ -0,0 +1,34 @@
+# pylint: disable=duplicate-code
+
+from typing_extensions import Self
+
+from pydantic import model_validator
+
+from s2python.common import Duration, NumberRange
+from s2python.generated.gen_s2 import (
+    FRBCFillLevelTargetProfileElement as GenFRBCFillLevelTargetProfileElement,
+)
+from s2python.validate_values_mixin import catch_and_convert_exceptions, S2Message
+
+
+@catch_and_convert_exceptions
+class FRBCFillLevelTargetProfileElement(
+    GenFRBCFillLevelTargetProfileElement, S2Message["FRBCFillLevelTargetProfileElement"]
+):
+    model_config = GenFRBCFillLevelTargetProfileElement.model_config
+    model_config["validate_assignment"] = True
+
+    duration: Duration = GenFRBCFillLevelTargetProfileElement.model_fields["duration"]  # type: ignore[assignment]
+    fill_level_range: NumberRange = GenFRBCFillLevelTargetProfileElement.model_fields[
+        "fill_level_range"
+    ]  # type: ignore[assignment]
+
+    @model_validator(mode="after")
+    def validate_start_end_order(self) -> Self:
+        if self.fill_level_range.start_of_range > self.fill_level_range.end_of_range:
+            raise ValueError(
+                self,
+                "start_of_range should not be higher than end_of_range for the fill_level_range",
+            )
+
+        return self

s2_python-0.2.2/src/s2python/frbc/frbc_leakage_behaviour_element.py

@@ -0,0 +1,30 @@
+# pylint: disable=duplicate-code
+
+from pydantic import model_validator
+from typing_extensions import Self
+
+from s2python.common import NumberRange
+from s2python.generated.gen_s2 import FRBCLeakageBehaviourElement as GenFRBCLeakageBehaviourElement
+from s2python.validate_values_mixin import catch_and_convert_exceptions, S2Message
+
+
+@catch_and_convert_exceptions
+class FRBCLeakageBehaviourElement(
+    GenFRBCLeakageBehaviourElement, S2Message["FRBCLeakageBehaviourElement"]
+):
+    model_config = GenFRBCLeakageBehaviourElement.model_config
+    model_config["validate_assignment"] = True
+
+    fill_level_range: NumberRange = GenFRBCLeakageBehaviourElement.model_fields[
+        "fill_level_range"
+    ]  # type: ignore[assignment]
+
+    @model_validator(mode="after")
+    def validate_start_end_order(self) -> Self:
+        if self.fill_level_range.start_of_range > self.fill_level_range.end_of_range:
+            raise ValueError(
+                self,
+                "start_of_range should not be higher than end_of_range for the fill_level_range",
+            )
+
+        return self

{s2_python-0.2.0 → s2_python-0.2.2}/src/s2python/s2_connection.py

@@ -6,6 +6,7 @@ import uuid
 from dataclasses import dataclass
 from typing import Optional, List, Type, Dict, Callable, Awaitable, Union
 
+import websockets
 from websockets.asyncio.client import ClientConnection as WSConnection, connect as ws_connect
 
 from s2python.common import (
@@ -193,7 +194,6 @@ class S2Connection:  # pylint: disable=too-many-instance-attributes
     _received_messages: asyncio.Queue
 
     _eventloop: asyncio.AbstractEventLoop
-    _background_tasks: Optional[asyncio.Task]
     _stop_event: asyncio.Event
 
     def __init__(
@@ -211,7 +211,6 @@ class S2Connection:  # pylint: disable=too-many-instance-attributes
         self._current_control_type = None
 
         self._eventloop = asyncio.new_event_loop()
-        self._background_tasks = None
 
         self.control_types = control_types
         self.role = role
@@ -222,13 +221,17 @@ class S2Connection:  # pylint: disable=too-many-instance-attributes
         self._handlers.register_handler(HandshakeResponse, self.handle_handshake_response_as_rm)
 
     def start_as_rm(self) -> None:
-        self._thread = threading.Thread(target=self._run_eventloop)
+        self._thread = threading.Thread(target=self._run_eventloop, daemon=False)
         self._thread.start()
         logger.debug("Started eventloop thread!")
 
     def _run_eventloop(self) -> None:
         logger.debug("Starting eventloop")
-        self._eventloop.run_until_complete(self._run_as_rm())
+        try:
+            self._eventloop.run_until_complete(self._run_as_rm())
+        except asyncio.CancelledError:
+            pass
+        logger.debug("S2 connection thread has stopped.")
 
     def stop(self) -> None:
         """Stops the S2 connection.
@@ -242,41 +245,51 @@ class S2Connection:  # pylint: disable=too-many-instance-attributes
                 "Do not call stop from the thread running the S2 connection. This results in an "
                 "infinite block!"
             )
-
-        asyncio.run_coroutine_threadsafe(self._do_stop(), self._eventloop).result()
+        if self._eventloop.is_running():
+            asyncio.run_coroutine_threadsafe(self._do_stop(), self._eventloop).result()
+        self._thread.join()
+        logger.info("Stopped the S2 connection.")
 
     async def _do_stop(self) -> None:
         logger.info("Will stop the S2 connection.")
-        if self._background_tasks:
-            self._background_tasks.cancel()
-            self._background_tasks = None
-
-        if self.ws:
-            await self.ws.close()
-            await self.ws.wait_closed()
+        self._stop_event.set()
 
     async def _run_as_rm(self) -> None:
         logger.debug("Connecting as S2 resource manager.")
         self._received_messages = asyncio.Queue()
+        self._stop_event = asyncio.Event()
         await self.connect_ws()
 
-        self._background_tasks = self._eventloop.create_task(
-            asyncio.wait(
-                (self._receive_messages(), self._handle_received_messages()),
-                return_when=asyncio.FIRST_EXCEPTION,
-            )
-        )
+        background_tasks = []
+        background_tasks.append(self._eventloop.create_task(self._receive_messages()))
+        background_tasks.append(self._eventloop.create_task(self._handle_received_messages()))
 
-        await self.connect_as_rm()
-        done: List[asyncio.Task]
-        pending: List[asyncio.Task]
-        (done, pending) = await self._background_tasks
+        async def wait_till_stop() -> None:
+            await self._stop_event.wait()
 
+        background_tasks.append(self._eventloop.create_task(wait_till_stop()))
+
+        await self.connect_as_rm()
+        (done, pending) = await asyncio.wait(background_tasks, return_when=asyncio.FIRST_COMPLETED)
         for task in done:
-            task.result()
+            try:
+                await task
+            except asyncio.CancelledError:
+                pass
+            except websockets.ConnectionClosedError:
+                logger.info("The other party closed the websocket connection.c")
 
         for task in pending:
-            task.cancel()
+            try:
+                task.cancel()
+                await task
+            except asyncio.CancelledError:
+                pass
+
+        if self.ws:
+            await self.ws.close()
+            await self.ws.wait_closed()
+        logger.debug("Finished S2 connection eventloop.")
 
     async def connect_ws(self) -> None:
         self.ws = await ws_connect(uri=self.url)

s2_python-0.2.0/src/s2python/common/number_range.py

@@ -1,32 +0,0 @@
-from typing import Any
-from typing_extensions import Self
-
-from pydantic import model_validator
-
-from s2python.validate_values_mixin import (
-    S2Message,
-    catch_and_convert_exceptions,
-)
-from s2python.generated.gen_s2 import NumberRange as GenNumberRange
-
-
-@catch_and_convert_exceptions
-class NumberRange(GenNumberRange, S2Message["NumberRange"]):
-    model_config = GenNumberRange.model_config
-    model_config["validate_assignment"] = True
-
-    @model_validator(mode="after")
-    def validate_start_end_order(self) -> Self:  # pylint: disable=duplicate-code
-        if self.start_of_range > self.end_of_range:
-            raise ValueError(self, "start_of_range should not be higher than end_of_range")
-
-        return self
-
-    def __hash__(self) -> int:
-        return hash(f"{self.start_of_range}|{self.end_of_range}")
-
-    def __eq__(self, other: Any) -> bool:
-        if isinstance(other, NumberRange):
-            return self.start_of_range == other.start_of_range and self.end_of_range == other.end_of_range
-
-        return False

s2_python-0.2.0/src/s2python/frbc/frbc_fill_level_target_profile_element.py

@@ -1,23 +0,0 @@
-from s2python.common import Duration, NumberRange
-
-from s2python.generated.gen_s2 import (
-    FRBCFillLevelTargetProfileElement as GenFRBCFillLevelTargetProfileElement,
-)
-from s2python.validate_values_mixin import (
-    catch_and_convert_exceptions,
-    S2Message,
-)
-
-
-@catch_and_convert_exceptions
-class FRBCFillLevelTargetProfileElement(
-    GenFRBCFillLevelTargetProfileElement,
-    S2Message["FRBCFillLevelTargetProfileElement"],
-):
-    model_config = GenFRBCFillLevelTargetProfileElement.model_config
-    model_config["validate_assignment"] = True
-
-    duration: Duration = GenFRBCFillLevelTargetProfileElement.model_fields["duration"]  # type: ignore[assignment]
-    fill_level_range: NumberRange = GenFRBCFillLevelTargetProfileElement.model_fields[
-        "fill_level_range"
-    ]  # type: ignore[assignment]

s2_python-0.2.0/src/s2python/frbc/frbc_leakage_behaviour_element.py

@@ -1,18 +0,0 @@
-from s2python.common import NumberRange
-from s2python.generated.gen_s2 import (
-    FRBCLeakageBehaviourElement as GenFRBCLeakageBehaviourElement,
-)
-from s2python.validate_values_mixin import (
-    catch_and_convert_exceptions,
-    S2Message,
-)
-
-
-@catch_and_convert_exceptions
-class FRBCLeakageBehaviourElement(GenFRBCLeakageBehaviourElement, S2Message["FRBCLeakageBehaviourElement"]):
-    model_config = GenFRBCLeakageBehaviourElement.model_config
-    model_config["validate_assignment"] = True
-
-    fill_level_range: NumberRange = GenFRBCLeakageBehaviourElement.model_fields[
-        "fill_level_range"
-    ]  # type: ignore[assignment]