bitcoinwatcher 0.1.3__py3-none-any.whl → 0.1.6__py3-none-any.whl

bitcoin/address_listener/address_listener.py

@@ -32,13 +32,21 @@ class AddressTxData:
 
 
 class AbstractAddressListener(AbstractTxListener, ABC):
-    def __init__(self, address_to_listen: [str]):
-        self.address_to_listen = address_to_listen
+    DECIMAL_SCALE = 5
+
+    def __init__(self, addresses_to_listen: {str}):
+        self.addresses_to_listen = addresses_to_listen
 
     @abstractmethod
-    def consume(self, address_tx_data: [AddressTxData]):
+    def consume(self, subscribed_address, address_tx_data: [AddressTxData]):
         pass
 
+    def filter_address_tx_data(self, address_tx_data: [AddressTxData]) -> [str]:
+        filtered_address_tx_data = list(filter(lambda x: x.address in self.addresses_to_listen and x.address != "",
+                                               address_tx_data))
+        # get all address
+        return list(map(lambda x: x.address, filtered_address_tx_data))
+
     def on_tx(self, tx: Transaction):
         # get all address in the inputs and outputs along with the amount
         inputs = tx.inputs
@@ -58,5 +66,6 @@ class AbstractAddressListener(AbstractTxListener, ABC):
                                                  tx_id=tx_id))
 
         # filter the address we are interested in
-        address_tx_data = list(filter(lambda x: x.address in self.address_to_listen, address_tx_data))
-        self.consume(address_tx_data)
+        addresses_for_events = self.filter_address_tx_data(address_tx_data)
+        for address in addresses_for_events:
+            self.consume(subscribed_address=address, address_tx_data=address_tx_data)

bitcoin/address_listener/simple_address_listener.py

@@ -1,13 +1,23 @@
 from bitcoin.address_listener.address_listener import AbstractAddressListener, AddressTxData, AddressTxType
+from bitcoin.tx_listener.zmq_listener import ZMQTXListener
 
 
 class SimpleAddressListener(AbstractAddressListener):
-    def consume(self, address_tx_data: [AddressTxData]):
-        for address_tx in address_tx_data:
-            if address_tx.type == AddressTxType.OUTPUT:
-                print(
-                    f"Address {address_tx.address} received {address_tx.amount_in_btc()} BTC in tx {address_tx.tx_id}")
-
-    def __init__(self, address):
-        self.address = address
-        super().__init__(address_to_listen=[address])
+    def consume(self, subscribed_address, address_tx_data: [AddressTxData]):
+        all_output = list(filter(lambda x: x.type == AddressTxType.OUTPUT and x.address == subscribed_address,
+                                 address_tx_data))
+        total_amount_in_output = sum(map(lambda x: x.amount_in_btc(), all_output))
+        # scale ito 4 decimal places
+        total_amount_in_output = round(total_amount_in_output, self.DECIMAL_SCALE)
+        print(f"Address {subscribed_address} received {total_amount_in_output} BTC in tx {address_tx_data[0].tx_id}")
+
+    def __init__(self, addresses_to_listen: {str}):
+        self.addresses = address
+        super().__init__(addresses_to_listen=addresses_to_listen)
+
+
+if __name__ == '__main__':
+    address = ["3P4WqXDbSLRhzo2H6MT6YFbvBKBDPLbVtQ"]
+    address_watcher = SimpleAddressListener(address)
+    zmq_listener = ZMQTXListener(address_watcher)
+    zmq_listener.start()

bitcoin/tests/test_address_listener.py

@@ -0,0 +1,36 @@
+from unittest import TestCase
+
+from bitcoin.address_listener.address_listener import AbstractAddressListener, AddressTxData, AddressTxType
+
+
+class DummyAddressListener(AbstractAddressListener):
+    def consume(self, address_tx_data):
+        pass
+
+    def __init__(self, address_to_listen: [str]):
+        super().__init__(address_to_listen)
+
+
+class TestAbstractAddressListener(TestCase):
+    def test_filter_address_tx_data_one_address(self):
+        address_listener = DummyAddressListener("3P4WqXDbSL")
+        input_tx_data = [AddressTxData(tx_id="tx id", address="3P4WqXDbSL", type= AddressTxType.OUTPUT, _amount=1000)]
+        address_tx_data = address_listener.filter_address_tx_data(input_tx_data)
+        self.assertEqual(len(address_tx_data), len(input_tx_data))
+
+    def test_filter_address_tx_data_multiple_addresses(self):
+        address_listener = DummyAddressListener(["3P4WqXDbSL", "3P4WqXDbSL2"])
+        input_tx_data = [AddressTxData(tx_id="tx id", address="3P4WqXDbSL", type= AddressTxType.OUTPUT, _amount=1000),
+                         AddressTxData(tx_id="tx id", address="3P4WqXDbSL2", type= AddressTxType.OUTPUT, _amount=1000),
+                            AddressTxData(tx_id="tx id", address="3P4WqXDbSL3", type= AddressTxType.OUTPUT, _amount=1000)]
+        address_tx_data = address_listener.filter_address_tx_data(input_tx_data)
+        self.assertEqual(len(address_tx_data), 2)
+        self.assertEqual(address_tx_data[0], "3P4WqXDbSL")
+        self.assertEqual(address_tx_data[1], "3P4WqXDbSL2")
+        self.assertNotEquals(len(address_tx_data), len(input_tx_data))
+
+    def test_filter_address_tx_data_empty_address_in_input(self):
+        address_listener = DummyAddressListener("3P4WqXDbSL")
+        input_tx_data = [AddressTxData(tx_id="tx id", address="", type= AddressTxType.OUTPUT, _amount=1000)]
+        address_tx_data = address_listener.filter_address_tx_data(input_tx_data)
+        self.assertEqual(len(address_tx_data), 0)

