Initial commit

This commit is contained in:
nolash
2021-06-13 11:09:48 +02:00
commit 819b07f173
4 changed files with 234 additions and 0 deletions

45
eth_cache/account.py Normal file
View File

@@ -0,0 +1,45 @@
# external imports
from hexathon import strip_0x
class AccountRegistry:
def __init__(self):
self.senders = {}
self.recipients = {}
def __normalize_address(self, address):
return bytes.fromhex(strip_0x(address))
def add(self, address, label):
self.add_sender(address, label)
self.add_recipient(address, label)
def add_sender(self, address, label):
a = self.__normalize_address(address)
self.senders[a] = label
def add_recipient(self, address, label):
a = self.__normalize_address(address)
self.recipients[a] = label
def have(self, address):
if self.get_sender(address) != None:
return True
if self.get_recipient(address) != None:
return True
return False
def get_sender(self, address):
a = self.__normalize_address(address)
return self.senders.get(a)
def get_recipient(self, address):
a = self.__normalize_address(address)
return self.recipients.get(a)

7
eth_cache/filter.py Normal file
View File

@@ -0,0 +1,7 @@
class GasFilter:
def __init__(self, store):
def filter(self, conn, block, tx, session):
self.store.put()

65
eth_cache/store.py Normal file
View File

@@ -0,0 +1,65 @@
# standard imports
import logging
import os
# external imports
from hexdir import HexDir
from chainlib.eth.tx import pack
from hexathon import strip_0x
logg = logging.getLogger(__name__)
class PointerHexDir(HexDir):
def __init__(self, root_path, key_length, levels=2, prefix_length=0):
super(PointerHexDir, self).__init__(root_path, key_length, levels, prefix_length)
self.pointers = {}
def register_pointer(self, label, dir_name=None):
if dir_name == None:
dir_name = label
pointer_dir = os.path.join(self.path, dir_name)
os.makedirs(pointer_dir, exist_ok=True)
label_file = os.path.join(pointer_dir, '.label')
try:
os.stat(label_file)
except FileNotFoundError:
f = open(label_file, 'w')
f.write(label)
f.close()
self.pointers[label] = pointer_dir
def add_pointer(self, pointer, pointer_relpath, destination_path):
if isinstance(pointer_relpath, list):
link_path = os.path.join(self.pointers[pointer], *pointer_relpath)
else:
link_path = os.path.join(self.pointers[pointer], pointer_relpath)
os.makedirs(os.path.dirname(link_path), exist_ok=True)
os.symlink(destination_path, link_path)
logg.debug('added link {} -> {}'.format(link_path, destination_path))
def add(self, key, content, prefix=b'', pointers={}):
(c, entry_path) = super(PointerHexDir, self).add(key, content, prefix=prefix)
for k in pointers.keys():
self.add_pointer(k, pointers[k], entry_path)
class TxFileStore:
def __init__(self, chain_spec, backend):
self.backend = backend
self.chain_spec = chain_spec
def put(self, block, tx, addresses, attrs={}):
for address in addresses:
tx_src = tx.as_dict()
tx_raw = pack(tx_src, self.chain_spec)
filename = '{}_{}_{}'.format(block.number, tx.index, strip_0x(tx.hash))
self.backend.add(bytes.fromhex(tx.hash), tx_raw, pointers={'address': [strip_0x(address),filename]})