Tweak linearize.py to give more flexibility

Add the ability to start at a non-zero height, and allow for appending to
an existing file.
This commit is contained in:
Micha 2014-01-12 13:48:39 +02:00
parent 0e469b5167
commit 7a29fb5940
Failed to extract signature

View File

@ -73,9 +73,9 @@ def get_blocks(settings):
rpc = BitcoinRPC(settings['host'], settings['port'], rpc = BitcoinRPC(settings['host'], settings['port'],
settings['rpcuser'], settings['rpcpass']) settings['rpcuser'], settings['rpcpass'])
outf = open(settings['output'], 'wb') outf = open(settings['output'], 'ab')
for height in xrange(settings['max_height']+1): for height in xrange(settings['min_height'], settings['max_height']+1):
data = getblock(rpc, settings, height) data = getblock(rpc, settings, height)
outhdr = settings['netmagic'] outhdr = settings['netmagic']
@ -114,6 +114,8 @@ if __name__ == '__main__':
settings['host'] = '127.0.0.1' settings['host'] = '127.0.0.1'
if 'port' not in settings: if 'port' not in settings:
settings['port'] = 8332 settings['port'] = 8332
if 'min_height' not in settings:
settings['min_height'] = 0
if 'max_height' not in settings: if 'max_height' not in settings:
settings['max_height'] = 250000 settings['max_height'] = 250000
if 'rpcuser' not in settings or 'rpcpass' not in settings: if 'rpcuser' not in settings or 'rpcpass' not in settings:
@ -122,6 +124,7 @@ if __name__ == '__main__':
settings['netmagic'] = settings['netmagic'].decode('hex') settings['netmagic'] = settings['netmagic'].decode('hex')
settings['port'] = int(settings['port']) settings['port'] = int(settings['port'])
settings['min_height'] = int(settings['min_height'])
settings['max_height'] = int(settings['max_height']) settings['max_height'] = int(settings['max_height'])
get_blocks(settings) get_blocks(settings)