Bladeren bron

Fix broken replay download in Python 3

Cammy 7 jaren geleden
bovenliggende
commit
efa154b473
1 gewijzigde bestanden met toevoegingen van 7 en 1 verwijderingen
  1. 7 1
      pyinstalive/downloader.py

+ 7 - 1
pyinstalive/downloader.py

@@ -138,7 +138,13 @@ def get_replays(user_id):
 			log("", "GREEN")
 			for index, broadcast in enumerate(broadcasts):
 				exists = False
-				for directory in (os.walk(save_path).next()[1]):
+
+				if sys.version.split(' ')[0].startswith('2'):
+					directories = (os.walk(save_path).next()[1])
+				else:
+					directories = (os.walk(save_path).__next__()[1])
+
+				for directory in directories:
 					if (str(broadcast['id']) in directory) and ("_live_" not in directory):
 						log("[W] Already downloaded a replay with ID '" + str(broadcast['id']) + "', skipping...", "GREEN")
 						exists = True