censor-code.py

import re
import sys
import random

def replace_char(x):
    if x.isalpha():
        return random.choice("αβγδεζηθικλμνξοπρστυφχψω")
    else:
        return x

def censor(input_string):
    """
    Replaces all characters in a string with random characters,
    while preserving ANSI escape codes.
    """
    # https://stackoverflow.com/a/14693789/18696276
    ansi_escape_pattern = r"\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])"
    parts = re.split(f"({ansi_escape_pattern})", input_string)
    replaced_parts = [
        (
            "".join(replace_char(c) for c in part)
            if not re.match(ansi_escape_pattern, part)
            else part
        )
        for part in parts
    ]

    return "".join(replaced_parts)


x = sys.stdin.read()
print(censor(x))