distributed-state-network 0.6.0__tar.gz → 0.6.2__tar.gz

{distributed_state_network-0.6.0 → distributed_state_network-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: distributed-state-network
-Version: 0.6.0
+Version: 0.6.2
 Summary: A tool to distribute the state of a network device to other devices on the network
 Project-URL: Homepage, https://github.com/erinclemmer/distributed_state_network
 Project-URL: Issues, https://github.com/erinclemmer/distributed_state_network/issues

{distributed_state_network-0.6.0 → distributed_state_network-0.6.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "distributed-state-network"
-version = "0.6.0"
+version = "0.6.2"
 authors = [{name="Erin Clemmer", email="erin.c.clemmer@gmail.com"}]
 description = "A tool to distribute the state of a network device to other devices on the network"
 keywords = ["distributed", "networking"]

{distributed_state_network-0.6.0 → distributed_state_network-0.6.2}/src/distributed_state_network/dsnode.py

@@ -4,7 +4,6 @@ import random
 import logging
 import threading
 import requests
-import tracing
 from typing import Dict, List, Optional, Callable
 
 from distributed_state_network.objects.endpoint import Endpoint
@@ -405,9 +404,10 @@ class DSNode:
         if not pkt.verify_signature(self.cred_manager.read_public(pkt.node_id)):
             raise Exception(401, "Could not verify ECDSA signature of data packet")
         
-        try:
-            self.receive_cb(data)
-        except Exception as e:
-            traceback.print_exc(e)
+        if self.receive_cb is not None:
+            try:
+                self.receive_cb(data)
+            except Exception as e:
+                traceback.print_exc(e)
 
         return b'OK'

{distributed_state_network-0.6.0 → distributed_state_network-0.6.2}/src/distributed_state_network/handler.py

@@ -8,7 +8,7 @@ from distributed_state_network.objects.config import DSNodeConfig
 from distributed_state_network.util.aes import generate_aes_key
 from distributed_state_network.util import stop_thread
 
-VERSION = "0.6.0"
+VERSION = "0.6.2"
 logging.basicConfig(level=logging.INFO)
 
 # Silence Flask and Werkzeug logging
@@ -172,3 +172,12 @@ class DSNodeServer:
     
     def node_id(self) -> str:
         return self.config.node_id
+
+    def set_receive_cb(self, cb: Callable):
+        self.node.receive_cb = cb
+
+    def set_update_cb(self, cb: Callable):
+        self.node.update_cb = cb
+
+    def set_disconnect_cb(self, cb: Callable):
+        self.node.disconnect_cb = cb