merge back Jason change

This commit is contained in:
Tarek Ziade 2012-02-21 11:52:32 +01:00
commit 1307356ba7
2 changed files with 8 additions and 1 deletions

View file

@ -1272,6 +1272,9 @@ def process_options(arglist=None):
__version__)
parser = OptionParser(version=version,
usage="%prog [options] input ...")
parser.add_option('--builtins', default=[], action="append",
help="append builtin function (pyflakes "
"_MAGIC_GLOBALS)")
parser.add_option('--max-complexity', default=-1, action='store',
type='int', help="McCabe complexity treshold")
parser.add_option('-v', '--verbose', default=0, action='count',

View file

@ -47,14 +47,18 @@ def _get_python_files(paths):
def main():
options, args = pep8.process_options()
complexity = options.max_complexity
builtins = set(options.builtins)
warnings = 0
if builtins:
orig_builtins = set(pyflakes._MAGIC_GLOBALS)
pyflakes._MAGIC_GLOBALS = orig_builtins | builtins
if args:
for path in _get_python_files(args):
warnings += check_file(path, complexity)
else:
stdin = sys.stdin.read()
warnings += check_code(stdin, complexity)
raise SystemExit(warnings > 0)