commit 5e5e4bcf482b961dfa8beaeef48a4feeae8bc67f Author: Karsten Loesing karsten.loesing@gmx.net Date: Tue May 26 08:58:01 2020 +0200
Work around Networkx's API change.
Turns out that Networkx renamed the node attribute to nodes some time between version 2.2.1 and 2.4. This workaround avoids accessing that attribute to add another key/value pair to an existing node and instead uses two different key/value pair sets when adding the node in the first place. Not pretty, but works around the issue.
Fixes #34298. --- onionperf/model.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/onionperf/model.py b/onionperf/model.py index cd44118..1f4d9fe 100644 --- a/onionperf/model.py +++ b/onionperf/model.py @@ -69,9 +69,10 @@ class TorperfModel(GeneratableTGenModel): server_str = ','.join(self.tgen_servers) g = DiGraph()
- g.add_node("start", serverport=self.tgen_port, peers=server_str, loglevel="info", heartbeat="1 minute") if self.socksproxy is not None: - g.node["start"]["socksproxy"] = self.socksproxy + g.add_node("start", serverport=self.tgen_port, peers=server_str, loglevel="info", heartbeat="1 minute", socksproxy=self.socksproxy) + else: + g.add_node("start", serverport=self.tgen_port, peers=server_str, loglevel="info", heartbeat="1 minute") g.add_node("pause", time="5 minutes") g.add_node("transfer5m", type="get", protocol="tcp", size="5 MiB", timeout="270 seconds", stallout="0 seconds")
@@ -98,9 +99,10 @@ class OneshotModel(GeneratableTGenModel): server_str = ','.join(self.tgen_servers) g = DiGraph()
- g.add_node("start", serverport=self.tgen_port, peers=server_str, loglevel="info", heartbeat="1 minute") if self.socksproxy is not None: - g.node["start"]["socksproxy"] = self.socksproxy + g.add_node("start", serverport=self.tgen_port, peers=server_str, loglevel="info", heartbeat="1 minute", socksproxy=self.socksproxy) + else: + g.add_node("start", serverport=self.tgen_port, peers=server_str, loglevel="info", heartbeat="1 minute") g.add_node("transfer5m", type="get", protocol="tcp", size="5 MiB", timeout="15 seconds", stallout="10 seconds")
g.add_edge("start", "transfer5m")