forked from forks/qmk_firmware
a3e7f3e7c5
* add a test and dry-run to qmk generate-api * add a dry-run to qmk pyformat * Add a --dry-run to qmk cformat * reverse the order of nose2 and flake8 tests * run CI test against cformat and pyformat * fix programming errors * tweak job name * fix argument * refine the files we select * fix stack trace in --ci * make cformat exit clean * fix c file extensions * decouple CI from pyformat * remove --ci arg * make ci happy * use the environment var instead * change output to text * fix log message * replace tabs
27 lines
977 B
Python
Executable file
27 lines
977 B
Python
Executable file
"""Format python code according to QMK's style.
|
|
"""
|
|
from milc import cli
|
|
|
|
import subprocess
|
|
|
|
|
|
@cli.argument('-n', '--dry-run', arg_only=True, action='store_true', help="Flag only, don't automatically format.")
|
|
@cli.subcommand("Format python code according to QMK's style.", hidden=False if cli.config.user.developer else True)
|
|
def pyformat(cli):
|
|
"""Format python code according to QMK's style.
|
|
"""
|
|
edit = '--diff' if cli.args.dry_run else '--in-place'
|
|
yapf_cmd = ['yapf', '-vv', '--recursive', edit, 'bin/qmk', 'lib/python']
|
|
try:
|
|
cli.run(yapf_cmd, check=True, capture_output=False)
|
|
cli.log.info('Python code in `bin/qmk` and `lib/python` is correctly formatted.')
|
|
return True
|
|
|
|
except subprocess.CalledProcessError:
|
|
if cli.args.dry_run:
|
|
cli.log.error('Python code in `bin/qmk` and `lib/python` incorrectly formatted!')
|
|
else:
|
|
cli.log.error('Error formatting python code!')
|
|
|
|
return False
|