erc20-pool/python/erc20_pool/quote.py

99 lines
2.6 KiB
Python
Raw Normal View History

# standard imports
import logging
import os
import enum
# external imports
from chainlib.eth.constant import ZERO_ADDRESS
from chainlib.eth.constant import ZERO_CONTENT
from chainlib.eth.contract import (
ABIContractEncoder,
ABIContractDecoder,
ABIContractType,
abi_decode_single,
)
from chainlib.eth.jsonrpc import to_blockheight_param
from chainlib.eth.error import RequestMismatchException
from chainlib.eth.tx import (
TxFactory,
TxFormat,
)
from chainlib.jsonrpc import JSONRPCRequest
from chainlib.block import BlockSpec
from hexathon import (
add_0x,
strip_0x,
)
from chainlib.eth.cli.encode import CLIEncoder
# local imports
from erc20_pool.data import data_dir
logg = logging.getLogger()
class DecimalQuote(TxFactory):
__abi = None
__bytecode = None
def constructor(self, sender_address, tx_format=TxFormat.JSONRPC, version=None):
code = self.cargs(version=version)
tx = self.template(sender_address, None, use_nonce=True)
tx = self.set_code(tx, code)
return self.finalize(tx, tx_format)
@staticmethod
def cargs(version=None):
code = DecimalQuote.bytecode(version=version)
enc = ABIContractEncoder()
args = enc.get()
code += args
logg.debug('constructor code: ' + args)
return code
@staticmethod
def gas(code=None):
return 4000000
@staticmethod
def abi():
if DecimalQuote.__abi == None:
f = open(os.path.join(data_dir, 'DecimalQuote.json'), 'r')
DecimalQuote.__abi = json.load(f)
f.close()
return DecimalQuote.__abi
@staticmethod
def bytecode(version=None):
if DecimalQuote.__bytecode == None:
f = open(os.path.join(data_dir, 'DecimalQuote.bin'))
DecimalQuote.__bytecode = f.read()
f.close()
return DecimalQuote.__bytecode
def value_for(self, contract_address, token_address_out, token_address_in, value_in, sender_address=ZERO_ADDRESS, id_generator=None):
j = JSONRPCRequest(id_generator)
o = j.template()
o['method'] = 'eth_call'
enc = ABIContractEncoder()
enc.method('valueFor')
enc.typ(ABIContractType.ADDRESS)
enc.typ(ABIContractType.ADDRESS)
enc.typ(ABIContractType.UINT256)
enc.address(token_address_out)
enc.address(token_address_in)
enc.uint256(value_in)
data = add_0x(enc.get())
tx = self.template(sender_address, contract_address)
tx = self.set_code(tx, data)
o['params'].append(self.normalize(tx))
o['params'].append('latest')
o = j.finalize(o)
return o