From 3379e462498cee56b909f2e1a9a8cdfdabdae4e5 Mon Sep 17 00:00:00 2001 From: SomberNight Date: Thu, 9 Oct 2025 14:57:41 +0000 Subject: [PATCH] follow-up prev: add type-hints, etc --- electrum/transaction.py | 12 ++++++------ electrum/txbatcher.py | 2 +- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/electrum/transaction.py b/electrum/transaction.py index eed3820e7..cca06ff18 100644 --- a/electrum/transaction.py +++ b/electrum/transaction.py @@ -2420,19 +2420,19 @@ class PartialTransaction(Transaction): self.add_outputs([funding_output]) delattr(self, '_script_to_output_idx') - def get_change_outputs(self): - return [o for o in self._outputs if o.is_change] + def get_change_outputs(self) -> Sequence[PartialTxOutput]: + return [o for o in self._outputs if o.is_change] - def has_change(self): + def has_change(self) -> bool: return len(self.get_change_outputs()) > 0 def get_dummy_output(self, dummy_addr: str) -> Optional['PartialTxOutput']: idxs = self.get_output_idxs_from_address(dummy_addr) if not idxs: - return + return None assert len(idxs) == 1 - for i in idxs: - return self.outputs()[i] + idx = list(idxs)[0] + return self.outputs()[idx] def set_rbf(self, rbf: bool) -> None: nSequence = 0xffffffff - (2 if rbf else 1) diff --git a/electrum/txbatcher.py b/electrum/txbatcher.py index 3dcf159fd..6451bfb33 100644 --- a/electrum/txbatcher.py +++ b/electrum/txbatcher.py @@ -559,7 +559,7 @@ class TxBatch(Logger): self._prevout = None @locked - def _new_base_tx(self, tx: Transaction) -> None: + def _new_base_tx(self, tx: PartialTransaction) -> None: self._prevout = tx.inputs()[0].prevout.to_str() self.storage['prevout'] = self._prevout if tx.has_change():