def __init__(self, text=None, filename=None):
if filename == None and text == None:
raise AttributeError("need a filename or text for a Code object")
return
if filename:
self.filename = filename
self.text = get_text(filename)
elif text:
self.text = text
self.lines = self.text.split("\n")
self.linebreak_indices = get_linebreak_indices(self.text)
self.levels = get_indent_levels(self.lines)
self.ignore_lines = get_ignore_lines(self.lines)
self.stripped = [] # lines without whitespace or comments
for line_no, line in enumerate(self.lines):
if self.ignore_lines[line_no]:
self.stripped.append("")
else:
self.stripped.append(strip(line))
def show_indent(f):
"""Utility function to show the expected and actual indents for a file"""
code = get_text(f)
lines = code.split("\n")
levels = get_indent_levels(lines)
errors = FeatureIndentation(code)
ignore = get_ignore_lines(lines)
for line_no, line in enumerate(lines):
if line_no in errors:
print
print
for i in range(line_no - 3, line_no):
print str(levels[i]) + "\t" + lines[i]
print str(levels[line_no]) + "\t" + line
print " ^^^ "
print errors[line_no]
print
for i in range(line_no+1, line_no+3):
print str(levels[i]) + "\t" + lines[i]
print
print
def compute(self, code, annotations):
indentations = {} # how far indented levels should be
lines = code.lines
levels = get_indent_levels(lines)
# find the distribution of indentations for each indent level
for line_no, line in enumerate(lines):
_levels = levels[line_no]
if _levels == None: # lines ignored because they are comments
continue
indent = get_indent(line)
for level in _levels:
if level not in indentations:
indentations[level] = [indent]
else:
indentations[level].append(indent)
# find the most common indentations for each indent level
# ensuring that each successive level is deeper
try:
common_indentation = {}
for level, indents in indentations.iteritems():
s = set(indents)
if level == 0:
common_indentation[level] = max(s, key=indents.count)
else:
last_indentation = common_indentation[level - 1]
most_common = max(s, key=indents.count)
# if indent is not deeper
if most_common <= last_indentation:
# TODO(fil): pass error when indent is not deeper
while most_common <= last_indentation:
s -= set([most_common])
if len(s) == 0:
most_common = max(set(indents), key=indents.count)
break
most_common = max(s, key=indents.count)
common_indentation[level] = most_common
except KeyError as e:
print "Parsing error in feature_indentation (%s)" % e
# find which lines don't match the expected indentation
for line_no, line in enumerate(lines):
_levels = levels[line_no]
if _levels == None:
continue
expected_indents = []
for level in _levels:
try:
expected_indents.append(common_indentation[level])
except KeyError:
continue
indent = get_indent(line)
if expected_indents and indent not in expected_indents:
error = "expected indent %i, got %i" % (expected_indents[0], indent)
self.add_to_annotations(line_no, error, annotations)