Issue #13 - Fix TypeError when using regexes

bellybutton/cli.py

@@ -175,10 +175,11 @@ def lint(modified_only=False, project_directory='.', verbose=False):
         )
         for failure in failure_results:
             failures += 1
+            lines = file_contents.splitlines()
             print(failure_message.format(
                 path=relpath,
                 lineno=failure.lineno,
-                line=file_contents.splitlines()[failure.lineno - 1],
+                line=lines[min(failure.lineno, len(lines)) - 1] if lines else '',
                 rule=failure.rule,
             ))
 

bellybutton/linting.py

@@ -62,7 +62,7 @@ def lint_file(filepath, file_contents, rules):
         elif isinstance(rule.expr, re._pattern_type):
             matching_lines = {
                 file_contents[:match.start()].count('\n') + 1  # TODO - slow
-                for match in re.finditer(rule.expr)
+                for match in re.finditer(rule.expr, file_contents)
             }
         elif callable(rule.expr):
             matching_lines = set(rule.expr(file_contents))

setup.py

@@ -4,7 +4,7 @@
     name='bellybutton',
     packages=['bellybutton'],
     platforms='any',
-    version='0.2.1',
+    version='0.2.2',
     description='Custom Python linting through AST expressions.',
     author='H. Chase Stevens',
     author_email='chase@chasestevens.com',