logos-blockchain-specs/mixnet/v2/sim/simulation.py

import simpy

from node import Node
from p2p import P2p


class Simulation:
    def __init__(self, num_nodes: int, node_params: Node.Parameters):
        self.env = simpy.Environment()
        self.p2p = P2p(self.env)
        self.nodes = [Node(i, self.env, self.p2p, node_params) for i in range(num_nodes)]
        self.p2p.add_node(self.nodes)

    def run(self, until):
        self.env.run(until=until)