runtimepy 5.4.3__py3-none-any.whl → 5.4.4__py3-none-any.whl

runtimepy/__init__.py

@@ -1,7 +1,7 @@
 # =====================================
 # generator=datazen
 # version=3.1.4
-# hash=d1fd99eada85e38ef836802f8c0a3c50
+# hash=5a37b52aedb2af82fb174fbea8ce0c1f
 # =====================================
 
 """
@@ -10,7 +10,7 @@ Useful defaults and other package metadata.
 
 DESCRIPTION = "A framework for implementing Python services."
 PKG_NAME = "runtimepy"
-VERSION = "5.4.3"
+VERSION = "5.4.4"
 
 # runtimepy-specific content.
 METRICS_NAME = "metrics"

runtimepy/net/udp/tftp/__init__.py

@@ -21,6 +21,7 @@ from runtimepy.net.udp.tftp.base import (
     DEFAULT_TIMEOUT_S,
     REEMIT_PERIOD_S,
     BaseTftpConnection,
+    TftpErrorHandler,
 )
 from runtimepy.net.udp.tftp.enums import DEFAULT_MODE
 from runtimepy.util import PossiblePath, as_path
@@ -204,6 +205,7 @@ async def tftp(
     connection_kwargs: dict[str, Any] = None,
     timeout_s: float = DEFAULT_TIMEOUT_S,
     reemit_period_s: float = REEMIT_PERIOD_S,
+    error_handler: TftpErrorHandler = None,
 ) -> AsyncIterator[TftpConnection]:
     """Use a tftp connection as a managed context."""
 
@@ -218,6 +220,11 @@ async def tftp(
     conn = await TftpConnection.create_connection(
         remote_addr=(addr.name, addr.port), **connection_kwargs
     )
+
+    # Add error handlers.
+    if error_handler is not None:
+        conn.error_handlers.append(error_handler)
+
     async with conn.process_then_disable(**process_kwargs):
         # Set parameters.
         conn.endpoint_timeout.value = timeout_s
@@ -235,6 +242,7 @@ async def tftp_write(
     connection_kwargs: dict[str, Any] = None,
     timeout_s: float = DEFAULT_TIMEOUT_S,
     reemit_period_s: float = REEMIT_PERIOD_S,
+    error_handler: TftpErrorHandler = None,
 ) -> bool:
     """Attempt to perform a tftp write."""
 
@@ -244,6 +252,7 @@ async def tftp_write(
         connection_kwargs=connection_kwargs,
         timeout_s=timeout_s,
         reemit_period_s=reemit_period_s,
+        error_handler=error_handler,
     ) as conn:
         # Perform tftp interaction.
         result = await conn.request_write(
@@ -262,6 +271,7 @@ async def tftp_read(
     connection_kwargs: dict[str, Any] = None,
     timeout_s: float = DEFAULT_TIMEOUT_S,
     reemit_period_s: float = REEMIT_PERIOD_S,
+    error_handler: TftpErrorHandler = None,
 ) -> bool:
     """Attempt to perform a tftp read."""
 
@@ -271,6 +281,7 @@ async def tftp_read(
         connection_kwargs=connection_kwargs,
         timeout_s=timeout_s,
         reemit_period_s=reemit_period_s,
+        error_handler=error_handler,
     ) as conn:
         result = await conn.request_read(
             destination, filename, mode=mode, addr=addr

runtimepy/net/udp/tftp/base.py

@@ -8,7 +8,7 @@ from contextlib import AsyncExitStack
 from io import BytesIO
 import logging
 from pathlib import Path
-from typing import BinaryIO, Union
+from typing import BinaryIO, Callable, Union
 
 # third-party
 from vcorelib.math import metrics_time_ns
@@ -30,6 +30,8 @@ from runtimepy.primitives import Double, Uint16
 REEMIT_PERIOD_S = 0.20
 DEFAULT_TIMEOUT_S = 1.0
 
+TftpErrorHandler = Callable[[TftpErrorCode, str, tuple[str, int]], None]
+
 
 class BaseTftpConnection(UdpConnection):
     """A class implementing a basic tftp interface."""
@@ -73,6 +75,7 @@ class BaseTftpConnection(UdpConnection):
 
         self.error_code = Uint16(time_source=metrics_time_ns)
         self.env.channel("error_code", self.error_code, enum="TftpErrorCode")
+        self.error_handlers: list[TftpErrorHandler] = []
 
         self.endpoint_period = Double(value=REEMIT_PERIOD_S)
         self.env.channel(
@@ -346,10 +349,14 @@ class BaseTftpConnection(UdpConnection):
         """Handle an error message."""
 
         # Update underlying primitive.
-        error_code = self.error_code.from_stream(stream)
-        self.endpoint(addr).handle_error(
-            TftpErrorCode(error_code), stream.read().decode()
-        )
+        error_code = TftpErrorCode(self.error_code.from_stream(stream))
+        message = stream.read().decode()
+
+        # Call extra error handlers.
+        for handler in self.error_handlers:
+            handler(error_code, message, addr)
+
+        self.endpoint(addr).handle_error(error_code, message)
 
     async def process_datagram(
         self, data: bytes, addr: tuple[str, int]

{runtimepy-5.4.3.dist-info → runtimepy-5.4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: runtimepy
-Version: 5.4.3
+Version: 5.4.4
 Summary: A framework for implementing Python services.
 Home-page: https://github.com/vkottler/runtimepy
 Author: Vaughn Kottler
@@ -17,10 +17,10 @@ Classifier: License :: OSI Approved :: MIT License
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
+Requires-Dist: websockets
 Requires-Dist: vcorelib >=3.3.1
 Requires-Dist: svgen >=0.6.8
 Requires-Dist: psutil
-Requires-Dist: websockets
 Provides-Extra: test
 Requires-Dist: pylint ; extra == 'test'
 Requires-Dist: flake8 ; extra == 'test'
@@ -44,11 +44,11 @@ Requires-Dist: uvloop ; (sys_platform != "win32" and sys_platform != "cygwin") a
     =====================================
     generator=datazen
     version=3.1.4
-    hash=76ca13ab011b845de4580fd0bc1f9deb
+    hash=dbac0e58214ea7e4abbe7161a3c64749
     =====================================
 -->
 
-# runtimepy ([5.4.3](https://pypi.org/project/runtimepy/))
+# runtimepy ([5.4.4](https://pypi.org/project/runtimepy/))
 
 [![python](https://img.shields.io/pypi/pyversions/runtimepy.svg)](https://pypi.org/project/runtimepy/)
 ![Build Status](https://github.com/vkottler/runtimepy/workflows/Python%20Package/badge.svg)

{runtimepy-5.4.3.dist-info → runtimepy-5.4.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-runtimepy/__init__.py,sha256=_OzprSplcibI12WvqExDf_PsXnZWlDNJh_jotslRyXw,390
+runtimepy/__init__.py,sha256=baR2RhS2-y7ZXgCnsPoi2qZNQvYUTMTNIsSR2qKnRNg,390
 runtimepy/__main__.py,sha256=OPAed6hggoQdw-6QAR62mqLC-rCkdDhOq0wyeS2vDRI,332
 runtimepy/app.py,sha256=sTvatbsGZ2Hdel36Si_WUbNMtg9CzsJyExr5xjIcxDE,970
 runtimepy/dev_requirements.txt,sha256=j0dh11ztJAzfaUL0iFheGjaZj9ppDzmTkclTT8YKO8c,230
@@ -194,8 +194,8 @@ runtimepy/net/udp/connection.py,sha256=YtLty6iveVH6PQRODKriwR5Q6lFi0Ed4k0o5r0bgA
 runtimepy/net/udp/create.py,sha256=84YDfJbNBlN0ZwbNpvh6Dl7ZPecbZfmpjMNRRWcvJDk,2005
 runtimepy/net/udp/protocol.py,sha256=A4SRHf0CgcL2zDs1nAsGDqz0RxKBy1soS8wtNdS5S0I,1492
 runtimepy/net/udp/queue.py,sha256=DF-YscxQcGbGCYQLz_l_BMaSRfraZOhRwieTEdXLMds,637
-runtimepy/net/udp/tftp/__init__.py,sha256=mpRewPEqKeuFf5yRO1cY5H7fjFJhX6PskpqE7zNP2Sc,8741
-runtimepy/net/udp/tftp/base.py,sha256=FCrtpLdUFkQU4nkn5vJc6Gn5EsZauCf8eLhxx1r_vG8,11276
+runtimepy/net/udp/tftp/__init__.py,sha256=C4YI06y-xwHphYBc68jXL35CrLVNzds3-tIhxTp1PWk,9081
+runtimepy/net/udp/tftp/base.py,sha256=7nEgxv3j6ehqP8q2p769NjnkkMBOw3smt1XDqFBmdN4,11551
 runtimepy/net/udp/tftp/endpoint.py,sha256=so60LdPTG66N5tdhHhiX7j_TBHvNOTi4JIgLcg2MAm0,10890
 runtimepy/net/udp/tftp/enums.py,sha256=06juMd__pJZsyL8zO8p3hRucnOratt1qtz9zcxzMg4s,1579
 runtimepy/net/udp/tftp/io.py,sha256=w6cnUt-T-Ma6Vg8BWoRbsNnIWUv0HTY4am6bcLWxNJs,803
@@ -255,9 +255,9 @@ runtimepy/tui/task.py,sha256=nUZo9fuOC-k1Wpqdzkv9v1tQirCI28fZVgcC13Ijvus,1093
 runtimepy/tui/channels/__init__.py,sha256=evDaiIn-YS9uGhdo8ZGtP9VK1ek6sr_P1nJ9JuSET0o,4536
 runtimepy/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 runtimepy/ui/controls.py,sha256=yvT7h3thbYaitsakcIAJ90EwKzJ4b-jnc6p3UuVf_XE,1241
-runtimepy-5.4.3.dist-info/LICENSE,sha256=okYCYhGsx_BlzvFdoNVBVpw_Cfb4SOqHA_VAARml4Hc,1071
-runtimepy-5.4.3.dist-info/METADATA,sha256=4KcLsxZe_Li3WlDrpo5j6DerKciyPt33S9mMXxrjty8,8152
-runtimepy-5.4.3.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-runtimepy-5.4.3.dist-info/entry_points.txt,sha256=-btVBkYv7ybcopqZ_pRky-bEzu3vhbaG3W3Z7ERBiFE,51
-runtimepy-5.4.3.dist-info/top_level.txt,sha256=0jPmh6yqHyyJJDwEID-LpQly-9kQ3WRMjH7Lix8peLg,10
-runtimepy-5.4.3.dist-info/RECORD,,
+runtimepy-5.4.4.dist-info/LICENSE,sha256=okYCYhGsx_BlzvFdoNVBVpw_Cfb4SOqHA_VAARml4Hc,1071
+runtimepy-5.4.4.dist-info/METADATA,sha256=QwSQWMUYIK5S4grh2n898qjjXI_7gHHMFtU10gqAX7c,8152
+runtimepy-5.4.4.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+runtimepy-5.4.4.dist-info/entry_points.txt,sha256=-btVBkYv7ybcopqZ_pRky-bEzu3vhbaG3W3Z7ERBiFE,51
+runtimepy-5.4.4.dist-info/top_level.txt,sha256=0jPmh6yqHyyJJDwEID-LpQly-9kQ3WRMjH7Lix8peLg,10
+runtimepy-5.4.4.dist-info/RECORD,,