Add newlines at strategic places in code string for printing.

    Args:
        s: str, piece of code. If not str, will attempt to convert to str.

    Returns:
        str

def make_code_readable(s):
    """Add newlines at strategic places in code string for printing.

    Args:
        s: str, piece of code. If not str, will attempt to convert to str.

    Returns:
        str
    """

    s = s if isinstance(s, str) else str(s)

    MAP = {",": ",\n", "{": "{\n ", "}": "\n}"}

    ll = []

    state = "open"
    flag_single = False
    flag_double = False
    flag_backslash = False
    for ch in s:
        if flag_backslash:
            flag_backslash = False
            continue

        if ch == "\\":
            flag_backslash = True
            continue

        if flag_single:
            if ch == "'":
                flag_single = False
        elif not flag_double and ch == "'":
            flag_single = True

        if flag_double:
            if ch == '"':
                flag_double = False
        elif not flag_single and ch == '"':
            flag_double = True

        if flag_single or flag_double:
            ll.append(ch)
        else:
            ll.append(MAP.get(ch, ch))

    return "".join(ll)