bitcoinwatcher 1.0.1__py3-none-any.whl → 1.2.0__py3-none-any.whl

bitcoin/tests/test_address_listener.py

@@ -1,10 +1,12 @@
 from unittest import TestCase
 
-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
 
 
 class DummyAddressListener(AbstractAddressListener):
-    def consume(self, address_tx_data):
+
+    def consume(self, subscribed_address, address_tx_data: [AddressTxData]):
         pass
 
     def __init__(self, address_to_listen: [str]):

bitcoin/tests/test_bitcoin_utils.py

@@ -0,0 +1,32 @@
+from unittest import TestCase
+
+from bitcoinlib.transactions import Transaction
+
+from bitcoin.tests.data.transactions import batch_airdrop_tx_hex, tx_hex, non_reveal_tx_hex
+from bitcoin.utils.bitcoin_utils import is_reveal_tx, count_reveal_inputs
+
+
+class Test(TestCase):
+    def test_is_reveal_tx(self):
+        tx = Transaction.parse_hex(tx_hex, strict=False)
+        self.assertTrue(is_reveal_tx(tx))
+
+    def test_is_reveal_tx_in_second_input(self):
+        tx = Transaction.parse_hex(batch_airdrop_tx_hex, strict=False)
+        self.assertTrue(is_reveal_tx(tx))
+
+    def test_is_not_reveal_tx(self):
+        tx = Transaction.parse_hex(non_reveal_tx_hex, strict=False)
+        self.assertFalse(is_reveal_tx(tx))
+
+    def test_count_reveal_inputs(self):
+        tx = Transaction.parse_hex(tx_hex, strict=False)
+        self.assertEqual(count_reveal_inputs(tx), 1)
+
+    def test_count_reveal_inputs_not_reveal(self):
+        tx = Transaction.parse_hex(non_reveal_tx_hex, strict=False)
+        self.assertEqual(count_reveal_inputs(tx), 0)
+
+    def test_count_reveal_in_airdrop_tx(self):
+        tx = Transaction.parse_hex(batch_airdrop_tx_hex, strict=False)
+        self.assertEqual(count_reveal_inputs(tx), 1420)

bitcoin/utils/benchmark.py

@@ -0,0 +1,33 @@
+# benchmark between mempool and bitcoinrcp get transaction info
+import os
+import time
+import requests
+from ordipool.ordipool.mempoolio import Mempool
+
+from bitcoin.utils.bitcoin_rpc import BitcoinRPC
+from bitcoin.utils.constants import default_host
+
+host = os.environ.get("RPC_HOST", default_host)
+base_url = f"http://{host}:3006/api"
+
+mempool = Mempool(base_url)
+bitcoinrpc = BitcoinRPC()
+
+if __name__ == '__main__':
+
+    txid = 'a6293e898b056fbea0329d071b1b237e4449ff464cdbc7a9ed8a770b97aafd4c'
+    times = 1000
+    start = time.time()
+    print("mempool starting")
+    for i in range(times):
+        print(i)
+        mempool.get_transaction(txid)
+    end = time.time()
+    print(f"mempool took: {end - start}")
+
+    start = time.time()
+    for i in range(times):
+        bitcoinrpc.get_transaction(txid)
+    end = time.time()
+    print(f"bitcoinrpc took: {end - start}")
+

bitcoin/utils/bitcoin_rpc.py

@@ -0,0 +1,31 @@
+import os
+
+from bitcoinrpc.authproxy import AuthServiceProxy, JSONRPCException
+
+from bitcoin.utils.constants import default_host
+
+
+class BitcoinRPC:
+    rpc_user = os.environ.get("RPC_USER")
+    rpc_password = os.environ.get("RPC_PASSWORD")
+    rpc_host = os.environ.get("RPC_HOST", default_host)
+    rpc_port = os.environ.get("RPC_PORT", 8332)
+    rpc_connection: AuthServiceProxy
+
+    def __init__(self):
+        rpc_string = f"http://{self.rpc_user}:{self.rpc_password}@{self.rpc_host}:{self.rpc_port}"
+        print("rpc_string: ", rpc_string)
+        self.rpc_connection = AuthServiceProxy(rpc_string)
+
+    def get_transaction(self, txid: str) -> dict:
+        return self.rpc_connection.getrawtransaction(txid, True)
+
+
+if __name__ == '__main__':
+    rpc = BitcoinRPC()
+    txid = 'a6293e898b056fbea0329d071b1b237e4449ff464cdbc7a9ed8a770b97aafd4c'
+    try:
+        transaction = rpc.get_transaction(txid)
+        print(transaction)
+    except JSONRPCException as e:
+        print(f"An error occurred: {e}")

