Snap for 4557233 from e4158b37f3394e0398050be9e31d206063282bc6 to pi-release

Change-Id: I0116f29278aa1ceb2df573141e7552167e0a19ad

1 files changed, 8 insertions, 3 deletions

diff --git a/ktlint.py b/ktlint.py
index fa9ea0e..697a0d6 100755
--- a/ktlint.py
+++ b/ktlint.py
@@ -31,15 +31,20 @@ KTLINT_JAR = os.path.join(MAIN_DIRECTORY, 'ktlint-android-all.jar')
 def main(args=None):
   parser = argparse.ArgumentParser()
   parser.add_argument('--file', '-f', nargs='*')
+  parser.add_argument('--format', '-F', dest='format', action='store_true')
+  parser.add_argument('--noformat', dest='format', action='store_false')
+  parser.set_defaults(format=False)
   args = parser.parse_args()
-  kotlin_files = [f for f in args.file if f.endswith('.kt')]
-  if not kotlin_files:
+  ktlint_args = [f for f in args.file if f.endswith('.kt')]
+  if args.format:
+    ktlint_args += ['-F']
+  if not ktlint_args:
     sys.exit(0)
 
   ktlint_env = os.environ.copy()
   ktlint_env['JAVA_CMD'] = 'java'
   try:
-    check = subprocess.Popen(['java', '-jar', KTLINT_JAR] + kotlin_files,
+    check = subprocess.Popen(['java', '-jar', KTLINT_JAR] + ktlint_args,
                              stdout=subprocess.PIPE, env=ktlint_env)
     stdout, _ = check.communicate()
     if stdout: