2021-02-03 19:40:03 +01:00
# standard imports
import os
import sys
import logging
import time
import argparse
import sys
import re
# third-party imports
import confini
2021-02-11 09:02:17 +01:00
from chainlib . eth . connection import HTTPConnection
from chainsyncer . driver import HeadSyncer
from chainsyncer . db import dsn_from_config
from chainsyncer . db . models . base import SessionBase
from chainsyncer . backend import SyncerBackend
from chainsyncer . error import LoopDone
2021-02-03 19:40:03 +01:00
logging . basicConfig ( level = logging . WARNING )
logg = logging . getLogger ( )
2021-02-03 20:55:39 +01:00
config_dir = ' /usr/local/etc/cic-syncer '
2021-02-03 19:40:03 +01:00
2021-02-03 23:32:19 +01:00
class Handler :
def __init__ ( self , method , domain ) :
self . method = method
self . domain = domain
def handle ( self , getter , tx , chain ) :
logg . debug ( ' noop tx {} chain {} method {} domain {} ' . format ( tx , chain , self . method , self . domain ) )
handler = getattr ( Handler , ' handle ' )
2021-02-03 19:40:03 +01:00
argparser = argparse . ArgumentParser ( description = ' daemon that monitors transactions in new blocks ' )
2021-02-03 20:55:39 +01:00
argparser . add_argument ( ' -p ' , ' --provider ' , dest = ' p ' , type = str , help = ' chain rpc provider address ' )
2021-02-03 19:40:03 +01:00
argparser . add_argument ( ' -c ' , type = str , default = config_dir , help = ' config root to use ' )
argparser . add_argument ( ' -i ' , ' --chain-spec ' , type = str , dest = ' i ' , help = ' chain spec ' )
argparser . add_argument ( ' --abi-dir ' , dest = ' abi_dir ' , type = str , help = ' Directory containing bytecode and abi ' )
argparser . add_argument ( ' --env-prefix ' , default = os . environ . get ( ' CONFINI_ENV_PREFIX ' ) , dest = ' env_prefix ' , type = str , help = ' environment prefix for variables to overwrite configuration ' )
argparser . add_argument ( ' -q ' , type = str , default = ' cic-eth ' , help = ' celery queue to submit transaction tasks to ' )
argparser . add_argument ( ' -v ' , help = ' be verbose ' , action = ' store_true ' )
argparser . add_argument ( ' -vv ' , help = ' be more verbose ' , action = ' store_true ' )
args = argparser . parse_args ( sys . argv [ 1 : ] )
if args . v == True :
logging . getLogger ( ) . setLevel ( logging . INFO )
elif args . vv == True :
logging . getLogger ( ) . setLevel ( logging . DEBUG )
config_dir = os . path . join ( args . c )
os . makedirs ( config_dir , 0o777 , True )
config = confini . Config ( config_dir , args . env_prefix )
config . process ( )
# override args
args_override = {
2021-02-11 09:02:17 +01:00
' CHAIN_SPEC ' : getattr ( args , ' i ' ) ,
2021-02-03 20:55:39 +01:00
' ETH_PROVIDER ' : getattr ( args , ' p ' ) ,
2021-02-03 19:40:03 +01:00
}
config . dict_override ( args_override , ' cli flag ' )
config . censor ( ' PASSWORD ' , ' DATABASE ' )
config . censor ( ' PASSWORD ' , ' SSL ' )
logg . debug ( ' config loaded from {} : \n {} ' . format ( config_dir , config ) )
2021-02-03 20:55:39 +01:00
#app = celery.Celery(backend=config.get('CELERY_RESULT_URL'), broker=config.get('CELERY_BROKER_URL'))
2021-02-03 19:40:03 +01:00
queue = args . q
dsn = dsn_from_config ( config )
SessionBase . connect ( dsn )
2021-02-11 09:02:17 +01:00
c = HTTPConnection ( config . get ( ' ETH_PROVIDER ' ) )
chain = config . get ( ' CHAIN_SPEC ' )
2021-02-03 21:10:08 +01:00
2021-02-03 19:40:03 +01:00
def main ( ) :
2021-02-03 20:55:39 +01:00
block_offset = c . block_number ( )
2021-02-03 19:40:03 +01:00
2021-02-09 23:26:48 +01:00
syncer_backend = SyncerBackend . live ( chain , 0 )
2021-02-11 09:02:17 +01:00
syncer = HeadSyncer ( syncer_backend )
2021-02-03 19:40:03 +01:00
try :
2021-02-03 21:57:26 +01:00
logg . debug ( ' block offset {} {} ' . format ( block_offset , c ) )
syncer . loop ( int ( config . get ( ' SYNCER_LOOP_INTERVAL ' ) ) , c )
2021-02-03 19:40:03 +01:00
except LoopDone as e :
sys . stderr . write ( " sync ' {} ' done at block {} \n " . format ( args . mode , e ) )
sys . exit ( 0 )
if __name__ == ' __main__ ' :
main ( )