print_clearable.py

original_print = print
original_input = input
import io
from sys import stdout, stdin, stderr
from typing import Generic, Literal, TextIO, TypeVar
TextIO = TypeVar('TextIO')
class SupportsWrite(Generic[TextIO]):
    pass

global PRINT_BUFFER
PRINT_BUFFER:dict[SupportsWrite,str] = {}

def print(*args, sep:str | None = " ", end: str | None = "\n", file: SupportsWrite[str] | None = None, flush: Literal[False] = False):
    global PRINT_BUFFER
    g_file = file
    if file == None:
        file = stdout
    string = ''
    for i,e in enumerate(args):
        if isinstance(e, str):
            string += e
        else:
            string += str(e)
        if i != len(args) - 1 :
            string += sep if sep else ""
    string += end if end else ""
    if file in PRINT_BUFFER.keys():
        PRINT_BUFFER[file] += string
    else:
        PRINT_BUFFER[file] = string
    return original_print(string, sep=None, end="", flush=flush, file=g_file)

def input(__prompt):
    global PRINT_BUFFER
    string = ''
    if isinstance(__prompt, str):
        string += __prompt
    else:
        string += str(__prompt)
    if stdout in PRINT_BUFFER.keys():
        PRINT_BUFFER[stdout] += string
    else:
        PRINT_BUFFER[stdout] = string
    ret = original_input(string)
    if stdin in PRINT_BUFFER.keys():
        PRINT_BUFFER[stdin] += ret+"\n"
    else:
        PRINT_BUFFER[stdin] = ret+"\n"
    return ret

def clear(file:SupportsWrite[str]|None=None, remove_errors:bool=False):
    global PRINT_BUFFER
    g_file = file
    if file == None:
        file = stdout
    if not file in PRINT_BUFFER.keys():
        return
    try:
        original_print("\033[F\033[K\r" * PRINT_BUFFER[file].count("\n"), file=g_file, end="")
    except io.UnsupportedOperation:
        import sys
        original_print("\033[F\033[K\r" * PRINT_BUFFER[file].count("\n"), file=sys.stdout, end="")
        if not remove_errors:
            original_print(RuntimeWarning("File could not be written for clear! Using stdout."), file=sys.stderr)
    del PRINT_BUFFER[file]

def clear_cache(file:SupportsWrite[str]|None=None, all:bool=False):
    global PRINT_BUFFER
    if all:
        for i in list(PRINT_BUFFER.keys()):
            del PRINT_BUFFER[i]
        return
    if file == None:
        file = stdout
    if file in PRINT_BUFFER.keys():
        del PRINT_BUFFER[file]

import atexit
atexit.register(clear_cache, [], {all:True})