Python utility for writing to standard output

stdio.py

# coding=UTF-8
"""
@example
from stdio import stdio
stdio.write('hello world')
data = stdio.read()
print stdio.STDIN_FILENO
print stdio.STDOUT_FILENO
"""
import thread
import sys, os
import socket

# socket read/write in multiple threads may cause unexpected behaviors
# so use two separated sockets for stdin and stdout

def _sock_stdio():
	def stdin_thread(sock, console):
		"""	read data from stdin, and write the data to sock
		"""
		try:
			fd = sys.stdin.fileno()
			while True:
				# DO NOT use sys.stdin.read(), it is buffered
				data = os.read(fd, 1024)
				#print 'stdin read: ' + repr(data)
				if not data:
					break
				while True:
					nleft = len(data)
					nleft -= sock.send(data)
					if nleft == 0:
						break
		except:
			pass
		#print 'stdin_thread exit'
		sock.close()

	def stdout_thread(sock, console):
		"""	read data from sock, and write to stdout
		"""
		try:
			fd = sys.stdout.fileno()
			while True:
				data = sock.recv(1024)
				#print 'stdio_sock recv: ' + repr(data)
				if not data:
					break
				while True:
					nleft = len(data)
					nleft -= os.write(fd, data)
					if nleft == 0:
						break
		except:
			pass
		#print 'stdin_thread exit'
		sock.close()


	class Console:
		def __init__(self):
			self.serv = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
			self.serv.bind(('127.0.0.1', 0))
			self.serv.listen(5)
			port = self.serv.getsockname()[1]

			# data read from stdin will write to this socket
			self.stdin_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
			self.stdin_sock.connect(('127.0.0.1', port))
			self.s_in, addr = self.serv.accept()

			# data read from this socket will write to stdout
			self.stdout_sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
			self.stdout_sock.connect(('127.0.0.1', port))
			self.s_out, addr = self.serv.accept()

			self.read_str = '' # read buffer for readline

			self.STDIN_FILENO = self.s_in.fileno()
			self.STDOUT_FILENO = self.s_out.fileno()

			thread.start_new_thread(stdin_thread, (self.stdin_sock, self))
			thread.start_new_thread(stdout_thread, (self.stdout_sock, self))

		def close(self):
			self.s_in.close()
			self.s_out.close()
			self.stdin_sock.close()
			self.stdout_sock.close()
			self.serv.close()

		def write(self, data):
			try:
				return self.s_out.send(data)
			except:
				return -1

		def read(self):
			try:
				data = self.s_in.recv(4096)
			except:
				return ''
			ret = self.read_str + data
			self.read_str = ''
			return ret

		def readline(self):
			while True:
				try:
					data = self.s_in.recv(4096)
				except:
					return ''
				if not data:
					return ''
				pos = data.find('\n')
				if pos == -1:
					self.read_str += data
				else:
					left = data[0 : pos + 1]
					right = data[pos + 1 : ]
					ret = self.read_str + left
					self.read_str = right
					return ret

	stdio = Console()
	return stdio

def _os_stdio():
	class Console:
		def __init__(self):
			self.STDIN_FILENO = sys.stdin.fileno()
			self.STDOUT_FILENO = sys.stdout.fileno()

		def close(self):
			pass

		def write(self, data):
			try:
				return os.write(self.STDOUT_FILENO, data)
			except:
				return -1

		def read(self):
			try:
				return os.read(self.STDIN_FILENO, 4096)
			except:
				return ''

		def readline(self):
			try:
				return sys.stdin.readline()
			except:
				return ''

	stdio = Console()
	return stdio

if os.name == 'posix':
	stdio = _os_stdio()
else:
	stdio = _sock_stdio()