Adds parentheses to print statements in a python file.

add_print_parentheses.py

import re

def add_print_parentheses(file_name):
    match = None  # a regex match when we are in a print statment
    lines = []
    with open(file_name) as file:
        for line in file:
            line = line.rstrip()
            if match:  # we are in a print statement
                indentation = re.match(r'[ ]*', line).span()[1]
                if indentation > match_indentation:
                    match_lines.append(line)
                else:
                    # the last line was the final line in the print statement
                    match_lines[-1] = match_lines[-1] + ')'
                    lines.extend(match_lines)
                    lines.append(line)
                    match = None
            else:
                match = re.match(r'([ ]*)print', line)
                if match:
                    match_indentation = match.span(1)[1]
                    match_len = match.span()[1]
                    match_lines = [line[:match_len] + '(' + line[match_len+1:]]
                else:
                    lines.append(line)

    with open(file_name + '3', 'w+') as file:
        file.write('\n'.join(lines))