168 lines
4.4 KiB
Python
168 lines
4.4 KiB
Python
# standard imports
|
||
import logging
|
||
import json
|
||
import re
|
||
import base64
|
||
|
||
# external imports
|
||
from hexathon import (
|
||
add_0x,
|
||
strip_0x,
|
||
)
|
||
|
||
# local imports
|
||
from cic_cache.cache import (
|
||
BloomCache,
|
||
DataCache,
|
||
)
|
||
|
||
logg = logging.getLogger(__name__)
|
||
#logg = logging.getLogger()
|
||
|
||
re_transactions_all_bloom = r'/tx/(\d+)?/?(\d+)/?(\d+)?/?(\d+)?/?'
|
||
re_transactions_account_bloom = r'/tx/user/((0x)?[a-fA-F0-9]+)(/(\d+)(/(\d+))?)?/?'
|
||
re_transactions_all_data = r'/txa/(\d+)/?(\d+)?/?(\d+)?/?(\d+)?/?'
|
||
re_transactions_account_data = r'/txa/user/((0x)?[a-fA-F0-9]+)(/(\d+)(/(\d+))?)?/?'
|
||
|
||
DEFAULT_LIMIT = 100
|
||
|
||
|
||
def parse_query_account(r):
|
||
address = strip_0x(r[1])
|
||
limit = DEFAULT_LIMIT
|
||
g = r.groups()
|
||
if len(g) > 3:
|
||
limit = r[4]
|
||
offset = 0
|
||
if len(g) > 4:
|
||
offset = r[6]
|
||
|
||
logg.debug('account query is address {} offset {} limit {}'.format(address, offset, limit))
|
||
|
||
return (address, offset, limit,)
|
||
|
||
|
||
# r is an re.Match
|
||
def parse_query_any(r):
|
||
limit = DEFAULT_LIMIT
|
||
if r.lastindex > 0:
|
||
limit = int(r[1])
|
||
offset = 0
|
||
if r.lastindex > 1:
|
||
offset = int(r[2])
|
||
block_offset = None
|
||
if r.lastindex > 2:
|
||
block_offset = int(r[3])
|
||
block_end = None
|
||
if r.lastindex > 3:
|
||
block_end = int(r[4])
|
||
if block_end < block_offset:
|
||
raise ValueError('cart before the horse, dude')
|
||
|
||
logg.debug('data query is offset {} limit {} block_offset {} block_end {}'.format(offset, limit, block_offset, block_end))
|
||
|
||
return (offset, limit, block_offset, block_end,)
|
||
|
||
|
||
def process_transactions_account_bloom(session, env):
|
||
r = re.match(re_transactions_account_bloom, env.get('PATH_INFO'))
|
||
if not r:
|
||
return None
|
||
|
||
(address, offset, limit,) = parse_query_account(r)
|
||
|
||
c = BloomCache(session)
|
||
(lowest_block, highest_block, bloom_filter_block, bloom_filter_tx) = c.load_transactions_account(address, offset, limit)
|
||
|
||
o = {
|
||
'alg': 'sha256',
|
||
'low': lowest_block,
|
||
'high': highest_block,
|
||
'block_filter': base64.b64encode(bloom_filter_block).decode('utf-8'),
|
||
'blocktx_filter': base64.b64encode(bloom_filter_tx).decode('utf-8'),
|
||
'filter_rounds': 3,
|
||
}
|
||
|
||
j = json.dumps(o)
|
||
|
||
return ('application/json', j.encode('utf-8'),)
|
||
|
||
|
||
def process_transactions_all_bloom(session, env):
|
||
r = re.match(re_transactions_all_bloom, env.get('PATH_INFO'))
|
||
if not r:
|
||
return None
|
||
|
||
(limit, offset, block_offset, block_end,) = parse_query_any(r)
|
||
|
||
c = BloomCache(session)
|
||
(lowest_block, highest_block, bloom_filter_block, bloom_filter_tx) = c.load_transactions(offset, limit)
|
||
|
||
o = {
|
||
'alg': 'sha256',
|
||
'low': lowest_block,
|
||
'high': highest_block,
|
||
'block_filter': base64.b64encode(bloom_filter_block).decode('utf-8'),
|
||
'blocktx_filter': base64.b64encode(bloom_filter_tx).decode('utf-8'),
|
||
'filter_rounds': 3,
|
||
}
|
||
|
||
j = json.dumps(o)
|
||
|
||
return ('application/json', j.encode('utf-8'),)
|
||
|
||
|
||
def process_transactions_all_data(session, env):
|
||
r = re.match(re_transactions_all_data, env.get('PATH_INFO'))
|
||
if not r:
|
||
return None
|
||
#if env.get('HTTP_X_CIC_CACHE_MODE') != 'all':
|
||
# return None
|
||
|
||
logg.debug('got data request {}'.format(env))
|
||
|
||
(offset, limit, block_offset, block_end) = parse_query_any(r)
|
||
|
||
c = DataCache(session)
|
||
(lowest_block, highest_block, tx_cache) = c.load_transactions_with_data(offset, limit, block_offset, block_end, oldest=True) # oldest needs to be settable
|
||
|
||
for r in tx_cache:
|
||
r['date_block'] = r['date_block'].timestamp()
|
||
|
||
o = {
|
||
'low': lowest_block,
|
||
'high': highest_block,
|
||
'data': tx_cache,
|
||
}
|
||
|
||
|
||
j = json.dumps(o)
|
||
|
||
return ('application/json', j.encode('utf-8'),)
|
||
|
||
|
||
def process_transactions_account_data(session, env):
|
||
r = re.match(re_transactions_account_data, env.get('PATH_INFO'))
|
||
if not r:
|
||
return None
|
||
#if env.get('HTTP_X_CIC_CACHE_MODE') != 'all':
|
||
# return None
|
||
|
||
(address, offset, limit,) = parse_query_account(r)
|
||
|
||
c = DataCache(session)
|
||
(lowest_block, highest_block, tx_cache) = c.load_transactions_account_with_data(address, offset, limit)
|
||
|
||
for r in tx_cache:
|
||
r['date_block'] = r['date_block'].timestamp()
|
||
|
||
o = {
|
||
'low': lowest_block,
|
||
'high': highest_block,
|
||
'data': tx_cache,
|
||
}
|
||
|
||
j = json.dumps(o)
|
||
|
||
return ('application/json', j.encode('utf-8'),)
|