134 lines
3.5 KiB
Python
134 lines
3.5 KiB
Python
# standard imports
|
|
import logging
|
|
|
|
# ecxternal imports
|
|
from hexathon import (
|
|
add_0x,
|
|
strip_0x,
|
|
uniform,
|
|
)
|
|
|
|
logg = logging.getLogger(__name__)
|
|
|
|
|
|
def normalize_hex(k):
|
|
k = strip_0x(k)
|
|
return uniform(k)
|
|
|
|
|
|
class QueueEntry:
|
|
|
|
def __init__(self, store, tx_hash):
|
|
self.store = store
|
|
self.tx_hash = normalize_hex(tx_hash)
|
|
self.signed_tx = None
|
|
self.seq = None
|
|
self.k = None
|
|
self.synced = False
|
|
|
|
|
|
def serialize(self):
|
|
return self.signed_tx
|
|
|
|
|
|
def create(self, signed_tx):
|
|
signed_tx = normalize_hex(signed_tx)
|
|
self.k = self.store.put(self.tx_hash, signed_tx)
|
|
self.synced = True
|
|
|
|
|
|
def load(self):
|
|
(self.k, self.signed_tx) = self.store.get(self.tx_hash)
|
|
self.synced = True
|
|
|
|
|
|
def __match_state(self, state):
|
|
return bool(self.store.state(self.k) & state)
|
|
|
|
|
|
def waitforfunds(self):
|
|
if self.__match_state(self.store.INSUFFICIENT_FUNDS):
|
|
return
|
|
self.store.move(self.k, self.store.INSUFFICIENT_FUNDS)
|
|
|
|
|
|
def fubar(self):
|
|
if self.__match_state(self.store.UNKNOWN_ERROR):
|
|
return
|
|
self.store.set(self.k, self.store.UNKNOWN_ERROR)
|
|
|
|
|
|
def reject(self):
|
|
if self.__match_state(self.store.NODE_ERROR):
|
|
return
|
|
self.store.set(self.k, self.store.NODE_ERROR)
|
|
|
|
|
|
def override(self, manual=False):
|
|
if manual:
|
|
self.store.set(self.k, self.store.OBSOLETE | self.store.MANUAL)
|
|
else:
|
|
self.store.set(self.k, self.store.OBSOLETE)
|
|
|
|
|
|
def manual(self):
|
|
self.store.set(self.k, self.store.MANUAL)
|
|
|
|
|
|
def retry(self):
|
|
if self.__match_state(self.store.QUEUED):
|
|
return
|
|
self.store.change(self.k, self.store.QUEUED, self.store.INSUFFICIENT_FUNDS)
|
|
|
|
|
|
def readysend(self):
|
|
if self.__match_state(self.store.QUEUED):
|
|
return
|
|
self.store.change(self.k, self.store.QUEUED, self.store.INSUFFICIENT_FUNDS)
|
|
|
|
|
|
def sent(self):
|
|
if self.__match_state(self.store.IN_NETWORK):
|
|
return
|
|
self.store.change(self.k, self.store.IN_NETWORK, self.store.RESERVED | self.store.DEFERRED | self.store.QUEUED | self.store.LOCAL_ERROR | self.store.NODE_ERROR)
|
|
|
|
|
|
def sendfail(self):
|
|
if self.__match_state(self.store.NODE_ERROR):
|
|
return
|
|
self.store.change(self.k, self.store.LOCAL_ERROR | self.store.DEFERRED, self.store.RESERVED | self.store.QUEUED | self.store.INSUFFICIENT_FUNDS)
|
|
|
|
|
|
def reserve(self):
|
|
if self.__match_state(self.store.RESERVED):
|
|
return
|
|
self.store.change(self.k, self.store.RESERVED, self.store.QUEUED)
|
|
|
|
|
|
def fail(self, block, tx):
|
|
if self.__match_state(self.store.NETWORK_ERROR):
|
|
return
|
|
self.store.set(self.k, self.store.NETWORK_ERROR)
|
|
if self.store.cache:
|
|
self.store.cache.set_block(self.tx_hash, block, tx)
|
|
|
|
|
|
def cancel(self, confirmed=False):
|
|
if confirmed:
|
|
self.store.change(self.k, self.store.OBSOLETE | self.store.FINAL, self.store.RESERVED | self.store.QUEUED)
|
|
else:
|
|
self.store.change(self.k, self.store.OBSOLETE, self.store.RESERVED | self.store.QUEUED)
|
|
|
|
|
|
def succeed(self, block, tx):
|
|
self.store.set(self.k, self.store.FINAL)
|
|
if self.store.cache:
|
|
self.store.cache.set_block(self.tx_hash, block, tx)
|
|
|
|
|
|
def __str__(self):
|
|
v = self.store.get(self.tx_hash)
|
|
n = self.store.state(v[0])
|
|
s = self.store.name(n)
|
|
return '{}: {}'.format(self.tx_hash, s)
|