Handle unconfirmed tx in wallet_fetch_history()

jmclient/jmclient/wallet_utils.py

@@ -596,7 +596,7 @@ def wallet_fetch_history(wallet, options):
     con.row_factory = dict_factory
     tx_db = con.cursor()
     tx_db.execute("CREATE TABLE transactions(txid TEXT, "
-            "blockhash TEXT, blocktime INTEGER);")
+            "blockhash TEXT, blocktime INTEGER, conflicts INTEGER);")
     jm_single().debug_silence[0] = True
     wallet_name = jm_single().bc_interface.get_wallet_name(wallet)
     buf = range(1000)
@@ -605,15 +605,24 @@ def wallet_fetch_history(wallet, options):
         buf = jm_single().bc_interface.rpc('listtransactions', ["*",
             1000, t, True])
         t += len(buf)
-        tx_data = ((tx['txid'], tx['blockhash'], tx['blocktime']) for tx
+        # confirmed
+        tx_data = ((tx['txid'], tx['blockhash'], tx['blocktime'], 0) for tx
                 in buf if 'txid' in tx and 'blockhash' in tx and 'blocktime'
                 in tx)
-        tx_db.executemany('INSERT INTO transactions VALUES(?, ?, ?);',
+        tx_db.executemany('INSERT INTO transactions VALUES(?, ?, ?, ?);',
                 tx_data)
+        # unconfirmed
+        uc_tx_data = ((tx['txid'], None, None, len(tx['walletconflicts'])) for
+                tx in buf if 'txid' in tx and 'blockhash' not in tx and
+                'blocktme' not in tx)
+        tx_db.executemany('INSERT INTO transactions VALUES(?, ?, ?, ?);',
+                uc_tx_data)
 
     txes = tx_db.execute(
         'SELECT DISTINCT txid, blockhash, blocktime '
-        'FROM transactions ORDER BY blocktime').fetchall()
+        'FROM transactions '
+        'WHERE (blockhash IS NOT NULL AND blocktime IS NOT NULL) OR conflicts = 0 '
+        'ORDER BY blocktime').fetchall()
     wallet_script_set = set(wallet.get_script_path(p)
                             for p in wallet.yield_known_paths())
 
@@ -647,7 +656,9 @@ def wallet_fetch_history(wallet, options):
     if options.verbosity > 0: jmprint(s().join(field_names), "info")
     if options.verbosity <= 2: cj_batch = [0]*8 + [[]]*2
     balance = 0
+    unconfirmed_balance = 0
     utxo_count = 0
+    unconfirmed_utxo_count = 0
     deposits = []
     deposit_times = []
     tx_number = 0
@@ -655,6 +666,10 @@ def wallet_fetch_history(wallet, options):
         is_coinjoin, cj_amount, cj_n, output_script_values, blocktime, txd =\
             get_tx_info(tx['txid'])
 
+        # unconfirmed transactions don't have blocktime, get_tx_info() returns
+        # 0 in that case
+        is_confirmed = (blocktime != 0)
+
         our_output_scripts = wallet_script_set.intersection(
             output_script_values.keys())
 
@@ -740,8 +755,11 @@ def wallet_fetch_history(wallet, options):
             tx_type = 'unknown type'
             jmprint('our utxos: ' + str(len(our_input_scripts)) \
                   + ' in, ' + str(len(our_output_scripts)) + ' out')
+
+        if is_confirmed:
             balance += delta_balance
             utxo_count += (len(our_output_scripts) - utxos_consumed)
+
             index = '%4d'%(tx_number)
             tx_number += 1
             if options.verbosity > 0:
@@ -780,6 +798,10 @@ def wallet_fetch_history(wallet, options):
                 deposits.append(delta_balance)
                 deposit_times.append(blocktime)
 
+        else:
+            unconfirmed_balance += delta_balance
+            utxo_count += (len(our_output_scripts) - utxos_consumed)
+
     # we could have a leftover batch!
     if options.verbosity <= 2:
         n = cj_batch[0]
@@ -824,15 +846,19 @@ def wallet_fetch_history(wallet, options):
     # includes disabled utxos in accounting:
     total_wallet_balance = sum(wallet.get_balance_by_mixdepth(
         include_disabled=True).values())
-    if balance != total_wallet_balance:
+    if balance + unconfirmed_balance != total_wallet_balance:
         jmprint(('BUG ERROR: wallet balance (%s) does not match balance from ' +
             'history (%s)') % (sat_to_str(total_wallet_balance),
                 sat_to_str(balance)))
     wallet_utxo_count = sum(map(len, wallet.get_utxos_by_mixdepth_(
         include_disabled=True).values()))
-    if utxo_count != wallet_utxo_count:
+    if utxo_count + unconfirmed_utxo_count != wallet_utxo_count:
         jmprint(('BUG ERROR: wallet utxo count (%d) does not match utxo count from ' +
             'history (%s)') % (wallet_utxo_count, utxo_count))
+
+    if unconfirmed_balance != 0:
+        jmprint('unconfirmed balance change = %s BTC' % sat_to_str(unconfirmed_balance))
+
     # wallet-tool.py prints return value, so return empty string instead of None here
     return ''