lbry-sdk/lbrynet/wallet/account.py

171 lines
6.4 KiB
Python
Raw Normal View History

import json
2018-07-12 18:14:47 +02:00
import logging
2018-07-12 05:18:59 +02:00
from twisted.internet import defer
from torba.baseaccount import BaseAccount
from torba.basetransaction import TXORef
2018-07-12 05:18:59 +02:00
from lbryschema.claim import ClaimDict
from lbryschema.signer import SECP256k1, get_signer
2018-07-12 18:14:47 +02:00
log = logging.getLogger(__name__)
def generate_certificate():
secp256k1_private_key = get_signer(SECP256k1).generate().private_key.to_pem()
return ClaimDict.generate_certificate(secp256k1_private_key, curve=SECP256k1), secp256k1_private_key
class Account(BaseAccount):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.certificates = {}
2018-07-05 04:16:02 +02:00
def add_certificate_private_key(self, ref: TXORef, private_key):
assert ref.id not in self.certificates, 'Trying to add a duplicate certificate.'
self.certificates[ref.id] = private_key
2018-07-05 04:16:02 +02:00
def get_certificate_private_key(self, ref: TXORef):
return self.certificates.get(ref.id)
2018-07-12 05:18:59 +02:00
@defer.inlineCallbacks
def maybe_migrate_certificates(self):
if not self.certificates:
return
addresses = {}
results = {
'total': 0,
'not-a-claim-tx': 0,
'migrate-success': 0,
'migrate-failed': 0,
'previous-success': 0,
'previous-corrupted': 0
}
for maybe_claim_id in list(self.certificates):
results['total'] += 1
2018-07-12 07:47:34 +02:00
if ':' not in maybe_claim_id:
claims = yield self.ledger.network.get_claims_by_ids(maybe_claim_id)
if maybe_claim_id not in claims:
log.warning(
"Failed to migrate claim '%s', server did not return any claim information.",
maybe_claim_id
)
results['migrate-failed'] += 1
continue
2018-07-12 18:14:47 +02:00
claim = claims[maybe_claim_id]
tx = None
if claim:
tx = yield self.ledger.get_transaction(claim['txid'])
else:
log.warning(maybe_claim_id)
2018-07-12 18:14:47 +02:00
if tx is not None:
txo = tx.outputs[claim['nout']]
if not txo.script.is_claim_involved:
results['not-a-claim-tx'] += 1
raise ValueError(
"Certificate with claim_id {} doesn't point to a valid transaction."
.format(maybe_claim_id)
)
2018-07-12 18:14:47 +02:00
tx_nout = '{txid}:{nout}'.format(**claim)
self.certificates[tx_nout] = self.certificates[maybe_claim_id]
del self.certificates[maybe_claim_id]
log.info(
2018-07-15 07:20:44 +02:00
"Migrated certificate with claim_id '%s' ('%s') to a new look up key %s.",
maybe_claim_id, txo.script.values['claim_name'], tx_nout
2018-07-12 18:14:47 +02:00
)
results['migrate-success'] += 1
2018-07-12 18:14:47 +02:00
else:
if claim:
addresses.setdefault(claim['address'], 0)
addresses[claim['address']] += 1
log.warning(
"Failed to migrate claim '%s', it's not associated with any of your addresses.",
maybe_claim_id
)
else:
log.warning(
"Failed to migrate claim '%s', it appears abandoned.",
maybe_claim_id
)
results['migrate-failed'] += 1
else:
try:
txid, nout = maybe_claim_id.split(':')
tx = yield self.ledger.get_transaction(txid)
if tx.outputs[int(nout)].script.is_claim_involved:
results['previous-success'] += 1
else:
results['previous-corrupted'] += 1
except Exception:
log.exception("Couldn't verify certificate with look up key: %s", maybe_claim_id)
results['previous-corrupted'] += 1
self.wallet.save()
log.info('verifying and possibly migrating certificates:')
log.info(json.dumps(results, indent=2))
if addresses:
log.warning('failed for addresses:')
log.warning(json.dumps(
[{'address': a, 'number of certificates': c} for a, c in addresses.items()],
indent=2
))
2018-07-12 05:57:22 +02:00
def get_balance(self, confirmations=6, include_claims=False, **constraints):
if not include_claims:
constraints.update({'is_claim': 0, 'is_update': 0, 'is_support': 0})
return super().get_balance(confirmations, **constraints)
def get_unspent_outputs(self, include_claims=False, **constraints):
if not include_claims:
constraints.update({'is_claim': 0, 'is_update': 0, 'is_support': 0})
return super().get_unspent_outputs(**constraints)
2018-07-12 07:47:34 +02:00
@defer.inlineCallbacks
def get_channels(self):
utxos = yield super().get_unspent_outputs(
claim_type__any={'is_claim': 1, 'is_update': 1},
claim_name__like='@%'
)
channels = []
for utxo in utxos:
d = ClaimDict.deserialize(utxo.script.values['claim'])
channels.append({
'name': utxo.claim_name,
'claim_id': utxo.claim_id,
'txid': utxo.tx_ref.id,
'nout': utxo.position,
'have_certificate': utxo.ref.id in self.certificates
})
defer.returnValue(channels)
@classmethod
def get_private_key_from_seed(cls, ledger: 'baseledger.BaseLedger', seed: str, password: str):
return super().get_private_key_from_seed(
ledger, seed, password or 'lbryum'
)
2018-07-12 07:47:34 +02:00
@classmethod
def from_dict(cls, ledger, wallet, d: dict) -> 'Account':
account = super().from_dict(ledger, wallet, d)
2018-08-06 08:53:27 +02:00
account.certificates = d.get('certificates', {})
2018-07-12 07:47:34 +02:00
return account
def to_dict(self):
d = super().to_dict()
2018-07-12 07:47:34 +02:00
d['certificates'] = self.certificates
return d
2018-08-30 06:04:25 +02:00
@defer.inlineCallbacks
def get_details(self, **kwargs):
details = yield super().get_details(**kwargs)
details['certificates'] = len(self.certificates)
return details
def get_claim(self, claim_id):
2018-08-01 15:11:34 +02:00
return self.ledger.db.get_claim(self, claim_id)