os.walk is generator

2014-12-13 15:56:17 +01:00 · 2014-12-13 15:56:17 +01:00 · f1863faf50
parent 2493a6699c
commit f1863faf50
1 changed files with 1 additions and 1 deletions
--- a/oml/update.py
+++ b/oml/update.py
@ -83,7 +83,7 @@ def download():
                current_files.add(os.path.basename(module_tar))
        with open(os.path.join(settings.updates_path, 'release.json'), 'wb') as fd:
            fd.write(release_data)
-        for f in set(os.walk(settings.updates_path).next()[2])-current_files:
+        for f in set(next(os.walk(settings.updates_path))[2])-current_files:
            os.unlink(os.path.join(settings.updates_path, f))
        return True
    return True