Update second version(2.0).py

second version(2.0).py

@@ -8,43 +8,49 @@ f4 = ["ap13", "ap14", "ap15", "ap16"]
 building = [f1 , f2, f3, f4]
 bombs = []
 
-bq = random.randint(1,10)
+bq = int(input("How many bombs do you desire? "))
+print("please answer the following with yes or no")
+twoin1 = str(input("Do you want more then one bomb in an apartment? "))
+
+twoin1 =twoin1.lower()
+#I made it possible for the user to choose how many bombs he wants to be generated, and if he wants
+#to have the possibility for there to be more then one bomb in an apartment
 g = 1
 
 for g in range(bq):
-    a = random.randint(0,3)
-    b = random.randint(0,3)
+    a = random.randint(0, 3)
+    b = random.randint(0, 3)
     bl = building[a][b]
     bombs.append(bl)
-    bombs2 = set(dict.fromkeys(bombs))#I was testing with lists , then had to make a dictionary to remove duplicates,
-    # but for the next version i will use SETS
-    size = len(bombs2)
-    #print(bl, a, b, bq)
-print(bombs)
-print(bombs2)
+        # I was testing with lists , then had to make a dictionary to remove duplicates,
+        # but for the next version i will use SETS
 reroll = []
-while size != bq:
-    r = bq - size
-    for g in range(r):
-        c = random.randint(0,3)
-        d = random.randint(0,3)
-        bl2 = building[c][d]
-        reroll.append(bl2)
-        reroll2 = set(reroll)
-        size2 = len(reroll2)
-        sf = len(reroll2.intersection(bombs2))
-        print(reroll, r, bq, size, sf)
-        if sf >= 1:
-            reroll.clear()
-        if size2 == r and sf == 0:
-            bombsf = bombs2.union(reroll2)
-            break
-    else:
-        continue
-    break
+if twoin1 == "no":
+    bombs2 = set(dict.fromkeys(bombs))
+    size = len(bombs2)
+    while size != bq:
+        r = bq - size
+        for g in range(r):
+            c = random.randint(0, 3)
+            d = random.randint(0, 3)
+            bl2 = building[c][d]
+            reroll.append(bl2)
+            reroll2 = set(reroll)
+            size2 = len(reroll2)
+            sf = len(reroll2.intersection(bombs2))
+            print(reroll, r, bq, size, sf)
+            if sf >= 1:
+                reroll.clear()
+            if size2 == r and sf == 0:
+                bombsf = bombs2.union(reroll2)
+                break
+        else:
+            continue
+        break
+else:pass
 #I was able to make this verification to work, thus my code can be true to the bomb quantity that is randomly generated, now i need to make the floor verification
 #and make it so the user determines how many bombs will be generated
-exit()
+exit() #I will now star to program the next section of my challenge and i will upload it as a 3.0
 #I put this exit() to do some testing so please ignore what comes in the next blocks of code since i am testing a way
 # to generate random bombs and avoid duplicates
 #After achieving this i will make thhe user input so you may choose how many bombs you want, and use a bollean to