Format all files

kfrlib · Jul 20, 2016 · ff2dc29 · ff2dc29
1 parent 846a125
commit ff2dc29
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/format-all.py b/format-all.py
@@ -9,14 +9,13 @@
 
 path = os.path.dirname(os.path.realpath(__file__))
 
+masks = ['*.hpp', '*.h', '*.cpp', '*.c', '*.cxx']
+
 filenames = []
 for root, dirnames, files in os.walk(path, path):
-    for filename in fnmatch.filter(files, '*.hpp'):
-        filenames.append(os.path.join(root, filename))
-    for filename in fnmatch.filter(files, '*.h'):
-        filenames.append(os.path.join(root, filename))
-    for filename in fnmatch.filter(files, '*.cpp'):
-        filenames.append(os.path.join(root, filename))
+    for mask in masks:
+        for filename in fnmatch.filter(files, mask):
+            filenames.append(os.path.join(root, filename))
 
 for filename in filenames:
     print( filename, '...' )