summaryrefslogtreecommitdiff
path: root/lib/python
diff options
context:
space:
mode:
authorskullY <skullydazed@gmail.com>2019-08-22 13:30:50 -0700
committerFlorian Didron <fdidron@users.noreply.github.com>2019-09-26 10:08:39 +0900
commitffca0ecd0a847e33a9215b5fd33075511b49a787 (patch)
tree4c89ccb44004a6f0f7ecb4e3cc04d0b894b8062a /lib/python
parent56443ee486474b6375b60f753b9a90890e9856df (diff)
Add support for passing files at the command line
Diffstat (limited to 'lib/python')
-rw-r--r--lib/python/qmk/cli/cformat.py20
1 files changed, 11 insertions, 9 deletions
diff --git a/lib/python/qmk/cli/cformat.py b/lib/python/qmk/cli/cformat.py
index f7020f4c5f..0c209247d9 100644
--- a/lib/python/qmk/cli/cformat.py
+++ b/lib/python/qmk/cli/cformat.py
@@ -6,22 +6,24 @@ import subprocess
from milc import cli
+@cli.argument('files', nargs='*', help='Filename(s) to format.')
@cli.entrypoint("Format C code according to QMK's style.")
def main(cli):
"""Format C code according to QMK's style.
"""
clang_format = ['clang-format', '-i']
- code_files = []
- for dir in ['drivers', 'quantum', 'tests', 'tmk_core']:
- for dirpath, dirnames, filenames in os.walk(dir):
- if 'tmk_core/protocol/usb_hid' in dirpath:
- continue
- for name in filenames:
- if name.endswith('.c') or name.endswith('.h') or name.endswith('.cpp'):
- code_files.append(os.path.join(dirpath, name))
+ if not cli.args.files:
+ for dir in ['drivers', 'quantum', 'tests', 'tmk_core']:
+ for dirpath, dirnames, filenames in os.walk(dir):
+ if 'tmk_core/protocol/usb_hid' in dirpath:
+ continue
+ for name in filenames:
+ if name.endswith('.c') or name.endswith('.h') or name.endswith('.cpp'):
+ cli.args.files.append(os.path.join(dirpath, name))
try:
- subprocess.run(clang_format + code_files, check=True)
+ subprocess.run(clang_format + cli.args.files, check=True)
cli.log.info('Successfully formatted the C code.')
except subprocess.CalledProcessError:
cli.log.error('Error formatting C code!')
+ return False