toggle_comments.py

#!/bin/env python


def toggle_comment(lines, filename, startswith=False):
    '''Toggle comment on lines in filename

    Options:
        lines (list): lines to comment.
        filename (str): filname
        startswith (bool): allows providing start of a line vs exact matching
    '''
    uncommented_lines = ['%s\n' % line for line in lines]
    commented_lines = ['# %s' % line for line in uncommented_lines]
    new_lines = []
    with open(filename) as f:
        file_lines = f.readlines()

    for file_line in file_lines:
        if file_line in uncommented_lines:
            file_line = '# %s' % file_line
        elif file_line in commented_lines:
            file_line = file_line[2:]
        new_lines.append(file_line)

    with open(filename, 'w') as f:
        f.writelines(new_lines)