Pārlūkot izejas kodu

Fix broken replay download in Python 3

Cammy 7 gadi atpakaļ
vecāks
revīzija
efa154b473
1 mainītis faili ar 7 papildinājumiem un 1 dzēšanām
  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