Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Python: align style with PEP-008
[simgrid.git] / examples / java / bittorrent / generate.py
index 155957f..d416c5f 100755 (executable)
 # Usage: python generate.py nb_nodes nb_bits end_date percentage
 # Example: python generate.py 10000 5000
 
-import sys, random
+import sys
+import random
 
 if len(sys.argv) != 4:
-       print("Usage: python generate.py nb_nodes end_date seed_percentage > deployment_file.xml")
-       sys.exit(1)
+    print(
+        "Usage: python generate.py nb_nodes end_date seed_percentage > deployment_file.xml")
+    sys.exit(1)
 
 nb_nodes = int(sys.argv[1])
 end_date = int(sys.argv[2])
-seed_percentage = int(sys.argv[3]);
+seed_percentage = int(sys.argv[3])
 
 nb_bits = 24
 max_id = 2 ** nb_bits - 1
 all_ids = [42]
 
 sys.stdout.write("<?xml version='1.0'?>\n"
-"<!DOCTYPE platform SYSTEM \"http://simgrid.gforge.inria.fr/simgrid.dtd\">\n"
-"<platform version=\"3\">\n"
-"  <process host=\"c-0.me\" function=\"bittorrent.Tracker\"><argument value=\"%d\"/></process>\n" % end_date)
+                 "<!DOCTYPE platform SYSTEM \"http://simgrid.gforge.inria.fr/simgrid.dtd\">\n"
+                 "<platform version=\"3\">\n"
+                 "  <process host=\"c-0.me\" function=\"bittorrent.Tracker\"><argument value=\"%d\"/></process>\n" % end_date)
 
 for i in range(1, nb_nodes):
 
-  ok = False
-  while not ok:
-    my_id = random.randint(0, max_id)
-    ok = not my_id in all_ids
-  start_date = i * 10
-  line = "  <process host=\"c-%d.me\" function=\"bittorrent.Peer\"><argument value=\"%d\" /><argument value=\"%d\" />" % (i, my_id, end_date)
-  if random.randint(0,100) < seed_percentage:
-    line += "<argument value=\"1\" />"
-  line += "</process>\n";
-  sys.stdout.write(line)
-  all_ids.append(my_id)
+    ok = False
+    while not ok:
+        my_id = random.randint(0, max_id)
+        ok = not my_id in all_ids
+    start_date = i * 10
+    line = "  <process host=\"c-%d.me\" function=\"bittorrent.Peer\"><argument value=\"%d\" /><argument value=\"%d\" />" % (
+        i, my_id, end_date)
+    if random.randint(0, 100) < seed_percentage:
+        line += "<argument value=\"1\" />"
+    line += "</process>\n"
+    sys.stdout.write(line)
+    all_ids.append(my_id)
 sys.stdout.write("</platform>")
-