scripts/src_util/check_whitespace.py

   1 # -*- coding: utf-8 -*-
   2
   3 #-------------------------------------------------------------------------
   4 # drawElements Quality Program utilities
   5 # --------------------------------------
   6 #
   7 # Copyright 2015 The Android Open Source Project
   8 #
   9 # Licensed under the Apache License, Version 2.0 (the "License");
  10 # you may not use this file except in compliance with the License.
  11 # You may obtain a copy of the License at
  12 #
  13 #      http://www.apache.org/licenses/LICENSE-2.0
  14 #
  15 # Unless required by applicable law or agreed to in writing, software
  16 # distributed under the License is distributed on an "AS IS" BASIS,
  17 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18 # See the License for the specific language governing permissions and
  19 # limitations under the License.
  20 #
  21 #-------------------------------------------------------------------------
  22
  23 import sys
  24 from argparse import ArgumentParser
  25 from common import getChangedFiles, getAllProjectFiles, isTextFile
  26
  27 def checkFileWhitespace (file):
  28     f = open(file, 'rb')
  29     error = False
  30     for lineNum, line in enumerate(f):
  31         if line.endswith(" \n") or line.endswith("\t\n"):
  32             error = True
  33             print "%s:%i trailing whitespace" % (file, lineNum+1)
  34         if " \t" in line:
  35             error = True
  36             print "%s:%i merged <space><tab>" % (file, lineNum+1)
  37         if line.endswith("\r") or line.endswith("\r\n"):
  38             error = True
  39             print "%s:%i incorrect line ending" % (file, lineNum+1)
  40     f.close()
  41
  42     return not error
  43
  44 def checkWhitespace (files):
  45     error = False
  46     for file in files:
  47         if isTextFile(file):
  48             if not checkFileWhitespace(file):
  49                 error = True
  50
  51     return not error
  52
  53 if __name__ == "__main__":
  54     parser = ArgumentParser()
  55     parser.add_argument("-e", "--only-errors",  action="store_true", dest="onlyErrors",   default=False, help="Print only on error")
  56     parser.add_argument("-i", "--only-changed", action="store_true", dest="useGitIndex",  default=False, help="Check only modified files. Uses git.")
  57
  58     args = parser.parse_args()
  59
  60     if args.useGitIndex:
  61         files = getChangedFiles()
  62     else:
  63         files = getAllProjectFiles()
  64
  65     error = not checkWhitespace(files)
  66
  67     if error:
  68         print "One or more checks failed"
  69         sys.exit(1)
  70     if not args.onlyErrors:
  71         print "All checks passed"