diff --git a/render.py b/render.py index 484adb6..ecbe766 100755 --- a/render.py +++ b/render.py @@ -77,9 +77,10 @@ else: if not os.path.exists('DRONES.json'): DRONES = defaultdict(list) - for letter in os.listdir('drones'): - for fn in sorted(os.listdir(os.path.join('drones', letter))): - path = os.path.join('drones', letter, fn) + prefix = 'drones' + for letter in os.listdir(prefix): + for fn in sorted(os.listdir(os.path.join(prefix, letter))): + path = os.path.join(prefix, letter, fn) DRONES[letter[0]].append({ 'path': path, 'duration': ox.avinfo(path)['duration']