summaryrefslogtreecommitdiff
path: root/lib/python/qmk/cli/pyformat.py
diff options
context:
space:
mode:
authorskullY <skullydazed@gmail.com>2019-08-22 09:40:12 -0700
committerskullydazed <skullydazed@users.noreply.github.com>2019-09-07 07:58:41 -0700
commitdeb6fa6a87b12fcdbf577837c6faeb854cd287c7 (patch)
tree59037fc2c2bb3f096caa7bce3a19261896b2a218 /lib/python/qmk/cli/pyformat.py
parent533d6d6a464d41d23a39cecfe42d95d2e400d335 (diff)
Add a command to format python code
Diffstat (limited to 'lib/python/qmk/cli/pyformat.py')
-rwxr-xr-xlib/python/qmk/cli/pyformat.py16
1 files changed, 16 insertions, 0 deletions
diff --git a/lib/python/qmk/cli/pyformat.py b/lib/python/qmk/cli/pyformat.py
new file mode 100755
index 0000000000..b1f8c02b28
--- /dev/null
+++ b/lib/python/qmk/cli/pyformat.py
@@ -0,0 +1,16 @@
+"""Format python code according to QMK's style.
+"""
+from milc import cli
+
+import subprocess
+
+
+@cli.entrypoint("Format python code according to QMK's style.")
+def main(cli):
+ """Format python code according to QMK's style.
+ """
+ try:
+ subprocess.run(['yapf', '-vv', '-ri', 'bin/qmk', 'lib/python'], check=True)
+ cli.log.info('Successfully formatted the python code in `bin/qmk` and `lib/python`.')
+ except subprocess.CalledProcessError:
+ cli.log.error('Error formatting python code!')