1*e7776783SApple OSS Distributions#!/usr/bin/env python 2*e7776783SApple OSS Distributionsfrom __future__ import absolute_import, print_function 3*e7776783SApple OSS Distributions 4*e7776783SApple OSS Distributions 5*e7776783SApple OSS Distributionshelpdoc = """ 6*e7776783SApple OSS DistributionsA simple utility that verifies the syntax for python scripts. 7*e7776783SApple OSS DistributionsThe checks it does are : 8*e7776783SApple OSS Distributions * Check for 'tab' characters in .py files 9*e7776783SApple OSS Distributions * Compile errors in py sources 10*e7776783SApple OSS DistributionsUsage: 11*e7776783SApple OSS Distributions python syntax_checker.py <python_source_file> [<python_source_file> ..] 12*e7776783SApple OSS Distributions""" 13*e7776783SApple OSS Distributionsimport sys 14*e7776783SApple OSS Distributionsimport os 15*e7776783SApple OSS Distributionsimport re 16*e7776783SApple OSS Distributions 17*e7776783SApple OSS Distributionstabs_search_rex = re.compile("^\s*\t+",re.MULTILINE|re.DOTALL) 18*e7776783SApple OSS Distributions 19*e7776783SApple OSS Distributionsdef find_non_ascii(s): 20*e7776783SApple OSS Distributions for c in s: 21*e7776783SApple OSS Distributions if ord(c) >= 0x80: return True 22*e7776783SApple OSS Distributions return False 23*e7776783SApple OSS Distributions 24*e7776783SApple OSS Distributionsif __name__ == "__main__": 25*e7776783SApple OSS Distributions if len(sys.argv) < 2: 26*e7776783SApple OSS Distributions print("Error: Unknown arguments", file=sys.stderr) 27*e7776783SApple OSS Distributions print(helpdoc) 28*e7776783SApple OSS Distributions sys.exit(1) 29*e7776783SApple OSS Distributions for fname in sys.argv[1:]: 30*e7776783SApple OSS Distributions if not os.path.exists(fname): 31*e7776783SApple OSS Distributions print("Error: Cannot recognize %s as a file" % fname, file=sys.stderr) 32*e7776783SApple OSS Distributions sys.exit(1) 33*e7776783SApple OSS Distributions if fname.split('.')[-1] != 'py': 34*e7776783SApple OSS Distributions print("Note: %s is not a valid python file. Skipping." % fname) 35*e7776783SApple OSS Distributions continue 36*e7776783SApple OSS Distributions fh = open(fname) 37*e7776783SApple OSS Distributions strdata = fh.readlines() 38*e7776783SApple OSS Distributions lineno = 0 39*e7776783SApple OSS Distributions syntax_fail = False 40*e7776783SApple OSS Distributions for linedata in strdata: 41*e7776783SApple OSS Distributions lineno += 1 42*e7776783SApple OSS Distributions if len(tabs_search_rex.findall(linedata)) > 0 : 43*e7776783SApple OSS Distributions print("Error: Found a TAB character at %s:%d" % (fname, lineno), file=sys.stderr) 44*e7776783SApple OSS Distributions syntax_fail = True 45*e7776783SApple OSS Distributions if find_non_ascii(linedata): 46*e7776783SApple OSS Distributions print("Error: Found a non ascii character at %s:%d" % (fname, lineno), file=sys.stderr) 47*e7776783SApple OSS Distributions syntax_fail = True 48*e7776783SApple OSS Distributions if syntax_fail: 49*e7776783SApple OSS Distributions print("Error: Syntax check failed. Please fix the errors and try again.", file=sys.stderr) 50*e7776783SApple OSS Distributions sys.exit(1) 51*e7776783SApple OSS Distributions #now check for error in compilation 52*e7776783SApple OSS Distributions try: 53*e7776783SApple OSS Distributions with open(fname, 'r') as file: 54*e7776783SApple OSS Distributions source = file.read() + '\n' 55*e7776783SApple OSS Distributions compile_result = compile(source, fname, 'exec') 56*e7776783SApple OSS Distributions except Exception as exc: 57*e7776783SApple OSS Distributions print(str(exc), file=sys.stderr) 58*e7776783SApple OSS Distributions print("Error: Compilation failed. Please fix the errors and try again.", file=sys.stderr) 59*e7776783SApple OSS Distributions sys.exit(1) 60*e7776783SApple OSS Distributions print("Success: Checked %s. No syntax errors found." % fname) 61*e7776783SApple OSS Distributions sys.exit(0) 62*e7776783SApple OSS Distributions 63