View difference between Paste ID: i0J0kVAx and Ap1GZepy

 import sys
 from pywmo import Wmo
 
 wmo_name = input('Enter WMO Filename : ');
 building = Wmo(wmo_name);
 
 nLight = building.num_lights;
 
 print("There are " + str(nLight) + " lights in this WMO.");
 
 while(1):
-	command = input("What do you want to do ?\n\"a\" - Add a light\n\"q\" - Exit\n");
+	command = input("What do you want to do ?\n\"a\" - Add a light\n\"i\" - Get informations about a light\n\"q\" - Exit\n")
 
 	if(command == "q"):
-		print("Goodbye !");
+
 		while(1):
 			save = input("Do you want to save (yes/no) ? ")
 
 			if save.upper() == "YES":
 				print("Saving file")
 				building.write_file("./output/" + wmo_name)
-		flag1 = input("LightType : ");
+
-		flag2 = input("type : ");
+			elif save.upper() == "NO":
-		flag3 = input("useAtten : ");
+				print("Exiting without saving")
-		flag4 = input("pad : ");
+
 			else:
-		r = input("R : ");
+				print("You have to write \"yes\" or \"no\"");
-		g = input("G : ");
+
-		b = input("B : ");
+		print("See ya later");
-		a = input("A : ");
+
 	elif(command == "a"):
-		x = input("X : ");
+
-		y = input("Y : ");
+
-		z = input("Z : ");
+
 
-		intensity = input("intensity : ");
+		light["flag1"] = int(input("LightType : "))
-		attenuation_start = input("attenuation_start : ");
+		light["flag2"] = int(input("type : "))
-		attenuation_stop = input("attenuation_stop : ");
+		light["flag3"] = int(input("useAtten : "))
 		light["flag4"] = int(input("pad : "))
-		unknown1 = input("Unknown 1 : ");
+
-		unknown2 = input("Unknown 1 : ");
+		light["color"] = (float(input("R : ")), float(input("G : ")), float(input("B : ")), float(input("A : ")))
-		unknown3 = input("Unknown 1 : ");
+		light["position"] = (float(input("X : ")), float(input("Y : ")), float(input("Z : ")))
-		unknown4 = input("Unknown 1 : ");
+
 		light["intensity"] = float(input("intensity : "))
-		light["flag1"] = int(flag1);
+
-		light["flag2"] = int(flag2);
+		light["attenuation_start"] = float(input("attenuation_start : "))
-		light["flag3"] = int(flag3);
+		light["attenuation_stop"] = float(input("attenuation_stop : "))
-		light["flag4"] = int(flag4);
+
 		light["unknown1"] = float(input("Unknown 1 : "))
-		light["color"] = (float(r), float(g), float(b), float(a));
+		light["unknown2"] = float(input("Unknown 2 : "))
-		light["position"] = (float(x), float(y), float(z));
+		light["unknown3"] = float(input("Unknown 3 : "))
 		light["unknown4"] = float(input("Unknown 4 : "))
-		light["attenuation_start"] = float(attenuation_start);
+
-		light["attenuation_stop"] = float(attenuation_stop);
+
 		building.lights.append(light);
-		light["unknown1"] = float(unknown1);
+
-		light["unknown2"] = float(unknown2);
+		print("There are " + str(len(building.groups)) + " groups")
-		light["unknown3"] = float(unknown3);
+
-		light["unknown4"] = float(unknown4);
+		while(1):
 			group = input("In which group do you want to reference that light ( type \"s\" to return the main menu ) ? ")
 
 			if(group == "s"):
 				break;
-		print("There are " + len(building.groups) + " groups");
+
 			group = int(group)
-		while(1)
+
-		{
+			if(group >= 0 and group < len(building.groups)):
-			group = input("In which group do you want to reference that light ( type \"s\" to return the main menu )? ");
+
 				if not building.groups[group].flags & 0x200:
 					building.groups[group].flags &= 0x200
 					building.groups[group].light_refs = []
 
-			group = int(group);
+				building.groups[group].light_refs.append(building.num_lights)
 				print("Light ref added to group " + str(group))
-			if(group >= 0 && group < len(building.groups)):
+
 				print("Bad group.")
-				if(!(building.groups[group].flags & 0x200)):
+
-					building.groups[group].flags &= 0x200;
+
 		print("Light succefully added. There are now " + str(building.num_lights) + " lights.")
-				building.groups[group].light_refs.append(building.num_lights);
+
 
-				print("Bad group.");
+
-		}
+		print("Unknown command")