lbry-sdk/torba/basemanager.py

81 lines
2.6 KiB
Python
Raw Normal View History

2018-10-15 04:16:51 +02:00
import asyncio
import logging
from typing import Type, MutableSequence, MutableMapping
2018-05-25 08:03:25 +02:00
2018-06-12 16:02:04 +02:00
from torba.baseledger import BaseLedger, LedgerRegistry
2018-05-25 08:03:25 +02:00
from torba.wallet import Wallet, WalletStorage
log = logging.getLogger(__name__)
2018-05-25 08:03:25 +02:00
class BaseWalletManager:
2018-05-25 08:03:25 +02:00
def __init__(self, wallets: MutableSequence[Wallet] = None,
ledgers: MutableMapping[Type[BaseLedger], BaseLedger] = None) -> None:
2018-06-14 21:17:59 +02:00
self.wallets = wallets or []
self.ledgers = ledgers or {}
2018-05-25 08:03:25 +02:00
self.running = False
@classmethod
def from_config(cls, config: dict) -> 'BaseWalletManager':
2018-07-01 23:20:17 +02:00
manager = cls()
for ledger_id, ledger_config in config.get('ledgers', {}).items():
manager.get_or_create_ledger(ledger_id, ledger_config)
2018-05-25 08:03:25 +02:00
for wallet_path in config.get('wallets', []):
wallet_storage = WalletStorage(wallet_path)
wallet = Wallet.from_storage(wallet_storage, manager)
2018-07-01 23:20:17 +02:00
manager.wallets.append(wallet)
2018-05-25 08:03:25 +02:00
return manager
2018-06-12 16:02:04 +02:00
def get_or_create_ledger(self, ledger_id, ledger_config=None):
ledger_class = LedgerRegistry.get_ledger_class(ledger_id)
2018-05-25 08:03:25 +02:00
ledger = self.ledgers.get(ledger_class)
if ledger is None:
2018-06-14 21:17:59 +02:00
ledger = ledger_class(ledger_config or {})
2018-05-25 08:03:25 +02:00
self.ledgers[ledger_class] = ledger
return ledger
2018-07-05 03:17:45 +02:00
def import_wallet(self, path):
2018-06-14 21:17:59 +02:00
storage = WalletStorage(path)
wallet = Wallet.from_storage(storage, self)
self.wallets.append(wallet)
return wallet
2018-06-08 05:47:46 +02:00
2018-10-15 04:16:51 +02:00
async def get_detailed_accounts(self, confirmations=6, show_seed=False):
2018-08-30 17:50:11 +02:00
ledgers = {}
for i, account in enumerate(self.accounts):
2018-10-15 04:16:51 +02:00
details = await account.get_details(confirmations=confirmations, show_seed=True)
2018-08-30 17:50:11 +02:00
details['is_default_account'] = i == 0
ledger_id = account.ledger.get_id()
ledgers.setdefault(ledger_id, [])
ledgers[ledger_id].append(details)
return ledgers
2018-06-08 05:47:46 +02:00
2018-05-25 08:03:25 +02:00
@property
def default_wallet(self):
for wallet in self.wallets:
return wallet
@property
def default_account(self):
for wallet in self.wallets:
return wallet.default_account
2018-07-17 05:58:29 +02:00
@property
def accounts(self):
for wallet in self.wallets:
for account in wallet.accounts:
yield account
2018-10-15 04:16:51 +02:00
async def start(self):
2018-05-25 08:03:25 +02:00
self.running = True
2018-10-15 04:16:51 +02:00
await asyncio.gather(*(
2018-05-25 08:03:25 +02:00
l.start() for l in self.ledgers.values()
2018-10-15 04:16:51 +02:00
))
2018-05-25 08:03:25 +02:00
2018-10-15 04:16:51 +02:00
async def stop(self):
await asyncio.gather(*(
2018-05-25 08:03:25 +02:00
l.stop() for l in self.ledgers.values()
2018-10-15 04:16:51 +02:00
))
2018-05-25 08:03:25 +02:00
self.running = False