lbry-sdk/lbrynet/daemon/json_response_encoder.py

58 lines
1.9 KiB
Python
Raw Normal View History

2018-08-16 01:23:06 +02:00
from decimal import Decimal
from binascii import hexlify
from datetime import datetime
from json import JSONEncoder
2018-08-16 17:24:22 +02:00
from lbrynet.wallet.transaction import Transaction, Output
2018-08-16 01:23:06 +02:00
class JSONResponseEncoder(JSONEncoder):
def __init__(self, *args, ledger, **kwargs):
super().__init__(*args, **kwargs)
self.ledger = ledger
2018-08-16 17:24:22 +02:00
def default(self, obj): # pylint: disable=method-hidden
2018-08-16 01:23:06 +02:00
if isinstance(obj, Transaction):
return self.encode_transaction(obj)
if isinstance(obj, Output):
return self.encode_output(obj)
if isinstance(obj, datetime):
return obj.strftime("%Y%m%dT%H:%M:%S")
if isinstance(obj, Decimal):
return float(obj)
2018-08-16 21:55:33 +02:00
if isinstance(obj, bytes):
return obj.decode()
2018-08-16 01:23:06 +02:00
return super().default(obj)
def encode_transaction(self, tx):
return {
'txid': tx.id,
'height': tx.height,
2018-08-16 01:23:06 +02:00
'inputs': [self.encode_input(txo) for txo in tx.inputs],
'outputs': [self.encode_output(txo) for txo in tx.outputs],
'total_input': tx.input_sum,
'total_output': tx.input_sum - tx.fee,
'total_fee': tx.fee,
'hex': hexlify(tx.raw).decode(),
}
def encode_output(self, txo):
output = {
2018-09-19 15:58:50 +02:00
'txid': txo.tx_ref.id,
2018-08-16 01:23:06 +02:00
'nout': txo.position,
'amount': txo.amount,
2018-09-19 15:58:50 +02:00
'address': txo.get_address(self.ledger),
'is_claim': txo.script.is_claim_name,
'is_support': txo.script.is_support_claim,
'is_update': txo.script.is_update_claim,
2018-08-16 01:23:06 +02:00
}
if txo.is_change is not None:
output['is_change'] = txo.is_change
return output
2018-08-16 01:23:06 +02:00
def encode_input(self, txi):
return self.encode_output(txi.txo_ref.txo) if txi.txo_ref.txo is not None else {
'txid': txi.txo_ref.tx_ref.id,
'nout': txi.txo_ref.position
}