from block import Block
from simulation import Simulator
if __name__ == '__main__':
# Dummy node
n = Node(1, 1, True)
# Create a dummy simulator
s = Simulator(0, 0, 0, 0)
s.nodes = [n]
# Manually created blockchain
n.blocks.update({
0: Block(0, 0.0, 0, -1, 0),
1: Block(1, 0.0, 0, 0, 1),
2: Block(2, 0.0, 0, 1, 2),
3: Block(3, 5.0, 0, 2, 3),
# Fork
4: Block(4, 0.0, 0, 1, 2),
5: Block(5, 0.0, 0, 4, 3),
})
# Dump graphs
s.remove_graphs()
s.dump_node_chains()
s.prune_node_chains()
s.dump_node_chains(pruned=True)
s.convert_graphs()
P.add_argument('--network', action="store_true",
help='Dump network graph (may take a long time!)')
if __name__ == '__main__':
args = P.parse_args()
# If not given as a fraction, then assume percentage
if args.z > 1:
args.z /= 100
sim = Simulator(args.n, args.z, args.tm, args.bm)
print("\n >>>> Cleaning graphs directory ")
sim.remove_graphs()
print("\n >>>> Running simulation \n")
sim.run(args.until, args.q)
if args.network:
print("\n >>>> Dumping network graph ")
sim.dump_network()
print("\n >>>> Dumping blockchains of all nodes ")
sim.dump_node_chains()
print("\n >>>> Dumping blockchains of all nodes (pruned) ")
sim.prune_node_chains()
sim.dump_node_chains(pruned=True)
