generate_chain.py

import argparse

from testchain.motifs.general import SetupChain, FinalizeChain
from testchain.motifs.change import Change
from testchain.motifs.motifs import Motifs
from testchain.motifs.addresses import Addresses
from testchain.motifs.special import SpecialCases
from testchain.motifs.taint import Taint
from testchain.motifs.heuristics import Heuristics
from testchain.motifs.cash import BitcoinCash
from testchain.runner import Runner

parser = argparse.ArgumentParser(description='Generate a synthetic blockchain.')
parser.add_argument('--output-dir', dest='output_dir', default="../files/", help='Output directory')
parser.add_argument('--chain', dest='chain', default="btc", help='Chain [btc, bch]')
parser.add_argument('--exec', dest='exec', default="bitcoind", help="Path to bitcoind executable")
args = parser.parse_args()
generator = Runner(args.output_dir, args.chain, args.exec)

generator.add_generator(SetupChain)
generator.add_generator(Addresses)
generator.add_generator(Motifs)
generator.add_generator(Change)
generator.add_generator(SpecialCases)
generator.add_generator(Taint)
generator.add_generator(Heuristics)
generator.add_generator(BitcoinCash)
generator.add_generator(FinalizeChain)


generator.run()