Making main_window optional

So we can run test functions without a main window

Author: oren-z0

electrum/plugins/timelock_recovery/qt.py

@@ -49,6 +49,7 @@
 if TYPE_CHECKING:
     from electrum.gui.qt import ElectrumGui
     from electrum.transaction import PartialTransaction
+    from electrum.gui.qt.main_window import ElectrumWindow
     from PyQt6.QtWidgets import QStatusBar
 
 
@@ -120,7 +121,9 @@ def requires_settings(self) -> bool:
         return False
 
     def setup_dialog(self, status_bar: 'QStatusBar') -> bool:
-        context = TimelockRecoveryContext(status_bar.parent())
+        main_window: 'ElectrumWindow' = status_bar.parent()
+        context = TimelockRecoveryContext(main_window.wallet)
+        context.main_window = main_window
 
         if constants.net.NET_NAME == 'regtest':
             return self.create_plan_dialog(context)

electrum/plugins/timelock_recovery/timelock_recovery.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import TYPE_CHECKING, List, Optional, Tuple
+from typing import TYPE_CHECKING, Callable, List, Optional, Sequence, Tuple
 from electrum.bitcoin import address_to_script
 from electrum.plugin import BasePlugin
 from electrum.transaction import PartialTxOutput, PartialTxInput, TxOutpoint
@@ -29,9 +29,9 @@ def nsequence(self, value: int):
         pass # ignore override attempts
 
 class TimelockRecoveryContext:
-    main_window: 'ElectrumWindow'
     wallet: 'Abstract_Wallet'
     wallet_name: str
+    main_window: Optional['ElectrumWindow'] = None
     timelock_days: Optional[int] = None
     cancellation_address: Optional[str] = None
     outputs: Optional[List['PartialTxOutput']] = None
@@ -48,9 +48,8 @@ class TimelockRecoveryContext:
 
     ANCHOR_OUTPUT_AMOUNT_SATS = 600
 
-    def __init__(self, main_window: 'ElectrumWindow'):
-        self.main_window = main_window
-        self.wallet = main_window.wallet
+    def __init__(self, wallet: 'Abstract_Wallet'):
+        self.wallet = wallet
         self.wallet_name = str(self.wallet)
 
     def _get_address_by_label(self, label: str) -> str:
@@ -88,7 +87,7 @@ def make_unsigned_alert_tx(self, fee_est, *, confirmed_only=False) -> 'PartialTr
                 for output in self.outputs
             ]
         return self.wallet.make_unsigned_transaction(
-            coins=self.main_window.get_coins(confirmed_only=confirmed_only),
+            coins=self.wallet.get_spendable_coins(confirmed_only=confirmed_only),
             outputs=self._alert_tx_outputs,
             fee=fee_est,
             is_sweep=False,