From b353028ea5e0ac7b69e7fed4f224c7cb07015ec9 Mon Sep 17 00:00:00 2001
From: Akaash Suresh <casa.akaash@gmail.com>
Date: Sat, 22 Feb 2020 22:57:19 -0600
Subject: New functionality for cformat (#7893)

Fixing complexity

remove lambda

PR review fixes #1

Removing unneccesary string substitution

Handle -a and specified files

Complexity rewrite, use pathlib
---
 lib/python/qmk/path.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'lib/python/qmk/path.py')

diff --git a/lib/python/qmk/path.py b/lib/python/qmk/path.py
index d16928afb5..bfaa439249 100644
--- a/lib/python/qmk/path.py
+++ b/lib/python/qmk/path.py
@@ -68,3 +68,17 @@ def normpath(path):
         return Path(path)
 
     return Path(os.environ['ORIG_CWD']) / path
+
+
+def c_source_files(dir_names):
+    """Returns a list of all *.c, *.h, and *.cpp files for a given list of directories
+
+    Args:
+
+        dir_names
+            List of directories, relative pathing starts at qmk's cwd
+    """
+    files = []
+    for dir in dir_names:
+        files.extend(file for file in Path(dir).glob('**/*') if file.suffix in ['.c', '.h', '.cpp'])
+    return files
-- 
cgit v1.2.3