Optimise file walking

qmk · Jan 19, 2020 · 7b97f2d · 7b97f2d
1 parent e267bc5
commit 7b97f2d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/python/qmk/cli/cformat.py b/lib/python/qmk/cli/cformat.py
@@ -24,14 +24,14 @@ def cformat(cli):
     if cli.args.files:
         cli.args.files = [os.path.join(os.environ['ORIG_CWD'], file) for file in cli.args.files]
     else:
+        ignores = ['tmk_core/protocol/usb_hid', 'quantum/template']
         for dir in ['drivers', 'quantum', 'tests', 'tmk_core']:
             for dirpath, dirnames, filenames in os.walk(dir):
-                ignores = ['tmk_core/protocol/usb_hid', 'quantum/template']
                 if any(i in dirpath for i in ignores):
-                    continue
+                    dirnames.clear()
 
                 for name in filenames:
-                    if name.endswith('.c') or name.endswith('.h') or name.endswith('.cpp'):
+                    if name.endswith(('.c', '.h', '.cpp')):
                         cli.args.files.append(os.path.join(dirpath, name))
 
     # Run clang-format on the files we've found