{bitcoinwatcher-0.1.3.dist-info → bitcoinwatcher-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: bitcoinwatcher
-Version: 0.1.3
+Version: 0.1.6
 Summary: bitcoinwatcher is a Python library that implements a ZMQ subscriber and provides abstractions to build custom address watchers. This library is designed to make it easy for developers to monitor Bitcoin addresses and react to changes in their state.
 Author: twosatsmaxi
 License: Apache License
@@ -221,8 +221,6 @@ Requires-Dist: requests ~=2.31.0
 Requires-Dist: certifi ==2024.2.2
 Requires-Dist: charset-normalizer ==3.3.2
 Requires-Dist: idna ==3.6
-Requires-Dist: ordipool ==1.1.1
-Requires-Dist: PyGObject ==3.46.0
 Requires-Dist: pyzmq ==25.1.2
 Requires-Dist: requests ==2.31.0
 Requires-Dist: six ==1.16.0

{bitcoinwatcher-0.1.3.dist-info → bitcoinwatcher-0.1.6.dist-info}/RECORD

@@ -1,16 +1,17 @@
 bitcoin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bitcoin/address_listener/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bitcoin/address_listener/address_listener.py,sha256=Eeo7Q0pqjMzX-TYkJr6YjWPe_iSqh7xWMArX2yFwACY,2059
-bitcoin/address_listener/simple_address_listener.py,sha256=9PKR5cylYD0jivKElvLSDWs14k50hy_cKmLPe4uz0TY,582
+bitcoin/address_listener/address_listener.py,sha256=Tn-oCZrFaGdI8k9F-t3hPIKdbAV95tUysY8s7MPiDQs,2535
+bitcoin/address_listener/simple_address_listener.py,sha256=fHF2HH_c1-ioQMhs5QauHyGT-eDjyxGl8hTeB39_yIs,1145
 bitcoin/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bitcoin/tests/test_address_listener.py,sha256=K9ltdDGoz60KoDimWQ4tXuQ8KxPJDkQOvHnHUtXPh6w,1966
 bitcoin/tx_listener/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bitcoin/tx_listener/abstract_tx_listener.py,sha256=rGp9na-aOSl_aPaIqf44RTZaFF3SvrxIhZBujTnWqao,189
 bitcoin/tx_listener/zmq_listener.py,sha256=rKqYMPZOHq-WR0cL_Ucmwr0AfIpHbsLvOxfTpBjOyI4,990
 bitcoin/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bitcoin/utils/bitcoin_utils.py,sha256=ehzfwiuln_lPia6urtUSVE3zXC1zyZEKiNM36e7N00k,428
 bitcoin/utils/inscription_utils.py,sha256=8QbOJ1o1n1bMFsPREGLzwFjnGzfuARgJCPr6ORhP44o,193
-bitcoinwatcher-0.1.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-bitcoinwatcher-0.1.3.dist-info/METADATA,sha256=ISQYM1YVyR0Y_D3p38DXh2mbs2SP91PEC7oG_5YEios,14935
-bitcoinwatcher-0.1.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-bitcoinwatcher-0.1.3.dist-info/top_level.txt,sha256=YdUgzLdCiMlrwaKyDqHA1acEd23QFko5bv7D6nBANJ0,8
-bitcoinwatcher-0.1.3.dist-info/RECORD,,
+bitcoinwatcher-0.1.6.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+bitcoinwatcher-0.1.6.dist-info/METADATA,sha256=9Rvmep2jqL6OMv5T2a1FayompiGbmyFLihrkl0xQ8cM,14869
+bitcoinwatcher-0.1.6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+bitcoinwatcher-0.1.6.dist-info/top_level.txt,sha256=YdUgzLdCiMlrwaKyDqHA1acEd23QFko5bv7D6nBANJ0,8
+bitcoinwatcher-0.1.6.dist-info/RECORD,,