mypy related fix

This commit is contained in:
Lex Berezhny 2018-09-25 22:46:22 -04:00
parent a879c822c2
commit 5c5b0ace53
2 changed files with 4 additions and 3 deletions

View file

@ -1,5 +1,5 @@
import typing import typing
from typing import Dict, Tuple, Type, Optional, Any from typing import List, Dict, Tuple, Type, Optional, Any
from twisted.internet import defer from twisted.internet import defer
from torba.mnemonic import Mnemonic from torba.mnemonic import Mnemonic
@ -10,6 +10,7 @@ from torba.constants import COIN
if typing.TYPE_CHECKING: if typing.TYPE_CHECKING:
from torba import baseledger from torba import baseledger
from torba import wallet as basewallet from torba import wallet as basewallet
from torba import basetransaction
class AddressManager: class AddressManager:
@ -358,7 +359,7 @@ class BaseAccount:
def get_unspent_outputs(self, **constraints): def get_unspent_outputs(self, **constraints):
return self.ledger.db.get_utxos_for_account(self, **constraints) return self.ledger.db.get_utxos_for_account(self, **constraints)
def get_transactions(self): def get_transactions(self) -> 'List[basetransaction.BaseTransaction]':
return self.ledger.db.get_transactions(self) return self.ledger.db.get_transactions(self)
@defer.inlineCallbacks @defer.inlineCallbacks

View file

@ -270,7 +270,7 @@ class BaseDatabase(SQLiteMixin):
return None, None, None, False return None, None, None, False
@defer.inlineCallbacks @defer.inlineCallbacks
def get_transactions(self, account, offset=0, limit=100) -> List[BaseTransaction]: def get_transactions(self, account, offset=0, limit=100):
account_id = account.public_key.address account_id = account.public_key.address
tx_rows = yield self.run_query( tx_rows = yield self.run_query(
""" """