diff options
author | skullY <skullydazed@gmail.com> | 2019-08-22 13:30:50 -0700 |
---|---|---|
committer | Florian Didron <fdidron@users.noreply.github.com> | 2019-09-26 10:08:39 +0900 |
commit | ffca0ecd0a847e33a9215b5fd33075511b49a787 (patch) | |
tree | 4c89ccb44004a6f0f7ecb4e3cc04d0b894b8062a /lib/python/qmk/cli | |
parent | 56443ee486474b6375b60f753b9a90890e9856df (diff) |
Add support for passing files at the command line
Diffstat (limited to 'lib/python/qmk/cli')
-rw-r--r-- | lib/python/qmk/cli/cformat.py | 20 |
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 |