Reads stdin from the user console and stops input with Ctrl+C

main.py

"""
Reads stdin from the user console and stops input with Ctrl+C. Also support piped stdin.
Works only on Linux (and maybe MacOS ?)
"""
import sys
import os
import time
import fcntl

def read_user_stdin() -> bytes:
	os.system("stty cbreak") # Disable stdin buffering

	# Actually read from stdin
	BUFFER_SIZE=256
	data = bytearray()
	while True:
		try:
			chunk = sys.stdin.buffer.raw.read(BUFFER_SIZE)
		except KeyboardInterrupt:
			break
		if chunk == None:
			break
		data += chunk

	os.system("stty -cbreak") # Re-enable stdin buffering

	return bytes(data)

def read_pipe_stdin() -> bytes:
	return sys.stdin.read().encode()

def read_current_stdin() -> bytes:
	if os.isatty(0):
		return read_user_stdin()
	else:
		return read_pipe_stdin()

print(read_current_stdin())