bitcoinwatcher 1.0.1__tar.gz → 1.2.0__tar.gz

{bitcoinwatcher-1.0.1 → bitcoinwatcher-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: bitcoinwatcher
-Version: 1.0.1
+Version: 1.2.0
 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
@@ -227,6 +227,7 @@ Requires-Dist: six==1.16.0
 Requires-Dist: urllib3==2.2.1
 Requires-Dist: bitcoinlib==0.6.14
 Requires-Dist: ordipool==1.2.0
+Requires-Dist: python-bitcoinrpc==1.0
 
 # bitcoinwatcher
 

bitcoinwatcher-1.2.0/bitcoin/address_listener/address_listener.py

@@ -0,0 +1,51 @@
+import os
+from abc import ABC, abstractmethod
+
+from bitcoinlib.transactions import Transaction
+from ordipool.ordipool.mempoolio import Mempool
+
+from bitcoin.models.address_tx_data import AddressTxData
+from bitcoin.tx_listener.abstract_tx_listener import AbstractTxListener
+from bitcoin.utils.constants import default_host
+from bitcoin.utils.tx_address_data_extractor import DefaultTxAddressDataExtractor, MempoolTxAddressDataExtractor
+
+
+class AbstractAddressListener(AbstractTxListener, ABC):
+    DECIMAL_SCALE = 5
+    host = os.environ.get("RPC_HOST", default_host)
+    base_url = f"http://{host}:3006/api"
+    mempool = Mempool(base_url=base_url)
+    default_tx_extractor = DefaultTxAddressDataExtractor()
+    mempool_tx_extractor = MempoolTxAddressDataExtractor(mempool)
+
+    def __init__(self, addresses_to_listen: {str}):
+        self.addresses_to_listen = addresses_to_listen
+
+    @abstractmethod
+    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(set((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
+        if tx.coinbase:
+            return
+        address_tx_data = self.default_tx_extractor.extract(tx)
+        # filter the address we are interested in
+        addresses_for_events = self.filter_address_tx_data(address_tx_data)
+        if len(addresses_for_events) == 0:
+            return
+        # get the address tx data from mempool for full details if any address matches
+        try:
+            address_tx_data = self.mempool_tx_extractor.extract(tx)
+        except Exception as e:
+            print(f"Error in getting mempool tx data, taking defaults: {e}")
+            address_tx_data = address_tx_data
+
+        for address in addresses_for_events:
+            self.consume(subscribed_address=address, address_tx_data=address_tx_data)

{bitcoinwatcher-1.0.1 → bitcoinwatcher-1.2.0}/bitcoin/address_listener/simple_address_listener.py

@@ -1,4 +1,5 @@
-from bitcoin.address_listener.address_listener import AbstractAddressListener, AddressTxData, AddressTxType
+from bitcoin.address_listener.address_listener import AbstractAddressListener, AddressTxData
+from bitcoin.models.address_tx_data import AddressTxType
 from bitcoin.tx_listener.zmq_listener import ZMQTXListener
 
 
@@ -11,6 +12,7 @@ class SimpleAddressListener(AbstractAddressListener):
                                 address_tx_data))
         total_amount_in_input = sum(map(lambda x: x.amount_in_btc(), all_input))
         # scale ito 4 decimal places
+        print("Transaction status: ", address_tx_data[0].tx_status)
         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}")
         print(f"Address {subscribed_address} spent {total_amount_in_input} BTC in tx {address_tx_data[0].tx_id}")

bitcoinwatcher-1.2.0/bitcoin/models/address_tx_data.py

@@ -0,0 +1,36 @@
+import dataclasses
+from enum import Enum
+
+
+class AddressTxType(Enum):
+    INPUT = "input"
+    OUTPUT = "output"
+
+
+class AddressTxSource(Enum):
+    MEMPOOL = "mempool"
+    DEFAULT = "default"
+
+
+class TransactionStatus(Enum):
+    UNCONFIRMED = "unconfirmed"
+    CONFIRMED = "confirmed"
+
+
+@dataclasses.dataclass
+class AddressTxData:
+    tx_id: str
+    address: str
+    type: AddressTxType
+    # figure out the way to get amount in vin
+    _amount: int = 0
+    tx_status: str = "unconfirmed"
+    is_reveal_tx: bool = False
+
+    def get_amount(self):
+        return self._amount
+
+    def amount_in_btc(self):
+        if self.type == AddressTxType.INPUT:
+            return "Amount is not supported for input type, yet"
+        return self._amount / 100000000