diff options
| author | skullY <[email protected]> | 2019-08-22 13:30:50 -0700 | 
|---|---|---|
| committer | skullydazed <[email protected]> | 2019-08-31 08:50:25 -0700 | 
| commit | 1784d1bfac44a63bf343b6e2098f0cba81d58cb2 (patch) | |
| tree | 4c6410809c9fd54b064b7f5c79726a4558d119f2 /lib/python | |
| parent | 95477749629407e2a9e33c6ccf26ecc8b24ab07a (diff) | |
| download | qmk_firmware-1784d1bfac44a63bf343b6e2098f0cba81d58cb2.tar.gz qmk_firmware-1784d1bfac44a63bf343b6e2098f0cba81d58cb2.zip | |
Add support for passing files at the command line
Diffstat (limited to 'lib/python')
| -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 [email protected]('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 | 
