Extract all print statements from a Python project

print_extractor.py

import ast
import os
import sys

location = os.path.abspath(sys.argv[1])

exclude_from_check = ['twisted/plugins/tunnel_helper_plugin.py']

for (dirpath, dirnames, filenames) in os.walk(location):
    for filename in filenames:
        if not filename.endswith('.py'):
            continue
        file_contents = ''
        file_path = os.path.join(dirpath, filename)
        if any([file_path.endswith(excluded_path) for excluded_path in exclude_from_check]):
            continue
        with open(file_path, 'r') as file:
            file_contents = file.read()
        if file_contents:
            node = ast.parse(file_contents)
            for subnode in ast.walk(node):
                if isinstance(subnode, ast.Print):
                    print 'print at line {}, col {} in file {}'.format(subnode.lineno, subnode.col_offset, file_path)