bitcoin/utils/bitcoin_utils.py

@@ -1,16 +1,18 @@
-from bitcoin.core import CTransaction
-from bitcoinutils.keys import PublicKey
-from bitcoinutils.transactions import TxOutput
+from bitcoinlib.transactions import Transaction, Input
+from ordipool.utils.utils import convert_to_hex
 
 
-def get_tx_id_from_tx(tx: CTransaction) -> str:
-    tx_id = tx.GetTxid()
-    # convert it to hex
-    txid_hex = tx_id[::-1].hex()
-    return txid_hex
+def count_inscriptions(vin: Input) -> int:
+    ord_hex = convert_to_hex('ord')
+    # count occurences of ord in the witness script
+    return sum([witness.hex().count(ord_hex) for witness in vin.witnesses])
 
 
-def get_address_from_output(output: TxOutput):
-    public_key = output.script_pubkey.to_hex()
-    address = PublicKey(public_key)
-    print(address)
+def count_reveal_inputs(tx: Transaction) -> int:
+    return sum([count_inscriptions(vin) for vin in tx.inputs])
+
+
+def is_reveal_tx(tx: Transaction) -> bool:
+    count_of_inscriptions = count_reveal_inputs(tx)
+    return count_of_inscriptions > 0
+    # check if any of the witness contains ord, convert witness to hex

bitcoin/utils/tx_address_data_extractor.py

@@ -0,0 +1,76 @@
+from abc import ABC, abstractmethod
+
+
+from bitcoinlib.transactions import Transaction
+from ordipool.ordipool.mempoolio import Mempool
+
+from bitcoin.address_listener.address_listener import AddressTxData
+from bitcoin.models.address_tx_data import AddressTxType, TransactionStatus
+from bitcoin.utils.bitcoin_utils import is_reveal_tx, count_reveal_inputs
+
+
+class AbstractTxAddressDataExtractor(ABC):
+    @abstractmethod
+    def extract(self, tx: Transaction) -> [AddressTxData]:
+        pass
+
+
+class MempoolTxAddressDataExtractor(AbstractTxAddressDataExtractor):
+    mempool: Mempool
+
+    def __init__(self, mempool: Mempool):
+        self.mempool = mempool
+
+    def extract(self, tx: Transaction) -> [AddressTxData]:
+        outputs = tx.outputs
+        tx_id = tx.txid
+        has_inscriptions = is_reveal_tx(tx)
+
+        tx = self.mempool.get_transaction(tx_id)
+        tx_status = TransactionStatus.CONFIRMED if tx.confirmed else TransactionStatus.UNCONFIRMED
+        address_tx_data = []
+        # getting inputs data separately from mempool
+        # as current library doesn't provide rich data like previous outputs and its value
+        for vin in tx.vins:
+            address = vin.prev_out.address
+            amount = vin.prev_out.value
+            address_tx_data.append(AddressTxData(tx_status=tx_status.value,
+                                                 address=address,
+                                                 is_reveal_tx=has_inscriptions,
+                                                 type=AddressTxType.INPUT,
+                                                 _amount=amount,
+                                                 tx_id=tx_id))
+        for output in outputs:
+            amount = output.value
+            address_tx_data.append(AddressTxData(tx_status=tx_status.value,
+                                                 address=output.address,
+                                                 _amount=amount,
+                                                 is_reveal_tx=has_inscriptions,
+                                                 type=AddressTxType.OUTPUT,
+                                                 tx_id=tx_id))
+
+        return address_tx_data
+
+
+class DefaultTxAddressDataExtractor(AbstractTxAddressDataExtractor):
+    def extract(self, tx: Transaction) -> [AddressTxData]:
+        outputs = tx.outputs
+        address_tx_data = []
+        inputs = tx.inputs
+        tx_status = tx.status
+        for input in inputs:
+            address = input.address
+            amount = 0
+            address_tx_data.append(AddressTxData(tx_status=tx_status,
+                                                 address=address,
+                                                 type=AddressTxType.INPUT,
+                                                 _amount=amount,
+                                                 tx_id=tx.txid))
+        for output in outputs:
+            amount = output.value
+            address_tx_data.append(AddressTxData(tx_status=tx_status,
+                                                 address=output.address,
+                                                 _amount=amount,
+                                                 type=AddressTxType.OUTPUT,
+                                                 tx_id=tx.txid))
+        return address_tx_data

