Bypass IndexError when running without having a file in each slot for all categories

This commit is contained in:
2023-01-16 16:16:22 -05:00
parent b6da652914
commit a468bfb83c
+9 -1
View File
@@ -19,6 +19,7 @@ sorted_dir = args.sorted_dir[0].rstrip('/') + '/'
for i in range(1, 9):
# national
options = [f.path for f in scandir(sorted_dir + 'National/' + str(i)) if f.is_dir()]
try:
track = choice(options)
if islink(f"{database_dir}Beta_Nat_Track_Slot_{str(i)}.dx9.database"):
remove(f"{database_dir}Beta_Nat_Track_Slot_{str(i)}.dx9.database")
@@ -36,8 +37,11 @@ for i in range(1, 9):
f"{database_dir}Beta_Nat_Track_Slot_{str(i)}.dx9.package")
symlink(f"{track}/Beta_Nat_Track_Slot_{str(i)}.dx9.scene",
f"{database_dir}Beta_Nat_Track_Slot_{str(i)}.dx9.scene")
except IndexError:
pass
# supercross
options = [f.path for f in scandir(sorted_dir + 'Supercross/' + str(i)) if f.is_dir()]
try:
track = choice(options)
if islink(f"{database_dir}Beta_Sx_Track_Slot_{str(i)}.dx9.database"):
remove(f"{database_dir}Beta_Sx_Track_Slot_{str(i)}.dx9.database")
@@ -55,8 +59,11 @@ for i in range(1, 9):
f"{database_dir}Beta_Sx_Track_Slot_{str(i)}.dx9.package")
symlink(f"{track}/Beta_Sx_Track_Slot_{str(i)}.dx9.scene",
f"{database_dir}Beta_Sx_Track_Slot_{str(i)}.dx9.scene")
except IndexError:
pass
# free ride
options = [f.path for f in scandir(sorted_dir + 'Free Ride/' + str(i)) if f.is_dir()]
try:
track = choice(options)
if islink(f"{database_dir}Beta_Track_Slot_{str(i)}.dx9.database"):
remove(f"{database_dir}Beta_Track_Slot_{str(i)}.dx9.database")
@@ -74,4 +81,5 @@ for i in range(1, 9):
f"{database_dir}Beta_Track_Slot_{str(i)}.dx9.package")
symlink(f"{track}/Beta_Track_Slot_{str(i)}.dx9.scene",
f"{database_dir}Beta_Track_Slot_{str(i)}.dx9.scene")
except IndexError:
pass