45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
|
#!/awips2/python/bin/python
|
||
|
|
#
|
||
|
|
# find_divs.py
|
||
|
|
# Finds division operators in python files.
|
||
|
|
#
|
||
|
|
# Author: tgurney
|
||
|
|
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
import ast
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import traceback
|
||
|
|
|
||
|
|
if len(sys.argv) < 2:
|
||
|
|
print("Usage: " + sys.argv[0] + " DIR...")
|
||
|
|
print()
|
||
|
|
print("Finds and prints all division operators in Python files in the specified DIRs.")
|
||
|
|
print("This program rejects Python files with invalid syntax. " +
|
||
|
|
"A traceback will get printed to stderr for every such file.")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
def parse_file(path):
|
||
|
|
with open(path) as f:
|
||
|
|
contents = f.read()
|
||
|
|
tree = ast.parse(contents, os.path.abspath(path))
|
||
|
|
last_lineno = None
|
||
|
|
lines = None
|
||
|
|
for node in ast.walk(tree):
|
||
|
|
if hasattr(node, 'lineno'):
|
||
|
|
last_lineno = node.lineno
|
||
|
|
if isinstance(node, ast.Div):
|
||
|
|
if not lines:
|
||
|
|
lines = contents.split('\n')
|
||
|
|
print(path + ':' + str(last_lineno) + ':' + lines[last_lineno-1])
|
||
|
|
|
||
|
|
for arg in sys.argv[1:]:
|
||
|
|
for root, _, filenames in os.walk(arg):
|
||
|
|
for filename in filenames:
|
||
|
|
if filename.endswith('.py'):
|
||
|
|
try:
|
||
|
|
parse_file(os.path.join(root, filename))
|
||
|
|
except Exception as e:
|
||
|
|
sys.stderr.write(traceback.format_exc() + '\n')
|