{bitcoinwatcher-1.0.1.dist-info → bitcoinwatcher-1.2.0.dist-info}/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,26 @@
+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=sK0hrhPrYg8z6AC8MqNbF3UoaqvbhdhVcoyMLj8x4MQ,2223
+bitcoin/address_listener/simple_address_listener.py,sha256=iSMPBUzl9d4s8_ee2LAXz83zBiGfs1KhwSH_7FvxwKY,1609
+bitcoin/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bitcoin/models/address_tx_data.py,sha256=0zIleuEHern42zk1Pp7IteR5jQqVdAM1l4Gvy6YzGe0,754
+bitcoin/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bitcoin/tests/test_address_listener.py,sha256=oqVfJ_lQNHgRxCnDteUpkaUyHff5Wa8J2oYJnwQ-QaE,3113
+bitcoin/tests/test_bitcoin_utils.py,sha256=NYRHWfqaOgLO8osOYVqRudk9ksuhC3LC9HGrxj-VGu4,1213
+bitcoin/tests/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bitcoin/tests/data/transactions.py,sha256=To0FCF4JDdjZTV9H900WCuV8_mGPItk5bEiYgUj0KiY,394877
+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=_edaee2PYj9T1jxLenv68ekZRTM8-LUN3DloDLNQ5Bc,999
+bitcoin/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bitcoin/utils/benchmark.py,sha256=ZCfhIe4_meUNpQ7NEioZj3_uudCisg7J8d_LTMoRFto,875
+bitcoin/utils/bitcoin_rpc.py,sha256=Uq3eZnLaGOgltybEWeBRubnUhTHyB8gVI_L_SnrZ-ug,1012
+bitcoin/utils/bitcoin_utils.py,sha256=mrnRPqUa2U2EMKu7rrPV_bW1sL2CJUfbAom0Zdamydk,631
+bitcoin/utils/constants.py,sha256=irZLlArgica2VckyckEYxH5D5KjvdF52dtBMWswqw8k,52
+bitcoin/utils/inscription_utils.py,sha256=8QbOJ1o1n1bMFsPREGLzwFjnGzfuARgJCPr6ORhP44o,193
+bitcoin/utils/tx_address_data_extractor.py,sha256=M_WvBNNXYwd2qBG6oyfu8Vn6P8F0Rt5w1FnfvFO6IZc,3321
+bitcoinwatcher-1.2.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+bitcoinwatcher-1.2.0.dist-info/METADATA,sha256=i4U1BiZn-KQ9TQDPKGaKkctAFpgx1lRGaLywjvz5qoc,14940
+bitcoinwatcher-1.2.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+bitcoinwatcher-1.2.0.dist-info/top_level.txt,sha256=YdUgzLdCiMlrwaKyDqHA1acEd23QFko5bv7D6nBANJ0,8
+bitcoinwatcher-1.2.0.dist-info/RECORD,,

bitcoinwatcher-1.0.1.dist-info/RECORD

@@ -1,18 +0,0 @@
-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=WKDG8gtB9_2K-v0nYhkKDkaqxPz2uE2uVtwWRNzg7H8,3000
-bitcoin/address_listener/simple_address_listener.py,sha256=2qR012t3rHTNHkit6TvkYP9Co4IVySKG8YyzmxKc0cU,1499
-bitcoin/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bitcoin/tests/test_address_listener.py,sha256=_8ic7ldHaeICfYTepAkqcWFzZPhEKzbIkAfrAZ1ZdCs,3033
-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=_edaee2PYj9T1jxLenv68ekZRTM8-LUN3DloDLNQ5Bc,999
-bitcoin/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bitcoin/utils/bitcoin_utils.py,sha256=ehzfwiuln_lPia6urtUSVE3zXC1zyZEKiNM36e7N00k,428
-bitcoin/utils/constants.py,sha256=irZLlArgica2VckyckEYxH5D5KjvdF52dtBMWswqw8k,52
-bitcoin/utils/inscription_utils.py,sha256=8QbOJ1o1n1bMFsPREGLzwFjnGzfuARgJCPr6ORhP44o,193
-bitcoinwatcher-1.0.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-bitcoinwatcher-1.0.1.dist-info/METADATA,sha256=qpqrssZZ60vtfPxx3vziTyZ5krWFVQz1Y86PYs8JmH4,14901
-bitcoinwatcher-1.0.1.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-bitcoinwatcher-1.0.1.dist-info/top_level.txt,sha256=YdUgzLdCiMlrwaKyDqHA1acEd23QFko5bv7D6nBANJ0,8
-bitcoinwatcher-1.0.1.dist-info/RECORD,,