Normalize hex in queueentry
This commit is contained in:
parent
3880249683
commit
790c9ddf13
@ -1,27 +1,45 @@
|
|||||||
# standard imports
|
# standard imports
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
|
# ecxternal imports
|
||||||
|
from hexathon import (
|
||||||
|
add_0x,
|
||||||
|
strip_0x,
|
||||||
|
uniform,
|
||||||
|
)
|
||||||
|
|
||||||
logg = logging.getLogger(__name__)
|
logg = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
def to_key(k, v):
|
||||||
|
return '{:>010s}_{}'.format(k, v)
|
||||||
|
|
||||||
|
|
||||||
|
def from_key(k):
|
||||||
|
(seq_str, tx_hash) = k.split('_')
|
||||||
|
return (int(seq_str), tx_hash,)
|
||||||
|
|
||||||
|
|
||||||
|
def normalize_hex(k):
|
||||||
|
k = strip_0x(k)
|
||||||
|
return uniform(k)
|
||||||
|
|
||||||
|
|
||||||
class QueueEntry:
|
class QueueEntry:
|
||||||
|
|
||||||
def __init__(self, store, tx_hash):
|
def __init__(self, store, tx_hash):
|
||||||
self.store = store
|
self.store = store
|
||||||
self.tx_hash = tx_hash
|
self.tx_hash = normalize_hex(tx_hash)
|
||||||
self.signed_tx = None
|
self.signed_tx = None
|
||||||
self.seq = None
|
self.seq = None
|
||||||
self.k = None
|
self.k = None
|
||||||
self.synced = False
|
self.synced = False
|
||||||
|
|
||||||
|
|
||||||
def __to_key(self, k, v):
|
|
||||||
return '{:>010s}_{}'.format(k, v)
|
|
||||||
|
|
||||||
|
|
||||||
def create(self, seq, signed_tx):
|
def create(self, seq, signed_tx):
|
||||||
n = str(seq)
|
n = str(seq)
|
||||||
self.k = self.__to_key(n, self.tx_hash)
|
signed_tx = normalize_hex(signed_tx)
|
||||||
|
self.k = to_key(n, self.tx_hash)
|
||||||
self.store.put(self.k, signed_tx)
|
self.store.put(self.k, signed_tx)
|
||||||
self.store.put_seq(self.tx_hash, n)
|
self.store.put_seq(self.tx_hash, n)
|
||||||
self.synced = True
|
self.synced = True
|
||||||
@ -29,7 +47,7 @@ class QueueEntry:
|
|||||||
|
|
||||||
def load(self):
|
def load(self):
|
||||||
seq = self.store.get_seq(self.tx_hash)
|
seq = self.store.get_seq(self.tx_hash)
|
||||||
self.k = self.__to_key(seq, self.tx_hash)
|
self.k = to_key(seq, self.tx_hash)
|
||||||
self.signed_tx = self.store.get(self.k)
|
self.signed_tx = self.store.get(self.k)
|
||||||
self.synced = True
|
self.synced = True
|
||||||
|
|
||||||
|
@ -2,6 +2,9 @@
|
|||||||
import logging
|
import logging
|
||||||
import re
|
import re
|
||||||
|
|
||||||
|
# local imports
|
||||||
|
from .entry import from_key
|
||||||
|
|
||||||
logg = logging.getLogger(__name__)
|
logg = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
@ -45,4 +48,9 @@ class Store:
|
|||||||
if item_state & state != item_state:
|
if item_state & state != item_state:
|
||||||
continue
|
continue
|
||||||
hashes.append(k)
|
hashes.append(k)
|
||||||
return hashes
|
hashes.sort()
|
||||||
|
hashes_out = []
|
||||||
|
for h in hashes:
|
||||||
|
pair = from_key(h)
|
||||||
|
hashes_out.append(pair[1])
|
||||||
|
return hashes_out
|
||||||
|
@ -4,7 +4,10 @@ import logging
|
|||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
# external imports
|
# external imports
|
||||||
from hexathon import add_0x
|
from hexathon import (
|
||||||
|
add_0x,
|
||||||
|
strip_0x,
|
||||||
|
)
|
||||||
from shep.error import StateTransitionInvalid
|
from shep.error import StateTransitionInvalid
|
||||||
|
|
||||||
# local imports
|
# local imports
|
||||||
@ -32,7 +35,7 @@ class TestShep(TestShepBase):
|
|||||||
|
|
||||||
tx_retrieved = QueueEntry(self.store, tx_hash)
|
tx_retrieved = QueueEntry(self.store, tx_hash)
|
||||||
tx_retrieved.load()
|
tx_retrieved.load()
|
||||||
self.assertEqual(tx_retrieved.signed_tx, signed_tx)
|
self.assertEqual(tx_retrieved.signed_tx, strip_0x(signed_tx))
|
||||||
|
|
||||||
|
|
||||||
def test_shep_valid(self):
|
def test_shep_valid(self):
|
||||||
|
Loading…
Reference in New Issue
Block a user