readability enhancements

master
skullY 2019-08-22 13:33:34 -07:00 committed by skullydazed
parent 1784d1bfac
commit 2d688ad14e
1 changed files with 5 additions and 0 deletions

View File

@ -12,18 +12,23 @@ def main(cli):
"""Format C code according to QMK's style. """Format C code according to QMK's style.
""" """
clang_format = ['clang-format', '-i'] clang_format = ['clang-format', '-i']
# Find the list of files to format
if not cli.args.files: if not cli.args.files:
for dir in ['drivers', 'quantum', 'tests', 'tmk_core']: for dir in ['drivers', 'quantum', 'tests', 'tmk_core']:
for dirpath, dirnames, filenames in os.walk(dir): for dirpath, dirnames, filenames in os.walk(dir):
if 'tmk_core/protocol/usb_hid' in dirpath: if 'tmk_core/protocol/usb_hid' in dirpath:
continue continue
for name in filenames: for name in filenames:
if name.endswith('.c') or name.endswith('.h') or name.endswith('.cpp'): if name.endswith('.c') or name.endswith('.h') or name.endswith('.cpp'):
cli.args.files.append(os.path.join(dirpath, name)) cli.args.files.append(os.path.join(dirpath, name))
# Run clang-format on the files we've found
try: try:
subprocess.run(clang_format + cli.args.files, check=True) subprocess.run(clang_format + cli.args.files, check=True)
cli.log.info('Successfully formatted the C code.') cli.log.info('Successfully formatted the C code.')
except subprocess.CalledProcessError: except subprocess.CalledProcessError:
cli.log.error('Error formatting C code!') cli.log.error('Error formatting C code!')
return False return False