comm_rm.py

import sys

def remove_comments(source):
    inside_string = False
    inside_block_comment = False
    string_char = None
    block_comment_char = None
    new_source = []
    for line in source:
        new_line = ''
        i = 0
        while i < len(line):
            if not inside_string and line[i:i+3] in ['"""', "'''"]:  # Check for block comment start or end
                if inside_block_comment and block_comment_char == line[i:i+3]:
                    inside_block_comment = False
                elif not inside_block_comment:
                    inside_block_comment = True
                    block_comment_char = line[i:i+3]
                i += 3
            elif not inside_block_comment and line[i] in ['"', "'"]:  # Check for string start
                # Toggle inside_string and copy the string literal
                inside_string = not inside_string
                string_char = line[i]
                start = i
                i += 1
                # Handle string literal
                while i < len(line) and (line[i] != string_char or (i < len(line) - 1 and line[i:i+2] == string_char*2)):
                    i += 1
                i += len(string_char)
                new_line += line[start:i]
            elif not inside_string and not inside_block_comment and line[i] == '#':  # Comment detected outside of a string
                break  # Ignore the rest of the line
            else:
                new_line += line[i]  # Copy other characters
                i += 1
        if new_line.strip() and not inside_block_comment:  # Check if the line is not empty and not inside a block comment
            new_source.append(new_line)  # Add the line to the new source
    return new_source

def main():
    if len(sys.argv) != 2:
        print("Usage: python comm_rm.py input_file.py")
        sys.exit(1)
    
    input_filename = sys.argv[1]
    output_filename = input_filename.replace('.py', '_rm.py')
    
    try:
        with open(input_filename, 'r') as file:
            source_code = file.readlines()
        new_source = remove_comments(source_code)
        with open(output_filename, 'w') as file:
            file.writelines(new_source)  # Write lines without stripping newline characters
        print(f"Comments removed. Clean file saved as: {output_filename}")
    except FileNotFoundError:
        print(f"File {input_filename} not found.")
    except Exception as e:
        print(f"An error occurred: {e}")

if __name__ == '__main__':
    main()