lbry-sdk/lbrynet/wallet/database.py

76 lines
2.7 KiB
Python
Raw Normal View History

from binascii import hexlify
2018-07-12 05:18:59 +02:00
from twisted.internet import defer
2018-06-12 17:53:29 +02:00
from torba.basedatabase import BaseDatabase
2018-07-12 05:18:59 +02:00
from .certificate import Certificate
2018-06-12 17:53:29 +02:00
class WalletDatabase(BaseDatabase):
CREATE_TXO_TABLE = """
create table if not exists txo (
2018-07-15 05:02:19 +02:00
txid text references tx,
txoid text primary key,
address text references pubkey_address,
position integer not null,
amount integer not null,
script blob not null,
is_reserved boolean not null default 0,
2018-07-12 05:18:59 +02:00
2018-07-15 05:02:19 +02:00
claim_id text,
claim_name text,
is_claim boolean not null default 0,
is_update boolean not null default 0,
is_support boolean not null default 0
);
"""
2018-06-12 17:53:29 +02:00
CREATE_TABLES_QUERY = (
BaseDatabase.CREATE_TX_TABLE +
BaseDatabase.CREATE_PUBKEY_ADDRESS_TABLE +
CREATE_TXO_TABLE +
2018-06-12 17:53:29 +02:00
BaseDatabase.CREATE_TXI_TABLE
)
def txo_to_row(self, tx, address, txo):
row = super().txo_to_row(tx, address, txo)
row.update({
'is_claim': txo.script.is_claim_name,
'is_update': txo.script.is_update_claim,
'is_support': txo.script.is_support_claim,
})
if txo.script.is_claim_involved:
row['claim_name'] = txo.script.values['claim_name'].decode()
2018-07-12 07:05:24 +02:00
if txo.script.is_update_claim or txo.script.is_support_claim:
row['claim_id'] = hexlify(txo.script.values['claim_id'][::-1])
2018-07-12 07:15:59 +02:00
elif txo.script.is_claim_name:
row['claim_id'] = hexlify(tx.get_claim_id(txo.position)[::-1])
return row
2018-07-12 05:18:59 +02:00
@defer.inlineCallbacks
def get_certificates(self, name, private_key_accounts=None, exclude_without_key=False):
txos = yield self.db.runQuery(
"""
2018-07-16 02:54:55 +02:00
SELECT tx.txid, txo.position, txo.claim_id
FROM txo JOIN tx ON tx.txid=txo.txid
WHERE claim_name=? AND (is_claim OR is_update)
GROUP BY txo.claim_id ORDER BY tx.height DESC;
""", (name,)
2018-07-12 05:18:59 +02:00
)
2018-07-16 02:54:55 +02:00
certificates = []
2018-07-12 05:18:59 +02:00
# Lookup private keys for each certificate.
if private_key_accounts is not None:
2018-07-16 02:54:55 +02:00
for txhash, nout, claim_id in txos:
2018-07-12 05:18:59 +02:00
for account in private_key_accounts:
private_key = account.get_certificate_private_key(
2018-07-16 02:54:55 +02:00
txhash, nout
2018-07-12 05:18:59 +02:00
)
2018-07-16 02:54:55 +02:00
certificates.append(Certificate(txhash, nout, claim_id, name, private_key))
2018-07-12 05:18:59 +02:00
if exclude_without_key:
defer.returnValue([
c for c in certificates if c.private_key is not None
])
defer.returnValue(certificates)