prototypes: ouroboros: Use zip function

This uses the zip function to run through the different lists in
bootstrap_prototype, since it is a more elegant solution.

1 files changed, 6 insertions, 5 deletions

diff --git a/rumba/prototypes/ouroboros.py b/rumba/prototypes/ouroboros.py
index cb0bf08..716f1b8 100644
--- a/rumba/prototypes/ouroboros.py
+++ b/rumba/prototypes/ouroboros.py
@@ -148,11 +148,12 @@ class Experiment(mod.Experiment):
         logger.info("Creating IPCPs")
         self.create_ipcps()
         logger.info("Enrolling IPCPs...")
-        i = 0
-        for element in self.enrollments:
+
+        for element, mgmt, dt in zip(self.enrollments,
+                                     self.mgmt_flows,
+                                     self.dt_flows):
             self.enroll_dif(element)
-            self.setup_flows(self.mgmt_flows[i], comp="mgmt")
-            self.setup_flows(self.dt_flows[i], comp="dt")
-            i = i + 1
+            self.setup_flows(mgmt, comp="mgmt")
+            self.setup_flows(dt, comp="dt")
 
         logger.info("All done, have fun!")