diff --git a/chainsyncer/cli/config.py b/chainsyncer/cli/config.py index d840512..c62ebf7 100644 --- a/chainsyncer/cli/config.py +++ b/chainsyncer/cli/config.py @@ -1,19 +1,15 @@ -# external imports -from chainsyncer.cli import SyncFlag - - -def process_config(config, args, flags): +def process_config(config, arg, args, flags): args_override = {} args_override['SYNCER_BACKEND'] = getattr(args, 'backend') - if flags & SyncFlag.RANGE: + if arg.match('range', flags): args_override['SYNCER_OFFSET'] = getattr(args, 'offset') args_override['SYNCER_LIMIT'] = getattr(args, 'until') config.dict_override(args_override, 'local cli args') - if flags & SyncFlag.HEAD: + if arg.match('head', flags): config.add(getattr(args, 'keep_alive'), '_KEEP_ALIVE') config.add(getattr(args, 'head'), '_HEAD')